spring实现根据controller中接收请求参数不同走不同service

Posted on | Post modified: |

前言

前几天一个工程中，需要实现这样一个场景：根据前端发送过来的请求参数的不同，走不同的 service（可同事走多个），最初我的思路是尝试实现在 spring 中实现动态的依赖注入，也就是根据请求参数，动态的在 controller 中注入某个 service 接口的特定实现（接口有多个实现），但是发现这个实现不了，然后想了想，换了个思路，重新设计了一下，实现了需求中的场景。

正文

我的解决办法是，使用“生产线工人工作能力自己掂量机制”来解决，这名字我自己起的，实际上就是想要实现按参数选择走哪个 service 实现，可以一次性把所有 service 实现全都注入进来，然后依次请求，同时在每个 service 实现中写一套规则判别方法，判断当前请求自己是不是能够处理，如果能够处理，则进入处理方法，若自己没有处理能力，则退出，让请求走到其他 service 做同样的判断。形象点，可以想象一下，在一条生产线的传送带上传送着不同品类的待加工的元部件，有若干工人排列在传送带旁边，每个工人只会加工某一种元件，那么，当传送带上的元件传送到自己面前时，需要判断一下，自己有没有处理这个元件的能力（掂量一下自己的能力），若有，取过来处理，若没有，放过去让别人走流程。

理解了其中逻辑，我们就来看代码吧（片段）：

1
2
3
4
5
6
7
8
9

public interface ServiceProvider {

    // 掂量一下自己有没有能力加工当前的元件（也就是Request），能就返回 true 不能返回 false
    boolean support(Request request);

    // 具体加工元件的逻辑
    Response execute(Request request);
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

@Service
public class ServiceImpl implements Service {

    // 注入一系列 service 数量不定 https://stackoverflow.com/questions/2153298/how-to-autowire-factorybean
    @Resource(name = "serviceProviders")
    private List<ServiceProvider> serviceProviders;

    @Override
    public List<Response> execute(Request request) {
        return serviceProviders // 循环每个 service TODO 现在时间复杂度为 O(n) 可以尝试优化为 O(logn)
                  .stream()
                  .filter(serviceProvider -> serviceProvider.support(request)) // 按加工能力过滤
                  .map(serviceProvider -> serviceProvider.execute(request)) // 执行 service 得 execute 方法
                  .collect(Collectors.toList());
    }
}

这里有一点需要解释一下，在上面第二段代码中，有这么一段：

1
2
@Resource(name = "serviceProviders")
private List<ServiceProvider> serviceProviders;

这里是使用 spring 中的 FactoryBean 机制实现的，可以简单的这样理解 FactoryBean ：FactoryBean 是生成普通 Bean 的 Bean，当注入 FactoryBean 时，默认注入的是其生产出来的所有普通 Bean，而不是它自己。

在上边代码中，注入的名为 serviceProviders 的这个 Bean，实际上是这样定义出来的：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
@Component("serviceProviders") // 注意这个 Bean 的名字，当其他 Bean 中注入这个 Bean 时，会注入 createInstance() 返回类型的 Bean，而不是其自身的类型 ServiceProviderFactoryBean
public class ServiceProviderFactoryBean extends AbstractFactoryBean<List<ServiceProvider>> implements ApplicationContextAware {

    private ApplicationContext applicationContext;

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

    @Override
    protected List<ServiceProvider> createInstance() {
        // 扫描所有 provider 并从 Bean 容器取出放入 list
        Reflections reflections = new Reflections(ServiceProvider.class.getPackage().getName());
        return reflections
                .getSubTypesOf(ServiceProvider.class)
                .stream()
                .map((Function<Class<? extends ServiceProvider>, ServiceProvider>) serviceProviderClass -> applicationContext.getBean(serviceProviderClass))
                .collect(Collectors.toList());
    }

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.applicationContext = applicationContext;
    }
}

通过这样的设计，就完成了我们的需求，实际上我们等于把思路反转了一下，从想尽办法控制注入到不做控制，一股脑全部注入进去，然后按规则过滤。有时候，其实遇到一条思路走不通的时候，可以反过来想想，也许就会走通。

勘误

「后端圈」小伙伴指出，我这个注入一系列 Beans 的机制写复杂了，实际上在 Spring 中，注入 List<SomeBean> somebeans 就能自动实现将所有 SomeBean 全部注入进来了。压根不需要自己写 FactoryBean 的，这个机制的实现过程，可以参考 Spring 源码：org.springframework.beans.factory.support.DefaultListableBeanFactory#resolveMultipleBeans 和 org.springframework.beans.factory.support.DefaultListableBeanFactory#addCandidateEntry

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
// 单个Bean的类型为org.springframework.beans.factory.config.DependencyDescriptor，而List<Bean>的类型为org.springframework.beans.factory.support.DefaultListableBeanFactory.MultiElementDescriptor，从这里分成了两个流程
/**
 * Add an entry to the candidate map: a bean instance if available or just the resolved
 * type, preventing early bean initialization ahead of primary candidate selection.
 */
private void addCandidateEntry(Map<String, Object> candidates, String candidateName,
        DependencyDescriptor descriptor, Class<?> requiredType) {
    if (descriptor instanceof MultiElementDescriptor || containsSingleton(candidateName)) {
        Object beanInstance = descriptor.resolveCandidate(candidateName, requiredType, this);
        candidates.put(candidateName, (beanInstance instanceof NullBean ? null : beanInstance));
    }
    else {
        candidates.put(candidateName, getType(candidateName));
    }
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
@Nullable
private Object resolveMultipleBeans(DependencyDescriptor descriptor, @Nullable String beanName,
        @Nullable Set<String> autowiredBeanNames, @Nullable TypeConverter typeConverter) {
    Class<?> type = descriptor.getDependencyType();
    if (type.isArray()) {
        Class<?> componentType = type.getComponentType();
        ResolvableType resolvableType = descriptor.getResolvableType();
        Class<?> resolvedArrayType = resolvableType.resolve();
        if (resolvedArrayType != null && resolvedArrayType != type) {
            type = resolvedArrayType;
            componentType = resolvableType.getComponentType().resolve();
        }
        if (componentType == null) {
            return null;
        }
        Map<String, Object> matchingBeans = findAutowireCandidates(beanName, componentType, new MultiElementDescriptor(descriptor)); // 注意这里，这里 new 了一个 MultiElementDescriptor ，和上边那段代码的分支对应上了
        if (matchingBeans.isEmpty()) {
            return null;
        }
        if (autowiredBeanNames != null) {
            autowiredBeanNames.addAll(matchingBeans.keySet());
        }
        TypeConverter converter = (typeConverter != null ? typeConverter : getTypeConverter());
        Object result = converter.convertIfNecessary(matchingBeans.values(), type);
        if (getDependencyComparator() != null && result instanceof Object[]) {
            Arrays.sort((Object[]) result, adaptDependencyComparator(matchingBeans));
        }
        return result;
    }
    else if (Collection.class.isAssignableFrom(type) && type.isInterface()) {
        Class<?> elementType = descriptor.getResolvableType().asCollection().resolveGeneric();
        if (elementType == null) {
            return null;
        }
        Map<String, Object> matchingBeans = findAutowireCandidates(beanName, elementType,
                new MultiElementDescriptor(descriptor));
        if (matchingBeans.isEmpty()) {
            return null;
        }
        if (autowiredBeanNames != null) {
            autowiredBeanNames.addAll(matchingBeans.keySet());
        }
        TypeConverter converter = (typeConverter != null ? typeConverter : getTypeConverter());
        Object result = converter.convertIfNecessary(matchingBeans.values(), type);
        if (getDependencyComparator() != null && result instanceof List) {
            ((List<?>) result).sort(adaptDependencyComparator(matchingBeans));
        }
        return result;
    }
    else if (Map.class == type) {
        ResolvableType mapType = descriptor.getResolvableType().asMap();
        Class<?> keyType = mapType.resolveGeneric(0);
        if (String.class != keyType) {
            return null;
        }
        Class<?> valueType = mapType.resolveGeneric(1);
        if (valueType == null) {
            return null;
        }
        Map<String, Object> matchingBeans = findAutowireCandidates(beanName, valueType,
                new MultiElementDescriptor(descriptor));
        if (matchingBeans.isEmpty()) {
            return null;
        }
        if (autowiredBeanNames != null) {
            autowiredBeanNames.addAll(matchingBeans.keySet());
        }
        return matchingBeans;
    }
    else {
        return null;
    }
}