feat: Détection des annotations ManyToOne, OneToMany et ManyToMany da…

…ns l'entité liée au Repository
green-code-initiative · Apr 6, 2023 · 85c56a9 · 85c56a9
1 parent 3472481
commit 85c56a9
Show file tree

Hide file tree

Showing 3 changed files with 68 additions and 2 deletions.
diff --git a/...ugin/src/main/java/fr/greencodeinitiative/java/checks/AvoidNPlusOneQueryProblemCheck.java b/...ugin/src/main/java/fr/greencodeinitiative/java/checks/AvoidNPlusOneQueryProblemCheck.java
@@ -2,10 +2,15 @@
 
 import java.util.Arrays;
 import java.util.List;
+import java.util.Optional;
+import java.util.Optional;
+
 
 import org.sonar.check.Priority;
 import org.sonar.check.Rule;
 import org.sonar.plugins.java.api.IssuableSubscriptionVisitor;
+import org.sonar.plugins.java.api.semantic.Type;
+import org.sonar.plugins.java.api.semantic.Type;
 import org.sonar.plugins.java.api.tree.*;
 
 @Rule(key = "EC206",
@@ -23,6 +28,12 @@ public class AvoidNPlusOneQueryProblemCheck extends IssuableSubscriptionVisitor
     private static final String LEFT_JOIN = "LEFT JOIN";
     private static final String VALUE = "value";
 
+    private static final List<String> SPRING_PROBLEMATIC_ANNOTATIONS = List.of(
+            "javax.persistence.OneToMany",
+            "javax.persistence.ManyToOne",
+            "javax.persistence.ManyToMany"
+    );
+
     private final AvoidNPlusOneQueryProblemCheckVisitor visitorInFile = new AvoidNPlusOneQueryProblemCheckVisitor();
 
     @Override
@@ -32,11 +43,39 @@ public List<Tree.Kind> nodesToVisit() {
 
     @Override
     public void visitNode(Tree tree) {
-        if (((ClassTree) tree).symbol().type().isSubtypeOf(SPRING_REPOSITORY)) {
+        ClassTree classTree = (ClassTree) tree;
+
+        if (isSpringRepository(classTree) && hasManyToOneAnnotations(classTree)) {
             tree.accept(visitorInFile);
         }
     }
 
+    private boolean hasManyToOneAnnotations(ClassTree classTree) {
+        Optional<Type> crudRepositoryInterface = classTree.symbol().interfaces().stream()
+                .filter(t -> t.isSubtypeOf(SPRING_REPOSITORY))
+                .findFirst();
+
+        return crudRepositoryInterface.map(type -> type
+                        .typeArguments()
+                        .get(0)
+                        .symbol()
+                        .declaration()
+                        .members()
+                        .stream()
+                        .filter(t -> t.is(Tree.Kind.VARIABLE))
+                        .anyMatch(t -> ((VariableTree) t).modifiers()
+                                .annotations()
+                                .stream()
+                                .anyMatch(a ->
+                                        SPRING_PROBLEMATIC_ANNOTATIONS.stream().anyMatch(a.symbolType()::isSubtypeOf)
+                                )))
+                .orElse(false);
+    }
+
+    private static boolean isSpringRepository(ClassTree classTree) {
+        return classTree.symbol().type().isSubtypeOf(SPRING_REPOSITORY);
+    }
+
     private class AvoidNPlusOneQueryProblemCheckVisitor extends BaseTreeVisitor {
 
         @Override

diff --git a/java-plugin/src/test/files/AvoidNPlusOneQueryProblemBad.java b/java-plugin/src/test/files/AvoidNPlusOneQueryProblemBad.java
@@ -1,5 +1,7 @@
 import org.springframework.data.repository.CrudRepository;
 import java.util.List;
+import javax.persistence.Entity;
+import javax.persistence.OneToMany;
 
 public interface UserRepository extends CrudRepository<User, Long> {
 
@@ -12,6 +14,31 @@ public interface UserRepository extends CrudRepository<User, Long> {
     List<User> findAllWithRoles();
 }
 
+@Entity
 public class User {
 
+    @OneToMany
+    private List<Role> roles;
+
+    public List<Role> getRoles() {
+        return roles;
+    }
+
+    public void setRoles(List<Role> roles) {
+        this.roles = roles;
+    }
+}
+
+@Entity
+public class Role {
+
+    private String name;
+
+    public String getName() {
+        return name;
+    }
+
+    public void setName(String name) {
+        this.name = name;
+    }
 }
diff --git a/...ecks/AvoidOnePlusOneQueryProblemTest.java → ...checks/AvoidNPlusOneQueryProblemTest.java b/...ecks/AvoidOnePlusOneQueryProblemTest.java → ...checks/AvoidNPlusOneQueryProblemTest.java
@@ -4,7 +4,7 @@
 import org.junit.jupiter.api.Test;
 import org.sonar.java.checks.verifier.CheckVerifier;
 
-class AvoidOnePlusOneQueryProblemTest {
+class AvoidNPlusOneQueryProblemTest {
 
     /**
      * @formatter:off