面试时总被问，spring 中使用了哪些设计模式，你在实际开发中又使用哪些设计模式。给他手指一个方向跟他说，这就是一个模式：go out!。
这就是一种模式：策略模式，一个接口的多个实现方式(算法)。本文梳理了使用 springboot 实现的三种实现策略模式的应用
我们知道，springboot 应用初始化的过程是通过事件机制进行的。主要是通过 EventPublishingRunListener 在不同的初始化阶段发送不同的 SpringApplicationEvent (不同的子)事件，触发相应逻辑(这里的逻辑指 class 的加载)的加载和初始化。
当 ApplicationPreparedEvent 事件发送后，对于应用来讲，说明整个初始化过程已完成，也意味着所有的类已放入 spring ioc 中。
这时我们就可以结合自己的业务逻辑实现策略模式的应用，我们通过以下三种方式实现策略的应用
方式一：使用 ApplicationListener 和 ContextRefreshedEvent
核心使用的是 ApplicationContext.getBeanNamesForAnnotation(Class annotationType)方法，基于注解类，获取标有指定注解类的所有实例
我们的业务逻辑是这样的：应用 Api 接收规则参数(authType)，Api 根据 authType 值的不同，使用不同的 auth service，执行相应的规则业务逻辑。
public interface UserValidator<D, R> { String check(D data, R rule);œ}@Service@Validator(authType = AuthType.B_USER)public class BUserValidator implements UserValidator<String, String> { @Override public String check(String data, String rule) { System.out.println("客官，这里B端用户逻辑"); return ""; }}@Service@Validator(authType = AuthType.C_USER)public class CUserValidator implements UserValidator<String, String> { @Override public String check(String data, String rule) { System.out.println("客官，这里C端用户逻辑"); return ""; }}public enum AuthType { B_USER(1, "b端用户"), C_USER(2, "c端用户");
public final int type;
public final String code;
AuthType(int type, String code) { this.type = type; this.code = code; }}@Target(ElementType.TYPE)@Retention(RetentionPolicy.RUNTIME)@Documented@Component@Inheritedpublic @interface Validator {
AuthType authType();}@Componentpublic class AuthContainer implements ApplicationListener<ContextRefreshedEvent> { private ConcurrentHashMap<Integer, UserValidator> validatorMap = new ConcurrentHashMap<>();
@Override public void onApplicationEvent(ContextRefreshedEvent contextRefreshedEvent) { String[] names = context.getBeanNamesForAnnotation(Validator.class); if (names.length > 0) { for (String name : names) { System.out.println("UserValidator:" + name); UserValidator userValidator = context.getBean(name, UserValidator.class); Validator validator = userValidator.getClass().getAnnotation(Validator.class); validatorMap.put(validator.authType().type, userValidator); } } }
public ConcurrentHashMap<Integer, UserValidator> getValidatorMap() { return validatorMap; }}
复制代码
Api 接口定义如下，根据不同的 authType 值，执行不同的 auth service
@RequestMapping("/client_auth")@RestControllerpublic class Client3 { @Autowired private AuthContainer authContainer;
@RequestMapping("getAuth") public String getRule(@RequestParam("authType") Integer authType) { // if authType=1，B_USER; if authType=2，C_USER; ConcurrentHashMap<Integer, UserValidator> map = authContainer.getValidatorMap(); UserValidator userValidator = map.get(authType); String res = userValidator.check("hi", "看看什么规则"); return res; }}
复制代码
方式二：使用 ApplicationContextAware 和 @PostConstruct
我们的业务逻辑是这样的：应用 Api 接收规则参数(ruleType)，Api 根据 ruleType 值的不同，使用不同的 rule service，执行相应的规则业务逻辑。
核心使用的是 ApplicationContext.getBeanNamesForAnnotation(Class annotationType)方法，基于注解类，获取标有指定注解类的所有实例
代码结构：
核心代码如下：
public abstract class RuleValidator<D, R> { public abstract String check(D data, R rule);}@RuleMapping(ruleCodeEnum = RuleCodeEnum.COUNT)@Servicepublic class CountRuleValidator extends RuleValidator<String, String> { @Override public String check(String data, String rule) { System.out.println("客官，这里是数量规则区域"); return ""; }}@RuleMapping(ruleCodeEnum = RuleCodeEnum.PRICE)@Servicepublic class PriceRuleValidator extends RuleValidator<String, String> { @Override public String check(String data, String rule) { System.out.println("客官，这里是价格规则区域"); return ""; }}public enum RuleCodeEnum { /** 价格规则 */ PRICE(1, "price"), /** 数量规则 */ COUNT(2, "count");
public final int type; public final String code;
RuleCodeEnum(int type, String code) { this.type = type; this.code = code; }}@Target(ElementType.TYPE)@Retention(RetentionPolicy.RUNTIME)@Documented@Component@Inheritedpublic @interface RuleMapping { String ruleCode() default ""; String ruleDesc() default ""; RuleCodeEnum ruleCodeEnum();}
@Componentpublic class RuleValidatorInit implements ApplicationContextAware {
private static ApplicationContext context; private Map<Integer, RuleValidator> validatorMap = new HashMap<>();
@Override public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { this.context = applicationContext; } public Map<Integer, RuleValidator> getValidatorMap() { return validatorMap; } @PostConstruct public void afterPropertiesSet() { String[] names = context.getBeanNamesForAnnotation(RuleMapping.class); if (names.length > 0) { for (String name : names) { RuleValidator ruleValidator = context.getBean(name, RuleValidator.class); RuleMapping ruleMapping = ruleValidator.getClass().getAnnotation(RuleMapping.class); validatorMap.put(ruleMapping.ruleCodeEnum().type, ruleValidator); } } }}
复制代码
Api 接口定义如下，根据不同的 ruleType 值，执行不同的 rule service
@RequestMapping("/client")@RestControllerpublic class Client { @Autowired private RuleValidatorInit ruleValidatorInit;
@RequestMapping("getRule") public String getRule(@RequestParam("ruleType") Integer ruleType) { Map<Integer, RuleValidator> map = ruleValidatorInit.getValidatorMap(); System.out.println(map); RuleValidator<String, String> ruleValidator = map.get(ruleType); String res = ruleValidator.check("hi", "看看什么规则"); return res; }}
复制代码
式三：使用 ApplicationContextAware
对比方式一，这里不需要定义指定注解了
核心使用的是 ApplicationContext.getBeansOfType(classType)方法，获取接口 classType 的所有子类实例
代码结构：
核心代码如下：
public interface RoleValidator<D, R> { String check(D data, R rule); RoleCodeEnum source();}@Servicepublic class BRoleValidator implements RoleValidator<String, String> { @Override public String check(String data, String rule) { System.out.println("客官，这里B端用户逻辑"); return ""; } @Override public RoleCodeEnum source() { return RoleCodeEnum.B_USER; }}@Servicepublic class CRoleValidator implements RoleValidator<String, String> { @Override public String check(String data, String rule) { System.out.println("客官，这里C端用户逻辑"); return ""; } @Override public RoleCodeEnum source() { return RoleCodeEnum.C_USER; }}public enum RoleCodeEnum { B_USER(1, "b端用户"), C_USER(2, "c端用户");
public final int type; public final String code;
RoleCodeEnum(int type, String code) { this.type = type; this.code = code; }}@Componentpublic class RoleValidatorInit implements ApplicationContextAware { private static Map<RoleCodeEnum, RoleValidator> builderMap = new HashMap<>();
@Override public void setApplicationContext(ApplicationContext applicationContext) throws BeansException { for (RoleValidator roleValidator : applicationContext.getBeansOfType(RoleValidator.class).values()) { builderMap.put(roleValidator.source(), roleValidator); } } public static RoleValidator getRoleValidator(RoleCodeEnum role) { return builderMap.get(role); }}
复制代码
Api 接口定义如下，根据不同的 roleType 值，执行不同的 role service
@RequestMapping("/client_role")@RestControllerpublic class Client2 { @Autowired private RoleValidatorInit roleValidatorInit;
@RequestMapping("getRole") public String getRule(@RequestParam("roleType") Integer roleType) { // if roleType=1，B_USER; if roleType=2，C_USER; RoleValidator roleValidator = roleValidatorInit.getRoleValidator(RoleCodeEnum.B_USER); String res = roleValidator.check("hi", "看看什么规则"); return res; }}
复制代码
总结
通过三种方式的对比，对于核心部分，其实就是几行代码的不同。无论是利用 Event 事件还是通过 Aware，本质都是拿到或利用 ApplicationContext 去解析接口，拿到实现类的实例，放入集合，然后在客户端(或其他)获取集合，根据枚举标识拿到对应的子类，执行对应的业务逻辑。
评论