fix: refactor PanacheMongoReposioryEnhancer after #3742

quarkusio · Aug 28, 2019 · c8edaad · c8edaad
1 parent 71658ee
commit c8edaad
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/...t/src/main/java/io/quarkus/mongodb/panache/deployment/PanacheMongoRepositoryEnhancer.java b/...t/src/main/java/io/quarkus/mongodb/panache/deployment/PanacheMongoRepositoryEnhancer.java
@@ -1,6 +1,7 @@
 package io.quarkus.mongodb.panache.deployment;
 
 import org.jboss.jandex.ClassInfo;
+import org.jboss.jandex.DotName;
 import org.jboss.jandex.IndexView;
 import org.objectweb.asm.ClassVisitor;
 import org.objectweb.asm.MethodVisitor;
@@ -11,36 +12,35 @@
 import io.quarkus.panache.common.deployment.PanacheRepositoryEnhancer;
 
 public class PanacheMongoRepositoryEnhancer extends PanacheRepositoryEnhancer {
-    public final static String PANACHE_REPOSITORY_BASE_NAME = PanacheMongoRepositoryBase.class.getName();
-    public final static String PANACHE_REPOSITORY_BASE_BINARY_NAME = PANACHE_REPOSITORY_BASE_NAME.replace('.', '/');
+    public final static DotName PANACHE_REPOSITORY_BASE_NAME = DotName.createSimple(PanacheMongoRepositoryBase.class.getName());
 
-    public final static String PANACHE_REPOSITORY_NAME = PanacheMongoRepository.class.getName();
-    public final static String PANACHE_REPOSITORY_BINARY_NAME = PANACHE_REPOSITORY_NAME.replace('.', '/');
+    public final static DotName PANACHE_REPOSITORY_NAME = DotName.createSimple(PanacheMongoRepository.class.getName());
 
     public PanacheMongoRepositoryEnhancer(IndexView index) {
         super(index, PanacheResourceProcessor.DOTNAME_PANACHE_REPOSITORY_BASE);
     }
 
     @Override
     public ClassVisitor apply(String className, ClassVisitor outputClassVisitor) {
-        return new PanacheMongoRepositoryClassVisitor(className, outputClassVisitor, panacheRepositoryBaseClassInfo);
+        return new PanacheMongoRepositoryClassVisitor(className, outputClassVisitor, panacheRepositoryBaseClassInfo,
+                this.indexView);
     }
 
     static class PanacheMongoRepositoryClassVisitor extends PanacheRepositoryClassVisitor {
 
         public PanacheMongoRepositoryClassVisitor(String className, ClassVisitor outputClassVisitor,
-                ClassInfo panacheRepositoryBaseClassInfo) {
-            super(className, outputClassVisitor, panacheRepositoryBaseClassInfo);
+                ClassInfo panacheRepositoryBaseClassInfo, IndexView indexView) {
+            super(className, outputClassVisitor, panacheRepositoryBaseClassInfo, indexView);
         }
 
         @Override
-        protected String getPanacheRepositoryBinaryName() {
-            return PANACHE_REPOSITORY_BINARY_NAME;
+        protected DotName getPanacheRepositoryDotName() {
+            return PANACHE_REPOSITORY_NAME;
         }
 
         @Override
-        protected String getPanacheRepositoryBaseBinaryName() {
-            return PANACHE_REPOSITORY_BASE_BINARY_NAME;
+        protected DotName getPanacheRepositoryBaseDotName() {
+            return PANACHE_REPOSITORY_BASE_NAME;
         }
 
         @Override