Spring/Spring Boot中如何按自定义作用域实例选择特定Bean

针对你的多租户Spring Boot应用场景，以下是符合Spring惯用方式的优雅解决方案，无需注入List<IService>手动选择Bean：

1. 为租户Bean添加标识注解 #

首先定义一个注解，用来标记每个IService实现对应的租户ID，清晰关联租户与Bean：

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
public @interface TenantIdentifier {
    String value(); // 对应租户唯一标识，如"tenantA"、"tenantB"
}

修改你的租户Service实现，添加该注解：

@Component
@TenantScope
@TenantIdentifier("tenantA")
class TenantAService implements IService {
    @Override
    public String compute() {
        return "Tenant A";
    }
}

@Component
@TenantScope
@TenantIdentifier("tenantB")
class TenantBService implements IService {
    @Override
    public String compute() {
        return "Tenant B";
    }
}

2. 实现IService动态代理工厂Bean #

创建一个FactoryBean，生成IService的动态代理，代理会在运行时根据当前租户上下文自动切换到对应实现：

@Component
public class TenantServiceProxyFactory implements FactoryBean<IService>, ApplicationContextAware {
    private ApplicationContext applicationContext;
    private Map<String, String> tenantBeanMap; // 缓存租户ID与Bean名称的映射

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.applicationContext = applicationContext;
        // 初始化时缓存映射关系，避免每次调用遍历Bean
        tenantBeanMap = new HashMap<>();
        for (String beanName : applicationContext.getBeanNamesForType(IService.class)) {
            Class<?> beanClass = applicationContext.getType(beanName);
            if (beanClass != null) {
                TenantIdentifier annotation = beanClass.getAnnotation(TenantIdentifier.class);
                if (annotation != null) {
                    tenantBeanMap.put(annotation.value(), beanName);
                }
            }
        }
    }

    @Override
    public IService getObject() throws Exception {
        return (IService) Proxy.newProxyInstance(
                IService.class.getClassLoader(),
                new Class[]{IService.class},
                (proxy, method, args) -> {
                    String tenantId = TenantContext.getCurrentTenant();
                    String beanName = tenantBeanMap.get(tenantId);
                    if (beanName == null) {
                        throw new IllegalArgumentException("未找到租户[" + tenantId + "]对应的IService实现");
                    }
                    // 从自定义TenantScope中获取当前租户的Bean实例
                    IService targetService = applicationContext.getBean(beanName, IService.class);
                    return method.invoke(targetService, args);
                }
        );
    }

    @Override
    public Class<?> getObjectType() {
        return IService.class;
    }

    @Override
    public boolean isSingleton() {
        return true;
    }
}

3. 直接注入IService到单例组件 #

现在你的CommonService可以直接注入IService，Spring会自动使用上述代理Bean，调用方法时自动切换到对应租户的实现：

@Component
class CommonService {
    @Autowired
    private IService service;

    public String doCompute() {
        return this.service.compute(); // 根据当前请求头的租户ID，自动返回"Tenant A"或"Tenant B"
    }
}

方案优势 #

• 完全符合Spring依赖注入的惯用方式，无需手动处理Bean选择逻辑
• 代理实现逻辑复用，所有注入IService的组件都能自动适配租户
• 新增租户只需添加带@TenantIdentifier注解的IService实现，扩展性强
• 初始化时缓存租户与Bean的映射，避免运行时重复遍历，性能更优

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