看到同事的代码,想到多年以前自己刚开始工作的时候写的代码,即使有再多的需求,再多的业务逻辑,都是满屏的if-else解决的,全然忘记什么叫做“面向对象编程”,但是写的多了都忘记了哪里是头,哪里是尾,但是当时并没有感觉到有多别扭,等慢慢的工作之后,回头再去看之前的代码,我艹,这是哪个**写的代码。
可能刚入门的同行可能会忽略一点,if-else可能是面向过程的实现。
那么如何解决掉满屏的if-else呢?让代码看起来变的更加的优雅、简洁呢?
我一般都是使用工厂模式 + 自定义注解 + 反射 + 枚举 + Map集合来实现一个策略模式,但也不是只要出现if-else就使用策略模式,当if-else => 4 , 我就会考虑使用这一套策略模式。
那么具体如何使用呢?我们通过一段业务来说明。
比如现在某一个平台有不同等级的用户,分别分为普通用户、黄金用户、铂金用户、钻石用户,需要通过不同的等级的用户查询相关的数据,比如优惠力度等。
一、首先定义一个自定义注解 :StrategyType,用于标记具体的策略类。
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.TYPE)
public @interface StrategyType {
/**
* 策略类型对应的具体的策略的编码
* @return 策略类型对应的具体的策略的编码
*/
String code() ;
/**
* 策略类型枚举编码
* @return 枚举类型编码
*/
StrategyTypeEnum type() ;
}
二、然后定义一个枚举类 :StrategyTypeEnum,用于列举所有支持的策略。
public enum StrategyTypeEnum {
USER_STRATEGY_TYPE("userStrategyType" , "用户类型策略枚举编码");
/*
* @TODO 如果有其它策略也可以加,比如根据不同的角色加载不同的数据 RoleStrategyType
* ROLE_STRATEGY_TYPE("roleStrategyType" ," 角色策略枚举编码")
*/
/**
* 策略类型编码
*/
private final String strategyTypeCode;
/**
* 策略类型描述
*/
private final String strategyTypeDesc;
/**
*
* @param strategyTypeCode 策略类型编码
* @param strategyTypeDesc 策略类型描述
*/
StrategyTypeEnum(String strategyTypeCode, String strategyTypeDesc) {
this.strategyTypeCode = strategyTypeCode;
this.strategyTypeDesc = strategyTypeDesc;
}
public String getStrategyTypeCode() {
return strategyTypeCode;
}
public String getStrategyTypeDesc() {
return strategyTypeDesc;
}
/**
* 根据策略类型编码获取枚举
* @param strategyTypeCode 策略类型编码
* @return StrategyTypeEnum 策略类型枚举
* @throws NullPointerException
*/
public static StrategyTypeEnum getUserTypeEnumByCode(String strategyTypeCode) throws NullPointerException {
if(strategyTypeCode.isEmpty()){
throw new NullPointerException("策略类型编码不能为空");
}
for (StrategyTypeEnum strategyTypeEnum : StrategyTypeEnum.values()) {
if(strategyTypeEnum.strategyTypeCode.equals(strategyTypeCode)){
return strategyTypeEnum;
}
}
return null;
}
}
三、再然后、定义一个策略类型编码常量类 :StrategyTypeConstant,用来放不同策略对应的具体的策略的编码。
public interface StrategyTypeConstant {
/**
* 普通用户
*/
public static final String USER_TYPE_NORMAL = "NORMAL";
/**
* 普通用户
*/
public static final String USER_TYPE_GOLD = "GOLD";
/**
* 普通用户
*/
public static final String USER_TYPE_PLATINUM = "PLATINUM";
/**
* 普通用户
*/
public static final String USER_TYPE_DIAMOND = "DIAMOND";
/*
* @TODO 可以进行扩展 , 其它策略的编码
*/
}
四、接着,我们创建一个策略工厂StrategyFactory,负责根据注解值获取对应的策略实例。
@Component
public class StrategyTypeFactory implements ApplicationContextAware {
private final Map<String, UserDataHandler> handlerMap = new HashMap<>();
public UserDataHandler getUserDataHandler(String userTypeCode) {
return handlerMap.get(userTypeCode);
}
/**
* Set the ApplicationContext that this object runs in.
* Normally this call will be used to initialize the object.
* <p>Invoked after population of normal bean properties but before an init callback such
* as {@link InitializingBean#afterPropertiesSet()}
*
* @param applicationContext the ApplicationContext object to be used by this object
* @throws BeansException if thrown by application context methods
* @see BeanInitializationException
*/
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
// 添加用户数据处理策略
addUserDataHandlerStrategy(applicationContext);
// @TODO 可以进行扩展 , 添加其它策略....
}
private void addUserDataHandlerStrategy(ApplicationContext applicationContext){
Map<String, Object> beansWithAnnotation = applicationContext.getBeansWithAnnotation(StrategyType.class);
beansWithAnnotation.entrySet().iterator().forEachRemaining(entrySet ->{
StrategyType annotation = entrySet.getValue().getClass().getAnnotation(StrategyType.class);
// 获取到用户类型枚举
String userTypeCode = annotation.code();
// 获取到策略类型枚举
String strategyTypeCode = annotation.type().getStrategyTypeCode();
// 策略的类型编码以及用户类型编码拼接为key , 表示某某策略类型的某某编码对应的策略
// key : strategyTypeCode + ":" + userTypeCode表示 : 用户类型策略的普通用户
// value : 对应的普通用户的处理类
this.handlerMap.put(strategyTypeCode + ":" + userTypeCode , (UserDataHandler) entrySet.getValue());
});
}
}
五、继续、定义一个接口:UserDataHandler,所有具体用户策略类都要实现该接口。
public interface UserDataHandler {
/**
* 处理用户数据
*/
Object handleUserData();
}
六、然后,我们创建四个具体的用户策略类NormalUserStrategy、GoldUserStrategy、PlatinumUserStrategy、DiamondUserStrategy,并使用@StrategyType注解标记每个策略类的类型:
@StrategyType(code = StrategyTypeConstant.USER_TYPE_DIAMOND , type = StrategyTypeEnum.USER_STRATEGY_TYPE)
@Service
public class DiamondUserDataHandler implements UserDataHandler{
/**
* 处理用户数据
*
* @return 处理结果
*/
@Override
public Object handleUserData() {
// @TODO 对应的业务逻辑
System.out.println("钻石用户业务逻辑");
return "钻石用户业务逻辑";
}
}
@StrategyType(code = StrategyTypeConstant.USER_TYPE_GOLD , type = StrategyTypeEnum.USER_STRATEGY_TYPE)
@Service
public class GoldUserDataHandler implements UserDataHandler{
/**
* 处理用户数据
*
* @return 处理结果
*/
@Override
public Object handleUserData() {
// @TODO 对应的业务逻辑
System.out.println("黄金用户业务逻辑");
return "黄金用户业务逻辑";
}
}
@StrategyType(code = StrategyTypeConstant.USER_TYPE_NORMAL , type = StrategyTypeEnum.USER_STRATEGY_TYPE)
@Service
public class NormalUserDataHandler implements UserDataHandler{
/**
* 处理用户数据
*
* @return 处理结果
*/
@Override
public Object handleUserData() {
// @TODO 对应的业务逻辑
System.out.println("普通用户业务逻辑");
return "普通用户业务逻辑";
}
}
@StrategyType(code = StrategyTypeConstant.USER_TYPE_PLATINUM , type = StrategyTypeEnum.USER_STRATEGY_TYPE)
@Service
public class PlatinumUserDataHandler implements UserDataHandler{
/**
* 处理用户数据
*
* @return 处理结果
*/
@Override
public Object handleUserData() {
// @TODO 对应的业务逻辑
System.out.println("铂金用户业务逻辑");
return "铂金用户业务逻辑";
}
}
七、测试:
@RestController
@RequestMapping("sysUser")
public class UserDataController {
@Resource
private StrategyTypeFactory strategyTypeFactory;
@RequestMapping("userDataHandle")
public Result<Object> userDataHandle(String userTypeCode){
// 根据不同的用户类型执行不同的策略
UserDataHandler userDataHandler = strategyTypeFactory.getUserDataHandler(StrategyTypeEnum.USER_STRATEGY_TYPE.getStrategyTypeCode() + ":" + userTypeCode);
if(userDataHandler == null){
return Result.error(HttpStatusCodeEnum.ERROR);
}
return Result.success(userDataHandler.handleUserData());
}
}