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

2023-12-21 10:25:32

现象

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

优化

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

案例介绍

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

  1. 枚举类
public enum SmsTypeEnum {
    CUCC, CMCC, CTCC
}
  1. 定义接口
public interface Sender {
    /**
     * 短信类型
     * 
     * @return 短信枚举
     */
    SmsTypeEnum smsType();

    /**
     * 发送短信
     * 
     * @param sms 短信内容
     */
    void sender(String sms);
}
  1. 三种运营商发送代码实现类
    移动
@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);
    }
}

  1. 初始化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;
    }
}
  1. 业务代码
@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

文章来源:https://blog.csdn.net/youxiaxiaomage/article/details/135077693
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。