Added missing transactionManagers

mdaca · Dec 23, 2024 · 9e71e01 · 9e71e01
1 parent 016943b
commit 9e71e01
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/main/java/org/ohdsi/webapi/service/CDMResultsService.java b/src/main/java/org/ohdsi/webapi/service/CDMResultsService.java
@@ -44,6 +44,7 @@
 import org.springframework.jdbc.core.JdbcTemplate;
 import org.springframework.scheduling.annotation.Scheduled;
 import org.springframework.stereotype.Component;
+import org.springframework.transaction.PlatformTransactionManager;
 
 import jakarta.ws.rs.Consumes;
 import jakarta.ws.rs.GET;
@@ -134,6 +135,9 @@ public class CDMResultsService extends AbstractDaoService implements Initializin
     @Autowired
     private ConversionService conversionService;
 
+    @Autowired
+    private PlatformTransactionManager transactionManager;
+
     @Override
     public void afterPropertiesSet() throws Exception {
         queryRunner.init(this.getSourceDialect(), objectMapper);
@@ -555,14 +559,14 @@ private Step getAchillesStep(Source source, String jobStepName) {
         AchillesCacheTasklet achillesTasklet = new AchillesCacheTasklet(source, instance, cacheService,
                 queryRunner, objectMapper);
         return stepBuilderFactory.get(jobStepName + " achilles")
-                .tasklet(achillesTasklet)
+                .tasklet(achillesTasklet).transactionManager(transactionManager)
                 .build();
     }
 
     private Step getCountStep(Source source, String jobStepName) {
         CDMResultsCacheTasklet countTasklet = new CDMResultsCacheTasklet(source, cdmCacheService);
         return stepBuilderFactory.get(jobStepName + " counts")
-                .tasklet(countTasklet)
+                .tasklet(countTasklet).transactionManager(transactionManager)
                 .build();
     }