[incubator-kie-issues#1473] Add withTransactional method to Dependenc…

…yInjectionAnnotator (#6090) * [incubator-kie-issues#1473] Add withTransactional method to DependencyInjectionAnnotator * [incubator-kie-issues#1473] Refactoring DependencyInjectionAnnotator. Add "Patch" annotation to RestAnnotators * [incubator-kie-issues#1473] Restore formatting --------- Co-authored-by: Gabriele-Cardosi <[email protected]>
apache · Oct 3, 2024 · 5448192 · 5448192
1 parent 7052421
commit 5448192
Show file tree

Hide file tree

Showing 5 changed files with 28 additions and 6 deletions.
diff --git a/...degen-common/src/main/java/org/drools/codegen/common/di/DependencyInjectionAnnotator.java b/...degen-common/src/main/java/org/drools/codegen/common/di/DependencyInjectionAnnotator.java
@@ -160,6 +160,18 @@ default <T extends NodeWithAnnotations<?>> T withInjection(T node) {
      */
     <T extends NodeWithAnnotations<?>> T withConfigInjection(T node, String configKey, String defaultValue);
 
+    /**
+     * Annotates given node with Transactional annotation
+     *
+     * @param node node to be annotated
+     */
+    default <T extends NodeWithAnnotations<?>> T withTransactional(T node)  {
+        node.addAnnotation(getTransactionalAnnotation());
+        return node;
+    }
+
+    String getTransactionalAnnotation();
+
     /**
      * Annotates and enhances method used to produce messages
      *

diff --git a/...mmon/src/main/java/org/drools/codegen/common/di/impl/CDIDependencyInjectionAnnotator.java b/...mmon/src/main/java/org/drools/codegen/common/di/impl/CDIDependencyInjectionAnnotator.java
@@ -178,6 +178,11 @@ public <T extends NodeWithAnnotations<?>> T withFactoryMethod(T node) {
         return node;
     }
 
+    @Override
+    public String getTransactionalAnnotation() {
+        return "jakarta.transaction.Transactional";
+    }
+
     @Override
     public <T extends NodeWithAnnotations<?>> T withTagAnnotation(T node, NodeList<MemberValuePair> attributes) {
         node.addAnnotation(new NormalAnnotationExpr(new Name("org.eclipse.microprofile.openapi.annotations.tags.Tag"), attributes));

diff --git a/...n/src/main/java/org/drools/codegen/common/di/impl/SpringDependencyInjectionAnnotator.java b/...n/src/main/java/org/drools/codegen/common/di/impl/SpringDependencyInjectionAnnotator.java
@@ -18,6 +18,10 @@
  */
 package org.drools.codegen.common.di.impl;
 
+import java.util.Collection;
+import java.util.Collections;
+import java.util.Optional;
+
 import com.github.javaparser.ast.NodeList;
 import com.github.javaparser.ast.expr.BinaryExpr;
 import com.github.javaparser.ast.expr.BooleanLiteralExpr;
@@ -36,10 +40,6 @@
 import com.github.javaparser.ast.type.ClassOrInterfaceType;
 import org.drools.codegen.common.di.DependencyInjectionAnnotator;
 
-import java.util.Collection;
-import java.util.Collections;
-import java.util.Optional;
-
 public class SpringDependencyInjectionAnnotator implements DependencyInjectionAnnotator {
 
     @Override
@@ -181,6 +181,11 @@ public <T extends NodeWithAnnotations<?>> T withFactoryClass(T node) {
         return node;
     }
 
+    @Override
+    public String getTransactionalAnnotation() {
+        return "org.springframework.transaction.annotation.Transactional";
+    }
+
     @Override
     public <T extends NodeWithAnnotations<?>> T withFactoryMethod(T node) {
         node.addAnnotation("org.springframework.context.annotation.Bean");

diff --git a/...ls-codegen-common/src/main/java/org/drools/codegen/common/rest/impl/CDIRestAnnotator.java b/...ls-codegen-common/src/main/java/org/drools/codegen/common/rest/impl/CDIRestAnnotator.java
@@ -29,7 +29,7 @@ public class CDIRestAnnotator implements RestAnnotator {
 
     @Override
     public <T extends NodeWithAnnotations<?>> boolean isRestAnnotated(T node) {
-        return Stream.of("POST", "GET", "PUT", "DELETE")
+        return Stream.of("POST", "GET", "PUT", "DELETE", "PATCH")
                 .map(node::getAnnotationByName)
                 .anyMatch(Optional::isPresent);
     }

diff --git a/...codegen-common/src/main/java/org/drools/codegen/common/rest/impl/SpringRestAnnotator.java b/...codegen-common/src/main/java/org/drools/codegen/common/rest/impl/SpringRestAnnotator.java
@@ -29,7 +29,7 @@ public class SpringRestAnnotator implements RestAnnotator {
 
     @Override
     public <T extends NodeWithAnnotations<?>> boolean isRestAnnotated(T node) {
-        return Stream.of("PostMapping", "GetMapping", "PutMapping", "DeleteMapping")
+        return Stream.of("PostMapping", "GetMapping", "PutMapping", "DeleteMapping", "PatchMapping")
                 .map(node::getAnnotationByName)
                 .anyMatch(Optional::isPresent);
     }