项目中的if else太多了,该怎么重构?

2141
发表时间:2019-11-27 13:38

介绍

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


if (msgType = "文本") {

// dosomething

} else if(msgType = "图片") {

// doshomething

} else if(msgType = "视频") {

// doshomething

} else {

// doshomething

}

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


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


实现过程

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


@Data

@AllArgsConstructor

public class MessageInfo {


    // 消息类型

    private Integer type;

    // 消息内容

    private String content;


}

定义一个消息处理接口


public interface MessageService {


    void handleMessage(MessageInfo messageInfo);

}

有2个消息处理接口,分别处理不同的消息


处理文本消息


@Service

@MsgTypeHandler(value = MSG_TYPE.TEXT)

public class TextMessageService implements MessageService {


    @Override

    public void handleMessage(MessageInfo messageInfo) {

        System.out.println("处理文本消息 " + messageInfo.getContent());

    }

}

处理图片消息


@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来保存这种映射关系,来看看怎么做的把


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


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;

    }

}

定义一个注解


@Documented

@Inherited

@Target(ElementType.TYPE)

@Retention(RetentionPolicy.RUNTIME)

public @interface MsgTypeHandler {


    MSG_TYPE value();

}


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

处理哪种类型的消息


@Service

@MsgTypeHandler(value = MSG_TYPE.TEXT)

public class TextMessageService implements MessageService {


    @Override

    public void handleMessage(MessageInfo messageInfo) {

        System.out.println("处理文本消息 " + messageInfo.getContent());

    }

}


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


@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);

    }


}


最精彩的部分到了


@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对象中


@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代码,也更容易维护

————————————————

版权声明:本文为CSDN博主「Java识堂」的原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接及本声明。

原文链接:https://blog.csdn.net/zzti_erlie/article/details/102988486


文章分类: 技术论坛
热门文章

这篇文章很长,但绝对是精华,相信我,读完以后,你会知道学历不好的解决方案,记得帮我点赞哦。先说结论,无论赞不赞同,...

Java最常用的工具类,这些就够了

1、搜索引擎1.1、秘迹搜索一款无敌有良心、无敌安全的搜索引擎,不会收集私人信息,保护私隐,没有Cookie,并且...

就在昨天,一位叫小菜的读者微信我说了上面这段话。我当时看到这条微信的第一感觉是:小菜你也太菜了吧,这都不知道为啥啊...

包含的模块:本文分为十九个模块,分别是: Java 基础、容器、多线程、反射、对象拷贝、Java Web 、异常、...

官方公众号
打造IT教学生态圈,服务好每一位IT学员
官方微博
关于沃沃
与沃合作
核心课程
14天免费学习申请
UI设计表单
姓名
QQ号
手机号
提交