Embedded ActiveMQ Artemis在Spring Batch测试中提前停止致查询失败

核心原因 #

嵌入式ActiveMQ Artemis服务器在Spring上下文销毁流程中，可能早于队列查询逻辑被销毁，导致查询时JMS连接/会话已失效，抛出NullPointerException。即使添加broker.useShutdownHook=false，也无法解决Spring Bean的销毁顺序问题——Artemis相关Bean可能优先于测试中的查询逻辑被销毁。

具体解决方案 #

1. 强制等待批处理作业执行完成 #

确保Spring Batch作业完全执行完毕、消息已发送到队列后，再执行查询操作：

@Test
void testBatchMessageSending() throws Exception {
    // 启动作业并等待执行完成
    JobExecution execution = jobLauncher.run(yourBatchJob, new JobParameters());
    // 验证作业状态为完成
    Assertions.assertEquals(BatchStatus.COMPLETED, execution.getStatus());
    
    // 此时再查询队列消息数
    long messageCount = getQueueMessageCount("your-queue-name");
    Assertions.assertEquals(5, messageCount); // 替换为你的预期值
}

// 封装队列查询逻辑
private long getQueueMessageCount(String queueName) {
    return jmsTemplate.browse(queueName, (session, browser) -> {
        int count = 0;
        Enumeration<?> messages = browser.getEnumeration();
        while (messages.hasMoreElements()) {
            messages.nextElement();
            count++;
        }
        return (long) count;
    });
}

2. 手动控制嵌入式Artemis的生命周期 #

绕过Spring自动管理的销毁流程，在测试结束后手动停止服务器：

@Autowired
private EmbeddedActiveMQServer embeddedArtemisServer;

@Autowired
private JobLauncher jobLauncher;
@Autowired
private Job yourBatchJob;
@Autowired
private JmsTemplate jmsTemplate;

@BeforeEach
void setUp() throws Exception {
    // 确保服务器启动（如果Spring未自动启动）
    if (!embeddedArtemisServer.isStarted()) {
        embeddedArtemisServer.start();
    }
}

@Test
void testBatchMessageSending() throws Exception {
    // 执行作业
    JobExecution execution = jobLauncher.run(yourBatchJob, new JobParameters());
    Assertions.assertEquals(BatchStatus.COMPLETED, execution.getStatus());
    
    // 查询队列
    long messageCount = getQueueMessageCount("your-queue-name");
    Assertions.assertEquals(5, messageCount);
}

@AfterEach
void tearDown() throws Exception {
    // 确保查询完成后再停止服务器
    if (embeddedArtemisServer.isStarted()) {
        embeddedArtemisServer.stop();
    }
}

3. 调整Spring Bean的销毁顺序 #

通过@DependsOn或@Order注解，确保Artemis服务器Bean晚于测试中使用的JMS组件销毁：

@Configuration
public class ArtemisConfig {
    @Bean(destroyMethod = "stop")
    @Order(Ordered.LOWEST_PRECEDENCE) // 最后销毁
    public EmbeddedActiveMQServer embeddedActiveMQServer() throws Exception {
        EmbeddedActiveMQServer server = new EmbeddedActiveMQServer();
        // 配置服务器参数，包括broker.useShutdownHook=false
        server.setBrokerConfig("broker.xml");
        server.start();
        return server;
    }
}

4. 验证JmsItemWriter的同步发送配置 #

确保JmsItemWriter使用的JmsTemplate是同步发送模式，避免消息还未写入队列就触发服务器销毁：

@Bean
public JmsItemWriter<YourMessageType> jmsItemWriter() {
    JmsItemWriter<YourMessageType> writer = new JmsItemWriter<>();
    JmsTemplate jmsTemplate = new JmsTemplate(connectionFactory());
    jmsTemplate.setDeliveryMode(DeliveryMode.PERSISTENT); // 确保消息持久化并同步发送
    writer.setJmsTemplate(jmsTemplate);
    writer.setQueue("your-queue-name");
    return writer;
}

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