【一起学设计模式】策略模式实战一：基于消息发送的策略模式实战

前言

最近准备学习下之前项目中用到的设计模式，这里代码都只展示核心业务代码，省略去大多不重要的代码。

代码大多是之前一起工作的小伙伴coding出来的，我这里做一个学习和总结，我相信技术能力的提高都是先从模仿开始的，学习别人的代码及设计思想也是一种提升的方式。

 

后续还会有观察者模式、责任链模式的博客产出，都是工作中正式运用到的场景输出，希望对看文章的你也有启发和帮助。

一、业务需求

 

我之前做过在线问诊的需求，业务复杂，很多节点需要出发消息推送，比如用户下单 需要给医生推送短信和push、医生接诊 需要给用户发送短信、push、微信等。产品说后期会有很多不同的节点触发消息发送。

 

这里就开始抽象需求，首先是发送消息，很多消息是同样的策略，只是组装的数据是动态拼接的，所以抽象出：buildSms()、buildPush()、buildWechat() 等构造消息体的方法，对于拼接字段相同的都采用同一策略，列入消息A、B需要通过医生id拼接消息，消息C、D需要通过用户id拼接消息，那么A、B就采用同一策略，C、D采用另一策略。

 

流程图大致如下：

 

各个业务系统 根据策略构造自己的消息体，然后通过kafka发送个底层服务，进行消息统一推送。

 

 

二、策略模式

 

策略模式（Strategy Pattern）指的是对象具备某个行为，但是在不同的场景中，该行为有不同的实现算法。比如一个人的交税比率与他的工资有关，不同的工资水平对应不同的税率。

策略模式 使用的就是面向对象的继承和多态机制，从而实现同一行为在不同场景下具备不同实现。

策略模式本质：分离算法，选择实现

 

主要解决在有多重算法相似的情况下，使用if...else 或者switch...case所带来的的复杂性和臃肿性。

 

 

代码示例：

class Client {
    public static void main(String[] args) {
        ICalculator calculator = new Add();
        Context context = new Context(calculator);
        int result = context.calc(1,2);
        System.out.println(result);
    }
 
 
    interface ICalculator {
        int calc(int a, int b);
    }
 
 
    static class Add implements ICalculator {
        @Override
        public int calc(int a, int b) {
            return a + b;
        }
    }
 
 
    static class Sub implements ICalculator {
        @Override
        public int calc(int a, int b) {
            return a - b;
        }
    }
 
 
    static class Multi implements ICalculator {
        @Override
        public int calc(int a, int b) {
            return a * b;
        }
    }
 
 
    static class Divide implements ICalculator {
        @Override
        public int calc(int a, int b) {
            return a / b;
        }
    }
 
 
    static class Context {
        private ICalculator mCalculator;
 
 
        public Context(ICalculator calculator) {
            this.mCalculator = calculator;
        }
 
 
        public int calc(int a, int b) {
            return this.mCalculator.calc(a, b);
        }
    }}

 

三、工作中实际代码演示

 

为了代码简洁和易懂，这里用的都是核心代码片段，主要看策略使用的方式以及思想即可。

 

1、消息枚举类，这里因为消息出发节点众多，所以每一个节点都会对应一个枚举类，枚举中包含短信、push、微信、私信等内容。

@Getter
public enum MsgCollectEnum {
 
    /**
     * 枚举入口：用户首次提问 给医生 文案内容(医生id拼连接)
     */
    FIRST_QUESTION_CONTENT(2101, 1, MsgSmsEnum.SMS_FIRST_QUESTION_CONTENT, MsgPushEnum.PUSH_FIRST_QUESTION_CONTENT, MsgWechatEnum.WECHAT_FIRST_QUESTION_CONTENT);
 
 
   /**
     * 短信文案：用户首次提问 给医生 文案内容
     */
    SMS_FIRST_QUESTION_CONTENT(STTurnLinkEnum.DOCTOR_QUESTION_SETTING_PAGE.getStoapp(), "您好，有一位用户向您发起咨询，请确认接单，赶快进入APP查看吧!{0}");
 
 
    /**
     * Push文案：用户首次提问 给医生 文案内容
     */
    PUSH_FIRST_QUESTION_CONTENT(STTurnLinkEnum.DOCTOR_QUESTION_SETTING_PAGE.getStoapp(), STPushAudioEnum.PAY_SUCCESS.getType(), "您好, 有一位用户向您发起了咨询服务");
 
 
    ......
}

 

 

2，消息节点触发代码

这里是构造上下文MsgContext，主要策略分发的逻辑在最后一行，这里也会作为重点来讲解

 

MsgContext msgContext = new MsgContext();
msgContext.setDoctorId(questionDO.getDoctorId());
msgContext.setReceiveUid(questionDO.getDrUid());
msgContext.setMsgType(MsgCollectEnum.FIRST_QUESTION_CONTENT.getType());
this.stContextStrategyFactory.doStrategy(String.valueOf(msgContext.getMsgType()), QuestionMsgStrategy.class).handleSeniority(msgContext);

 

3，策略分发

首先，通过QuestionMsgStrategy.class 找到对应所有的beanMap，然后通过自定义注解找到所有对应策略类，最后通过msgType找到指定的实现类。接着我们看下策略实现类

 

@Slf4j
public class STContextStrategyFactory {
    public <O extends STIContext, T extends STIContextStrategy<O>> STIContextStrategy<O> doStrategy(String type, Class<T> clazz) {
        Map<String, T> beanMap = STSpringBeanUtils.getBeanMap(clazz);
        if (MapUtils.isEmpty(beanMap)) {
            log.error("获取class:{} 为空", clazz.getName());
        }
        try {
            for (Map.Entry<String, T> entry : beanMap.entrySet()) {
                Object real = STAopTargetUtils.getTarget(entry.getValue());
                STStrategyAnnotation annotation = real.getClass().getAnnotation(STStrategyAnnotation.class);
                List<String> keySet = Splitter.on("-").omitEmptyStrings().trimResults().splitToList(annotation.type());
                if (keySet.contains(type)) {
                    return entry.getValue();
                }
            }
        } catch (Exception e) {
            log.error("获取目标代理对象失败:{}", e);
        }
        log.error("strategy type = {} handle is null", type);
        return null;
    }
}

 

 

4，策略实现类

通过自定义注解，然后解析msgType值找到指定策略类，通过不同的策略类构造的msg 发送给kafka。

 

@Component
@STStrategyAnnotation(type = "2101-2104-2113-2016", description = "发给医生，无其他附属信息")
public class QuestionMsgSimpleToDoctorStrategyImpl extends AbstractQuestionSendMsgStrategy {
 
 
    @Autowired
    private RemoteMsgService remoteMsgService;
    @Autowired
    private QuestionDetailService questionDetailService;
 
 
    @Override
    public StarSmsIn buildSmsIn(MsgContext context) {
        // do something
    }
 
 
    @Override
    public StarPushIn buildPushIn(MsgContext context) {
        // do something
    }
 
 
    ......
 
 
}
 
 
@Slf4j
public abstract class AbstractQuestionSendMsgStrategy implements QuestionMsgStrategy {
    /**
     * 构建短信消息
     *
     * @param context
     * @return
     */
    public abstract StarSmsIn buildSmsIn(MsgContext context);
 
 
    /**
     * 构建push消息
     *
     * @param context
     * @return
     */
    public abstract StarPushIn buildPushIn(MsgContext context);
 
 
    /**
     * 构建微信公众号
     *
     * @param context
     * @return
     */
    public abstract StarWeChatIn buildWeChatIn(MsgContext context);
 
 
     @Override
    public STResultInfo handleSeniority(MsgContext msgContext) {
        // buildMsg and send kafka
    }
}

 

 

四，策略模式缺点

整个消息系统的设计起初是基于此策略模式来实现的，但是在后续迭代开发中会发现越来越不好维护，主要缺点如下：

a、接入消息推送的研发同学需要了解每个策略类，对于相同的策略进行复用

b、节点越来越多，策略类也越来越多，系统不易维护

c、触发节点枚举类散落在各个业务系统中，经常会有相同的节点而不同的msgType

 

针对于上述的缺点，又重构了一把消息系统，此次是完全采用节点配置化方案，提供一个可视化页面进行配置，将要构造的消息体通过配置写入到数据库中，代码中通过不同的占位符进行数据动态替换。

这里就不再展示新版系统的代码了，重构后 接入方只需要构造msgContext即可，再也不需要自己手动去写不同的策略类了。