乐趣区

关于后端:基于注解实现的策略模式步骤简单通俗易懂

jiezi

背景

在我的项目开发的过程中，咱们常常会遇到如下的一种场景：对于用户的申请须要依据不同的状况进行不同的解决。

  • 最简略粗犷的一种解决形式是应用 switch…case 或者 if…else。然而这样解决形式只实用于解决逻辑简略或者状况分类较少的状况，如学校发放校服，男同学发放男士衣服，女同学发放女士衣服。
  • 然而，如果解决逻辑毕竟简单，或者状况分类较多，甚至将来有可能减少状况分类，上一种解决形式就会显得力不从心。此时应用策略模式将会是一种更优的解决形式。

根底配置 & 步骤

以下的计划是基于注解实现的策略模式。根底步骤 & 配置如下：

  • 定义策略名称：该项应用枚举实现
  • 定义策略名称注解：应用注解进行定义
  • 定义策略行为接口：该接口定义了策略行为
  • 定义策略处理器：蕴含策略名称的注解，并实现了策略行为接口
  • 定义策略容器：此处应用 map 作为策略容器，key 为策略名称注解，value 为策略处理器的实例
  • 初始化策略：容器初始化时候，从容器中读取蕴含策略名称注解的实例，并将其放入到策略容器中。

代码实现

在以下的例子中，会针对用户申请的 Msg 进行解析，msgType 有两种：一种是聊天音讯 ChatMsg，还有一种是零碎音讯 SysMsg。实现形式如下所示：

定义策略名称 

public enum MsgType {

    CHAT_MSG,
    SYS_MSG;
}

定义策略名称注解 

@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
public @interface Type {MsgType value();
}

定义策略行为接口 

public interface BaseMsgHandler {void handleMsg(String content);
}

定义策略处理器 

@Component
@Type(value = MsgType.CHAT_MSG)
public class ChatMsgHandler implements BaseMsgHandler{

    @Override
    public void handleMsg(String msg) {System.out.println("This is chatMsg. Detail msg information is :" + msg);
    }
}

@Component
@Type(value = MsgType.SYS_MSG)
public class SysMsgHandler implements BaseMsgHandler{

    @Override
    public void handleMsg(String msg) {System.out.println("This is sysMsg. Detail msg information is :" + msg);
    }
}

定义策略容器 

public static final Map<Type, BaseMsgHandler> msgStrategyMap = new HashMap<>();

初始化策略 

@Component
public class MsgConfig implements ApplicationContextAware {public static final Map<Type, BaseMsgHandler> msgStrategyMap = new HashMap<>();

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {applicationContext.getBeansWithAnnotation(Type.class).entrySet().iterator().forEachRemaining(entrySet ->{msgStrategyMap.put(entrySet.getKey().getClass().getAnnotation(Type.class),
                    (BaseMsgHandler) entrySet.getValue());
        });
    }
}

上述筹备动作实现后，就能够编写调用代码了：

import lombok.Data;

@Data
public class Msg{
    private String content;
    private MsgType msgType;
}

@RestController
@RequestMapping("/")
public class MsgController {@RequestMapping("msg")
    public void handleMsg(@RequestBody Msg msg){BaseMsgHandler handler = MsgConfig.msgStrategyMap.get(msg.getMsgType());
        handler.handleMsg(msg.getContent());
    }
}

最初

欢送关注公众号：前程有光，支付一线大厂 Java 面试题总结 + 各知识点学习思维导 + 一份 300 页 pdf 文档的 Java 外围知识点总结！

退出移动版