前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >IM系统中我们是如何用策略模式避免大量的if else?

IM系统中我们是如何用策略模式避免大量的if else?

作者头像
Java识堂
发布2019-11-14 14:56:59
8280
发布2019-11-14 14:56:59
举报
文章被收录于专栏:Java识堂
介绍

最近跟着公司的大佬开发了一款IM系统,就是聊天软件。我们有一部分业务逻辑是这样的

代码语言:javascript
复制
if (msgType = "文本") {
    // dosomething
} else if(msgType = "图片") {
    // doshomething
} else if(msgType = "视频") {
    // doshomething
} else {
    // doshomething
}

就是根据消息的不同类型有不同的处理策略,每种消息的处理策略代码都很长,如果都放在这种if else代码块中,代码很难维护也很丑,所以我们一开始就用了策略模式来处理这种情况。

策略模式还挺简单的,就是定义一个接口,然后有多个实现类,每种实现类封装了一种行为。然后根据不同的条件选择不同的实现类。来看一下我们是怎么做的

实现过程

定义消息对象,当然真实的对象没有这么简单,省略了很多属性

代码语言:javascript
复制
@Data
@AllArgsConstructor
public class MessageInfo {

    // 消息类型
    private Integer type;
    // 消息内容
    private String content;

}

定义一个消息处理接口

代码语言:javascript
复制
public interface MessageService {

    void handleMessage(MessageInfo messageInfo);
}

有2个消息处理实现类,分别处理不同的消息

处理文本消息

代码语言:javascript
复制
@Service
@MsgTypeHandler(value = MSG_TYPE.TEXT)
public class TextMessageService implements MessageService {

    @Override
    public void handleMessage(MessageInfo messageInfo) {
        System.out.println("处理文本消息 " + messageInfo.getContent());
    }
}

处理图片消息

代码语言:javascript
复制
@Service
@MsgTypeHandler(value = MSG_TYPE.IMAGE)
public class ImageMessageService implements MessageService {

    @Override
    public void handleMessage(MessageInfo messageInfo) {
        System.out.println("处理图片消息 " + messageInfo.getContent());
    }
}

文章写到这,可能大多数人都会想到要需要如下一个Map, Map<消息类型,消息处理对象>,这样直接根据消息类型就能拿到消息处理对象,调用消息处理对象的方法即可。我们就是这样做的,但是我们不想手动维护这个Map对象,因为每次增加新的消息处理类,Map的初始化过程就得修改

我们使用了注解+ApplicationListener来保存这种映射关系

定义一个消息类型的枚举类

代码语言:javascript
复制
public enum MSG_TYPE {

    TEXT(1, "文本"),
    IMAGE(2, "图片"),
    VIDEO(3, "视频");

    public final int code;
    public final String name;

    MSG_TYPE(int code, String name) {
        this.code = code;
        this.name = name;
    }
}

定义一个注解

代码语言:javascript
复制
@Documented
@Inherited
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public @interface MsgTypeHandler {

    MSG_TYPE value();
}

不知道你注意到了没,前面的代码中,每种消息处理类上面都有一个@MsgTypeHandler注解,表明了这个处理类处理哪种类型的消息

代码语言:javascript
复制
@Service
@MsgTypeHandler(value = MSG_TYPE.TEXT)
public class TextMessageService implements MessageService {

    @Override
    public void handleMessage(MessageInfo messageInfo) {
        System.out.println("处理文本消息 " + messageInfo.getContent());
    }
}

用一个context对象保存了消息类型->消息处理对象的映射关系

代码语言:javascript
复制
@Component
public class MessageServiceContext {

    private final Map<Integer, MessageService> handlerMap = new HashMap<>();

    public MessageService getMessageService(Integer type) {
        return handlerMap.get(type);
    }

    public void putMessageService(Integer code, MessageService messageService) {
        handlerMap.put(code, messageService);
    }

}

最精彩的部分到了,看我们是如何初始化这个map的

代码语言:javascript
复制
@Component
public class MessageServiceListener implements ApplicationListener<ContextRefreshedEvent> {

    @Override
    public void onApplicationEvent(ContextRefreshedEvent event) {
        Map<String, Object> beans = event.getApplicationContext().getBeansWithAnnotation(MsgTypeHandler.class);
        MessageServiceContext messageServiceContext = event.getApplicationContext().getBean(MessageServiceContext.class);
        beans.forEach((name, bean) -> {
            MsgTypeHandler typeHandler = bean.getClass().getAnnotation(MsgTypeHandler.class);
            messageServiceContext.putMessageService(typeHandler.value().code, (MessageService) bean);
        });
    }
}

在spring的启动过程中,通过解析注解,将消息类型->消息处理对象的映射关系保存到MessageServiceContext对象中

写一个测试类

代码语言:javascript
复制
@Autowired
MessageServiceContext messageServiceContext;

@Test
public void contextLoads() {
    // 构建一个文本消息
    MessageInfo messageInfo = new MessageInfo(MSG_TYPE.TEXT.code, "消息内容");
    MessageService messageService = messageServiceContext.getMessageService(messageInfo.getType());
    // 输出为->处理文本消息 消息内容
    messageService.handleMessage(messageInfo);
}

可以看到文本消息被文本处理类所处理,通过策略模式避免了写大量的if else代码,也更容易维护

本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2019-11-13,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 Java识堂 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 实现过程
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档