策略模式簡潔的實現方式，你知道嗎？
作者：小識
策略模式還挺簡單的，就是定義一個接口，然后有多個實現類，每種實現類封裝了一種行為。然后根據條件的不同選擇不同的實現類。
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中獲取到對應的策略類，然后處理消息。
責任編輯：武曉燕
來源：
Java識堂
