java通过反射获取加了某个注解的所有的类
一、前言
有时候我们会碰到这样的情况：
有n个场景，每个场景都有自己的逻辑，即n个处理逻辑，
这时候我们就需要通过某个参数的值代表这n个场景，然后去加载每个场景不同的bean对象，即不同的类，这些类中都有一个同名的方法，但是里面的逻辑不同，类似策略模式、工厂模式等
假设这样的场景，银行卡分几种类型，比如普通会员，黄金会员，黑卡会员......
普通会员、黄金会员和黑卡的权限不一样，消费能力不一样等等内容，这里我们就以消费的能力举例，普通会员只能转账5万元，黄金只能转账20万元，黑卡100万元，
我们有3各类，分别是Putong.class，Huangjin.class，Heika.class，这三个类里面有一个同样的方法，名字叫：checkPermissions()方法，三个类上都加了一个自定义注解@MemberTypeDefinition
二、代码demo参考
1、自定义注解
package com.zygxsq.test;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
/**
* 自定义注解
*
**/
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
public @interface MemberTypeDefinition {
/**
* 数据源默认执行方法
*/
public static final String DEFAULT_DAO_METHOD_NAME = "checkPermissions";
/**
* 数据源执行的方法名称, 默认为getData, 该方法必须只有一个参数{@link BaseDataSpec}, 返回结果必须是{@link BaseResult}, 可见性必须是public
*
* @return 方法名称
*/
String value() default DEFAULT_DAO_METHOD_NAME;
}2、基础model类
Putong.class
package com.zygxsq.test;
import org.springframework.stereotype.Component;
/**
* 普通会员处理逻辑
*
**/
@MemberTypeDefinition
@Slf4j
@Component
public class Putong {
public BaseResult checkPermissions(BaseDataSpec baseDataSpec) {
// 省略中间的一些处理逻辑
// ......
// 省略中间的一些处理逻辑
// 最后封装返回结果
BaseResult baseResult = new BaseResult();
baseResult.setResponse("我是普通会员");
return baseResult;
}
}Huangjin.class
package com.zygxsq.test;
import org.springframework.stereotype.Component;
/**
* 黄金会员处理逻辑
*
**/
@MemberTypeDefinition
@Slf4j
@Component
public class Huangjin {
public BaseResult checkPermissions(BaseDataSpec baseDataSpec) {
// 省略中间的一些处理逻辑
// ......
// 省略中间的一些处理逻辑
// 最后封装返回结果
BaseResult baseResult = new BaseResult();
baseResult.setResponse("我是黄金会员");
return baseResult;
}
}Heika.class
package com.zygxsq.test;
import org.springframework.stereotype.Component;
/**
* 黑卡会员处理逻辑
*
**/
@MemberTypeDefinition
@Slf4j
@Component
public class Heika {
public BaseResult checkPermissions(BaseDataSpec baseDataSpec) {
// 省略中间的一些处理逻辑
// ......
// 省略中间的一些处理逻辑
// 最后封装返回结果
BaseResult baseResult = new BaseResult();
baseResult.setResponse("我是黑卡会员");
return baseResult;
}
}3、入参和返回参数对象
入参对象
package com.zygxsq.test;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.experimental.SuperBuilder;
import java.io.Serializable;
import java.util.List;
/**
* 入参数据对象
*
**/
@Data
@SuperBuilder
@NoArgsConstructor
@AllArgsConstructor
public class BaseDataSpec implements Serializable {
/**
* serialVersionUID
*/
private static final long serialVersionUID = -5669150877852962345L;
/**
* id
*/
private String id;
/**
* 名字
*/
private String name;
}返回结果对象
package com.zygxsq.test;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;
import java.io.Serializable;
/**
* 返回结果对象
*
**/
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class BaseResult implements Serializable {
/**
* serialVersionUID
*/
private static final long serialVersionUID = 7510262928468530569L;
private String response;
}4、反射核心代码
那如何通过反射进行加载呢，BaseDataAnnotationApplication.class
package com.zygxsq.test;
import com.google.common.collect.Maps;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Component;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.Map;
/**
* 启动加载
*
**/
@Slf4j
@Component
public class BaseDataAnnotationApplication implements ApplicationContextAware, InitializingBean {
private ApplicationContext applicationContext;
/**
* <className,V>
*/
private final Map<String, Object> loadMap = Maps.newConcurrentMap();
/**
* 特殊人群表缓存 <id,1> ,获取 SpecialPeople.class里面所有的表名
*/
public static Map<String,Object> dmTablesMap = Maps.newConcurrentMap();
@Override
public void afterPropertiesSet() throws Exception {
// 获取加了 MemberTypeDefinition 注解的源表bean
loadSourceDefinition();
// 获取SpecialPeople.class里面所有的表名
loadSpecialMap();
}
private void loadSourceDefinition() throws Exception{
Map<String, Object> beans = applicationContext.getBeansWithAnnotation(MemberTypeDefinition.class);
for (final Object serviceObject : beans.values()) {
final Class<? extends Object> calcuteClass = serviceObject.getClass();
MemberTypeDefinition annotation = calcuteClass.getAnnotation(MemberTypeDefinition.class);
if (null == annotation) {
log.error("类： {} 注解缺失", calcuteClass);
continue;
}
loadMap.put(calcuteClass.getName(), serviceObject);
}
}
/**
* 获取SpecialPeople.class里面所有的表名
* @throws Exception
*/
private void loadSpecialMap() throws Exception {
dmTablesMap = Maps.newConcurrentMap();
Field[] declaredFields = SpecialPeople.class.getDeclaredFields();
for (Field declaredField : declaredFields) {
Class<?> type = declaredField.getType();
String typeName = declaredField.getGenericType().getTypeName();
Class<?> aClass = Class.forName(typeName);
if (type == People.class) {
People people = (People) declaredField.get(aClass);
String id = people.getId().trim();
dmTablesMap.put(id, "1");
}
}
}
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
this.applicationContext = applicationContext;
}
public <T> BaseResult buildData(Class<T> clazz, BaseDataSpec baseDataSpec) throws Exception{
// 获取执行方法名, 默认为getData
String methodName;
MemberTypeDefinition annotation = clazz.getAnnotation(MemberTypeDefinition.class);
if (annotation == null || StringUtils.isBlank(annotation.value())) {
methodName = MemberTypeDefinition.DEFAULT_DAO_METHOD_NAME;
} else {
methodName = annotation.value();
}
Method method;
Object bean = loadMap.get(clazz.getName());
BaseResult result = null;
try {
method = bean.getClass().getMethod(methodName, BaseDataSpec.class);
result = (BaseResult) method.invoke(bean, baseDataSpec);
} catch (NoSuchMethodException e) {
throw new Exception(clazz.getName()+"未找到执行方法："+methodName);
} catch (Exception e2) {
throw new Exception(clazz.getName()+"未找到执行方法："+methodName);
}
return result;
}
public <T> BaseResult buildData(Class<?> sourceClass) {
return null;
}
}通过上面的application，就可以加对象加载到缓存里了，然后我们直接调用即可
5、测试接口
package com.zygxsq.test;
import com.zygxsq.test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
@RestController
@RequestMapping("/users")
public class TestController {
@Autowired
private BaseDataAnnotationApplication baseDataAnnotationApplication;
@RequestMapping("/getUserType")
public String getUserType() {
// ...
// 省略查询表等逻辑
// ...
Class<test.Putong> putongClass = Putong.class;
BaseDataSpec baseDataSpec = BaseDataSpec.builder()
.id("888888")
.build();
BaseResult baseResult = baseDataAnnotationApplication.buildData(putongClass,baseDataSpec);
String response = baseResult.getResponse();
System.out.println(response);
return response;
}
}6、代码结构
所有代码结构，大致是这样，具体的一些代码可以放在不同的package下面，我这里仅仅是让大家可以直观的看到有这些类
本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
原始发表：2023-09-23，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录