if else 太多了
最近跟着公司的大佬开发了一款IM系统，类似QQ和微信哈，就是聊天软件。我们有一部分业务逻辑是这样的
代码语言：javascript复制if (msgType = "文本") {
// dosomething
} else if(msgType = "图片") {
// doshomething
} else if(msgType = "视频") {
// doshomething
} else {
// doshomething
}
就是根据消息的不同类型有不同的处理策略，每种消息的处理策略代码都很长，如果都放在这种if else代码快中，代码很难维护也很丑，所以我们一开始就用了策略模式来处理这种情况。
策略模式还挺简单的，就是定义一个接口，然后有多个实现类，每种实现类封装了一种行为。然后根据条件的不同选择不同的实现类。
实现过程
消息对象，当然真实的对象没有这么简单，省略了很多属性
代码语言：javascript复制@Data
@AllArgsConstructor
public class MessageInfo {
// 消息类型
private MsgTypeEnum type;
// 消息内容
private String content;
}
消息类型是一个枚举类
代码语言：javascript复制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;
}
}
定义一个消息处理策略接口
代码语言：javascript复制public interface MessageStrategy {
void handleMessage(MessageInfo messageInfo);
}
有2个消息处理接口，分别处理不同的消息
处理文本消息
代码语言：javascript复制@Service
@MsgTypeHandler(value = MsgTypeEnum.TEXT)
public class TextMessageStrategy implements MessageStrategy {
@Override
public void handleMessage(MessageInfo messageInfo) {
System.out.println("处理文本消息 " messageInfo.getContent());
}
}
处理图片消息
代码语言：javascript复制@Service
@MsgTypeHandler(value = MsgTypeEnum.IMAGE)
public class ImageMessageStrategy implements MessageStrategy {
@Override
public void handleMessage(MessageInfo messageInfo) {
System.out.println("处理图片消息 " messageInfo.getContent());
}
}
可以看到上面我们用了一个MsgTypeHandler注解来表明策略类是用来处理哪种消息的？
代码语言：javascript复制@Documented
@Inherited
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public @interface MsgTypeHandler {
MsgTypeEnum value();
}
至此，所有代码就已经写完了，来跑一下测试代码
代码语言：javascript复制@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中获取到对应的策略类，然后处理消息
