Spring | 在Spring中使用@Resource注入List类型的Bean并按优先级排序

原创于 2025-05-25 16:16:07 发布 · 1.1k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#spring #java #spring boot

`CodingTechWork`

引言

在Spring框架中，依赖注入（DI）是核心功能之一，它允许开发者将Bean的创建和管理交给Spring容器，从而减少代码的耦合性。除了常见的单个Bean注入，Spring还支持注入集合类型的Bean，例如List、Set、Map等。本文将详细介绍如何使用@Resource注解注入List类型的Bean，并结合任务优先级注解来对实现类进行排序，最后通过一个业务示例来展示其应用场景。

为什么需要注入`List`类型的Bean并按优先级排序

在实际开发中，我们经常遇到需要组合多个服务（Service）来完成复杂任务的场景。例如，一个系统可能需要根据不同的业务规则生成多个子任务，并且这些任务可能有执行的先后顺序。通过注入List类型的Bean并按优先级排序，可以方便地管理和调用这些服务，而无需手动创建和管理它们的实例，同时保证任务按正确的顺序执行。

使用`@Resource`注入`List`类型的Bean并按优先级排序

基本概念

@Resource注解是Java的javax.annotation包中的注解，用于注入依赖。虽然@Autowired是Spring中更常用的注解，但@Resource同样可以实现依赖注入，并且在某些情况下（如注入List类型的Bean）表现得更为灵活。

示例代码

以下是一个完整的示例，展示如何使用@Resource注解注入List类型的Bean，并结合任务优先级注解来对实现类进行排序。

定义接口和实现类

首先，定义一个接口DemoBaseService，并实现多个具体的服务类。每个服务类通过@TaskPriority注解指定其优先级。

public interface DemoBaseService {
    List<String> generateTasks(Long domainId);
}

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@interface TaskPriority {
    int value();
}

@Component
@TaskPriority(1)
public class ServiceA implements DemoBaseService {
    @Override
    public List<String> generateTasks(Long domainId) {
        return Collections.singletonList("Task from Service A for Domain: " + domainId);
    }
}

@Component
@TaskPriority(2)
public class ServiceB implements DemoBaseService {
    @Override
    public List<String> generateTasks(Long domainId) {
        return Collections.singletonList("Task from Service B for Domain: " + domainId);
    }
}

注入`List`类型的Bean并按优先级排序

在主服务类中，使用@Resource注解注入List<DemoBaseService>，并根据@TaskPriority注解对服务进行排序。

@Service
public class TaskGenerationService {

    @Resource
    private List<DemoBaseService> services;

    @PostConstruct
    public void sortServicesByPriority() {
        services.sort(Comparator.comparingInt(service -> getPriority(service)));
    }

    private int getPriority(DemoBaseService service) {
        TaskPriority priority = service.getClass().getAnnotation(TaskPriority.class);
        return priority != null ? priority.value() : Integer.MAX_VALUE;
    }

    public List<String> generateAllTasks(Long domainId) {
        List<String> allTasks = new ArrayList<>();
        for (DemoBaseService service : services) {
            allTasks.addAll(service.generateTasks(domainId));
        }
        return allTasks;
    }
}

测试代码

编写一个简单的测试类来验证功能。

@SpringBootTest
public class TaskGenerationTest {

    @Autowired
    private TaskGenerationService taskGenerationService;

    @Test
    public void testGenerateAllTasks() {
        Long domainId = 1L;
        List<String> tasks = taskGenerationService.generateAllTasks(domainId);
        tasks.forEach(System.out::println);
    }
}

3.3 运行结果

运行测试代码后，输出如下：

Task from Service A for Domain: 1
Task from Service B for Domain: 1

业务场景示例

以下是一个更接近实际业务的示例，展示如何在复杂的业务逻辑中使用@Resource注入List类型的Bean并按优先级排序。

业务背景

假设我们正在开发一个任务管理系统，需要根据不同的业务规则生成多个子任务，并将这些子任务汇总后执行。每个服务类通过@TaskPriority注解指定其优先级，以确保任务按正确的顺序执行。

定义接口和实现类

定义一个接口DemoBaseService，并实现多个具体的服务类，每个服务类负责生成特定类型的子任务。

public interface DemoBaseService {
    List<String> generateTasks(Long domainId, List<Long> deviceIdList);
}

@Component
@TaskPriority(1)
public class ServiceA implements DemoBaseService {
    @Override
    public List<String> generateTasks(Long domainId, List<Long> deviceIdList) {
        return deviceIdList.stream()
                .map(deviceId -> "Task from Service A for Device: " + deviceId + " in Domain: " + domainId)
                .collect(Collectors.toList());
    }
}

@Component
@TaskPriority(2)
public class ServiceB implements DemoBaseService {
    @Override
    public List<String> generateTasks(Long domainId, List<Long> deviceIdList) {
        return deviceIdList.stream()
                .map(deviceId -> "Task from Service B for Device: " + deviceId + " in Domain: " + domainId)
                .collect(Collectors.toList());
    }
}

注入`List`类型的Bean并按优先级排序

在主服务类中，使用@Resource注解注入List<DemoBaseService>，并根据@TaskPriority注解对服务进行排序。

@Service
public class TaskGenerationService {

    @Resource
    private List<DemoBaseService> services;

    @PostConstruct
    public void sortServicesByPriority() {
        services.sort(Comparator.comparingInt(service -> getPriority(service)));
    }

    private int getPriority(DemoBaseService service) {
        TaskPriority priority = service.getClass().getAnnotation(TaskPriority.class);
        return priority != null ? priority.value() : Integer.MAX_VALUE;
    }

    public List<String> generateAllTasks(Long domainId, List<Long> deviceIdList) {
        List<String> allTasks = new ArrayList<>();
        for (DemoBaseService service : services) {
            allTasks.addAll(service.generateTasks(domainId, deviceIdList));
        }
        return allTasks;
    }
}

测试代码

编写一个简单的测试类来验证功能。

@SpringBootTest
public class TaskGenerationTest {

    @Autowired
    private TaskGenerationService taskGenerationService;

    @Test
    public void testGenerateAllTasks() {
        Long domainId = 1L;
        List<Long> deviceIdList = Arrays.asList(1L, 2L, 3L);
        List<String> tasks = taskGenerationService.generateAllTasks(domainId, deviceIdList);
        tasks.forEach(System.out::println);
    }
}

运行结果

运行测试代码后，输出如下：

Task from Service A for Device: 1 in Domain: 1
Task from Service A for Device: 2 in Domain: 1
Task from Service A for Device: 3 in Domain: 1
Task from Service B for Device: 1 in Domain: 1
Task from Service B for Device: 2 in Domain: 1
Task from Service B for Device: 3 in Domain: 1

总结

通过使用@Resource注解注入List类型的Bean并结合任务优先级注解来对实现类进行排序，我们可以在Spring框架中灵活地管理和调用多个服务。这种方式不仅简化了代码，还提高了代码的可维护性和可扩展性。在实际业务中，这种技术可以应用于多种场景，如任务生成、数据处理等，帮助开发者更高效地实现复杂的业务逻辑。