项目中的if else太多了,该怎么重构?
if else 太多了
if (msgType = "文本") { // dosomething } else if(msgType = "图片") { // doshomething } else if(msgType = "视频") { // doshomething } else { // doshomething }
就是根据消息的不同类型有不同的处理策略,每种消息的处理策略代码都很长,如果都放在这种if else代码快中,代码很难维护也很丑,所以我们一开始就用了策略模式来处理这种情况。
策略模式还挺简单的,就是定义一个接口,然后有多个实现类,每种实现类封装了一种行为。然后根据条件的不同选择不同的实现类。
实现过程
消息对象,当然真实的对象没有这么简单,省略了很多属性
@Data @AllArgsConstructor public class MessageInfo { // 消息类型 private MsgTypeEnum type; // 消息内容 private String content; }
消息类型是一个枚举类
public enum MsgTypeEnum { TEXT(1, "文本"), IMAGE(2, "图片"), VIDEO(3, "视频"); public final int code; public final String name; MsgTypeEnum(int code, String name) { this.code = code; this.name = name; } }
定义一个消息处理策略接口
public interface MessageStrategy { void handleMessage(MessageInfo messageInfo); }
有2个消息处理接口,分别处理不同的消息
处理文本消息
@Service @MsgTypeHandler(value = MsgTypeEnum.TEXT) public class TextMessageStrategy implements MessageStrategy { @Override public void handleMessage(MessageInfo messageInfo) { System.out.println("处理文本消息 " + messageInfo.getContent()); } }
处理图片消息
@Service @MsgTypeHandler(value = MsgTypeEnum.IMAGE) public class ImageMessageStrategy implements MessageStrategy { @Override public void handleMessage(MessageInfo messageInfo) { System.out.println("处理图片消息 " + messageInfo.getContent()); } }
可以看到上面我们用了一个MsgTypeHandler注解来表明策略类是用来处理哪种消息的?
@Documented @Inherited @Target(ElementType.TYPE) @Retention(RetentionPolicy.RUNTIME) public @interface MsgTypeHandler { MsgTypeEnum value(); }
至此,所有代码就已经写完了,来跑一下测试代码
@RunWith(SpringRunner.class) @SpringBootTest public class DemoApplicationTests { private Map<MsgTypeEnum, MessageStrategy> messageStrategyMap; @Resource private void setMessageStrategyMap(List<MessageStrategy> messageStrategies) { messageStrategyMap = messageStrategies.stream().collect( Collectors.toMap(item -> AnnotationUtils.findAnnotation(item.getClass(), MsgTypeHandler.class).value(), v -> v)); } @Test public void contextLoads() { MessageInfo messageInfo = new MessageInfo(MsgTypeEnum.TEXT, "这是一个文本消息"); MessageStrategy messageStrategy = messageStrategyMap.get(messageInfo.getType()); // 处理文本消息 这是一个文本消息 messageStrategy.handleMessage(messageInfo); } }
在setMessageStrategyMap方法上加@Resource注解,将消息类型及其对应的策略类保存到map中,当消息来临的时候就能从map中获取到对应的策略类,然后处理消息
参考博客
[1]https://juejin.im/post/5c5172d15188256a2334a15d
下一篇:
【Java SE】数据类型与变量