Springboot日常开发使用策略、工厂模式

现象

在日常开发中，可能存在很多代码逻辑相同的内容，只是有些指定的类型不同，所以代码中会出现很多if-else或者switch相关代码。比如我发送短信，短信内容都是相同的，只是运营商不同。写很多switch或者if-else会显得代码很臃肿，可读性比较差。

优化

其实上面的问题大家都能想到使用策略+工厂的方式去实现相应的业务逻辑，但是有很多人无法写好这一块代码，让代码简洁更加健壮。
下面就说说自己实际开发中的使用。

案例介绍

以发送短信为例
三大运营商，我们可以定义为枚举或者常量，个人喜欢使用枚举，看个人喜好

1. 枚举类

public enum SmsTypeEnum {
    CUCC, CMCC, CTCC
}

2. 定义接口

public interface Sender {
    /**
     * 短信类型
     * 
     * @return 短信枚举
     */
    SmsTypeEnum smsType();

    /**
     * 发送短信
     * 
     * @param sms 短信内容
     */
    void sender(String sms);
}

3. 三种运营商发送代码实现类
   移动

@Service
@Slf4j
public class CmccSender implements Sender {
    @Override
    public SmsTypeEnum smsType() {
        return SmsTypeEnum.CMCC;
    }

    @Override
    public void sender(String sms) {
        log.info("sms type {} send msg {}", smsType(), sms);
    }
}

联通

@Service
@Slf4j
public class CuccSender implements Sender {
    @Override
    public SmsTypeEnum smsType() {
        return SmsTypeEnum.CUCC;
    }

    @Override
    public void sender(String sms) {
        log.info("sms type {} send msg {}", smsType(), sms);
    }
}

电信

@Service
@Slf4j
public class CtccSender implements Sender {
    @Override
    public SmsTypeEnum smsType() {
        return SmsTypeEnum.CTCC;
    }

    @Override
    public void sender(String sms) {
        log.info("sms type {} send msg {}", smsType(), sms);
    }
}

4. 初始化Bean工厂(这里使用到springboot的扩展点@PostContruct)

@Component
@RequiredArgsConstructor
@Slf4j
public class SmsStrategy {
    private final List<Sender> senderList; // spring 可以注入List, Map
    private static Map<SmsTypeEnum, Sender> smsTypeEnumSenderMap = Collections.emptyMap();

    @PostConstruct
    public void init() {
        if (CollectionUtils.isEmpty(senderList)) {
            return;
        }
        smsTypeEnumSenderMap = senderList.stream().map(sender -> Pair.of(sender.smsType(), sender))
            .collect(Collectors.toMap(Pair::getKey, Pair::getValue, (v1, v2) -> v1));
        log.info("load sms sender map {}", smsTypeEnumSenderMap);
    }

    public Sender getSenderBySmsType(@NonNull SmsTypeEnum smsType) {
        Sender sender = smsTypeEnumSenderMap.get(smsType);
        Assert.notNull(sender, MessageFormat.format("smsType {0} sender is null", smsType));
        return sender;
    }
}

5. 业务代码

@Service
@RequiredArgsConstructor
public class SmsService {
    private final SmsStrategy smsStrategy;

    public void senderCmcc(String msg) {
        this.smsStrategy.getSenderBySmsType(SmsTypeEnum.CMCC).sender(msg);
    }

    public void senderCucc(String msg) {
        this.smsStrategy.getSenderBySmsType(SmsTypeEnum.CUCC).sender(msg);
    }

    public void senderCtcc(String msg) {
        this.smsStrategy.getSenderBySmsType(SmsTypeEnum.CTCC).sender(msg);
    }
}

示例代码

https://gitee.com/youxiaxiaomage/java-practices/tree/master/yxxmg-gof-sample/src/main/java/com/yxxmg/gof/spring/strategy