Spring Boot运行时动态切换SMS与Email消息服务的技术咨询

问题背景 #

我们使用Spring Boot 2.7.12（对应Spring版本5.3.27）开发，当前需修改application.properties中的spring.profiles.active=sms配置并重启应用，才能在SMSService与EmailService之间切换。现需要借助Spring框架的依赖注入（DI）实现无需重启应用的运行时动态切换。

原有代码 #

package net.portal;

public interface MessageService {
    void sendMessage(String message);
}

@Service
@Profile("!sms")
public class EmailService implements MessageService {
    @Override
    public void sendMessage(String message) {
        // 邮件发送逻辑
        System.out.println("Email sent: " + message);
    }
}

@Service
@Profile("sms")
public class SMSService implements MessageService {
    @Override
    public void sendMessage(String message) {
        // SMS发送逻辑
        System.out.println("SMS sent: " + message);
    }
}

@Component
public class MessageSender {
    private final MessageService messageService;
    
    @Autowired
    public MessageSender(MessageService messageService) {
        this.messageService = messageService;
    }
    
    public void sendMessage(String message) {
        messageService.sendMessage(message);
    }
}

// 应用启动类
private static ApplicationContext context;
public static void main(String[] args) {
    SpringApplication.run(ApplicationLauncher.class, args);
    MessageSender sender = context.getBean(MessageSender.class);
    sender.sendMessage("Hello, this is frm SMS message now!");
}

解决方案 #

步骤1：移除@Profile注解，让两个服务类都被Spring容器管理 #

去掉EmailService和SMSService上的@Profile注解，确保Spring启动时会将两个实现类都加载到容器中：

@Service
public class EmailService implements MessageService {
    @Override
    public void sendMessage(String message) {
        System.out.println("Email sent: " + message);
    }
}

@Service
public class SMSService implements MessageService {
    @Override
    public void sendMessage(String message) {
        System.out.println("SMS sent: " + message);
    }
}

注：Spring默认会将类名首字母小写作为bean名称，即emailService和smsService，后续可以通过这个名称获取对应实例。

步骤2：改造MessageSender，支持动态切换服务 #

修改MessageSender，注入所有MessageService实现类的Map，同时维护当前使用的服务类型，提供切换方法：

@Component
public class MessageSender {
    private final Map<String, MessageService> messageServiceMap;
    // 默认使用邮件服务
    private String currentServiceName = "emailService";

    // Spring会自动注入所有MessageService实现类，key为bean名称
    @Autowired
    public MessageSender(Map<String, MessageService> messageServiceMap) {
        this.messageServiceMap = messageServiceMap;
    }

    public void sendMessage(String message) {
        // 根据当前服务名称获取对应实例并执行发送逻辑
        messageServiceMap.get(currentServiceName).sendMessage(message);
    }

    // 提供外部调用的切换方法
    public void switchService(String serviceName) {
        if (messageServiceMap.containsKey(serviceName)) {
            this.currentServiceName = serviceName;
        } else {
            throw new IllegalArgumentException("不存在的服务类型：" + serviceName);
        }
    }
}

步骤3：添加切换触发方式（可选） #

如果需要通过外部接口触发切换，可以新增一个REST控制器：

@RestController
@RequestMapping("/message")
public class ServiceSwitchController {
    private final MessageSender messageSender;

    @Autowired
    public ServiceSwitchController(MessageSender messageSender) {
        this.messageSender = messageSender;
    }

    @PostMapping("/switch")
    public String switchService(@RequestParam String serviceName) {
        try {
            messageSender.switchService(serviceName);
            return "已成功切换到[" + serviceName + "]服务";
        } catch (IllegalArgumentException e) {
            return e.getMessage();
        }
    }
}

步骤4：测试动态切换 #

在main方法中可以直接测试切换逻辑：

private static ApplicationContext context;
public static void main(String[] args) {
    context = SpringApplication.run(ApplicationLauncher.class, args);
    MessageSender sender = context.getBean(MessageSender.class);
    
    // 初始发送邮件
    sender.sendMessage("初始测试消息");
    // 切换到SMS服务
    sender.switchService("smsService");
    sender.sendMessage("切换后的SMS消息");
}

原理说明 #

Spring容器会自动收集所有实现MessageService的bean，并注入到Map<String, MessageService>中，key对应bean的名称。通过维护当前使用的服务名称，在发送消息时动态从Map中获取对应的服务实例，从而实现运行时的服务切换，无需重启应用。

内容的提问来源于stack exchange，提问作者janespring