前段时间有个简单的短信登录功能,需要集成短信服务实现短信发送,一开始用的阿里云,后面领导又说用天翼云,还说阿里云的保留,以后说不准还会集成其他的服务商,于是用nacos的热更新来解决动态切换,又为了代码优雅一点,就用了策略模式来实现这个功能
策略模式结构
Strategy:策略接口
StrategyImpl:策略实现类
Factory:策略工厂
一、策略接口
public interface SmsStrategy {
/**
* 策略接口
*/
void sendSms(String phone);
}
二、策略实现类
@Component(value = "aliyun")
public class aliyunSmsStrategyImpl implements SmsStrategy {
@Override
public void sendSms(String phone) {
System.out.println("aliyun");
}
}
@Component(value = "tianyiyun")
public class tianyiyunSmsStrategyImpl implements SmsStrategy {
@Override
public void sendSms(String phone) {
System.out.println("tianyiyun");
}
}
三、策略工厂
这是策略模式最重要的,利用Spring的自动装配原则,在项目启动时就会将被 @Component 注解标识的策略实现类注入到容器中,通过策略工厂的构造方法即可拿到所有的策略实现类
@Component
public class SmsFactory {
private final Map<String, SmsStrategy> smsStrategies = Maps.newHashMapWithExpectedSize(4);
/**
* 利用构造函数在项目启动的时候将策略实现类注册到 map里
* @param strategyMap
*/
public SmsFactory(Map<String, SmsStrategy> strategyMap) {
this.smsStrategies.clear();
this.smsStrategies.putAll(strategyMap);
}
//可以使用@Getter注解代替,这样写方便读者理解在Service层调用Context执行策略
public Map<String, SmsStrategy> getSmsStrategyMap() {
return smsStrategies;
}
}
四、nacos热更新
springboot集成nacos这里不赘述了
将 @RefreshScope 注解加在用到了 @Value 注解的类上
@Service
@RefreshScope
public class SmsServiceImpl implements SmsService {
@Autowired
private SmsFactory smsFactory;
@Value("${sms.strategy}")
private String straetgyValue;
@Override
public void sendSms(String phone) {
smsFactory.getSmsStrategyMap().get(straetgyValue).sendSms("1353802907");
}
}
nacos配置
server:
port: 8991
spring:
application:
name: rpdemo
cloud:
nacos:
username: nacos
password: nacos
discovery:
enabled: true
server-addr: 192.168.233.1:8848
config:
enabled: true
server-addr: 192.168.233.1:8848
refresh-enabled: true
shared-configs:
- data-id: rpdemo.yaml
refresh: true