策略模式简洁的实现方式你知道吗？

​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 MapmessageStrategyMap;

@Resource

private void setMessageStrategyMap(ListmessageStrategies) {

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中获取到对应的策略类，然后处理消息。云服务器

很赞哦!（144）