if过多如何重构_多个else if用法

if过多如何重构_多个else if用法介绍最近跟着公司的大佬开发了一款IM系统,类似QQ和微信哈,就是聊天软件。我们有一部分业务逻辑是这样的if(msgType=”文本”){ //dosomething}elseif(msgType=”图片”){ //doshomething}elseif(msgType=”视频”){ //doshomething}else{ //doshom………

大家好,又见面了,我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全系列IDE稳定放心使用

在这里插入图片描述

if else 太多

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

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

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/186279.html原文链接:https://javaforall.cn

【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛

【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...

(0)
blank

相关推荐

发表回复

您的电子邮箱地址不会被公开。

关注全栈程序员社区公众号