在日常开发中,可能存在很多代码逻辑相同的内容,只是有些指定的类型不同,所以代码中会出现很多if-else
或者switch
相关代码。比如我发送短信,短信内容都是相同的,只是运营商不同。写很多switch或者if-else会显得代码很臃肿,可读性比较差。
其实上面的问题大家都能想到使用策略+工厂的方式去实现相应的业务逻辑,但是有很多人无法写好这一块代码,让代码简洁更加健壮。
下面就说说自己实际开发中的使用。
以发送短信为例
三大运营商,我们可以定义为枚举或者常量,个人喜欢使用枚举,看个人喜好
public enum SmsTypeEnum {
CUCC, CMCC, CTCC
}
public interface Sender {
/**
* 短信类型
*
* @return 短信枚举
*/
SmsTypeEnum smsType();
/**
* 发送短信
*
* @param sms 短信内容
*/
void sender(String sms);
}
@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);
}
}
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;
}
}
@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