Fix quarkusio#5274 for Mongo too

FroMage · Nov 7, 2019 · 6cbdeeb · 6cbdeeb
1 parent bec228a
commit 6cbdeeb
Showing 1 changed file with 20 additions and 17 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
@@ -50,23 +50,26 @@ protected String getPanacheOperationsBinaryName() {
 
         @Override
         public void visitEnd() {
-            // Bridge for findById
-            MethodVisitor mv = super.visitMethod(Opcodes.ACC_PUBLIC | Opcodes.ACC_SYNTHETIC | Opcodes.ACC_BRIDGE,
-                    "findById",
-                    "(Ljava/lang/Object;)Ljava/lang/Object;",
-                    null,
-                    null);
-            mv.visitParameter("id", 0);
-            mv.visitCode();
-            mv.visitIntInsn(Opcodes.ALOAD, 0);
-            mv.visitIntInsn(Opcodes.ALOAD, 1);
-            mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL,
-                    daoBinaryName,
-                    "findById",
-                    "(Ljava/lang/Object;)" + entitySignature, false);
-            mv.visitInsn(Opcodes.ARETURN);
-            mv.visitMaxs(0, 0);
-            mv.visitEnd();
+            // Bridge for findById, but only if we actually know the end entity (which we don't for intermediate
+            // abstract repositories that haven't fixed their entity type yet
+            if (!entitySignature.equals("Ljava/lang/Object;")) {
+                MethodVisitor mv = super.visitMethod(Opcodes.ACC_PUBLIC | Opcodes.ACC_SYNTHETIC | Opcodes.ACC_BRIDGE,
+                        "findById",
+                        "(Ljava/lang/Object;)Ljava/lang/Object;",
+                        null,
+                        null);
+                mv.visitParameter("id", 0);
+                mv.visitCode();
+                mv.visitIntInsn(Opcodes.ALOAD, 0);
+                mv.visitIntInsn(Opcodes.ALOAD, 1);
+                mv.visitMethodInsn(Opcodes.INVOKEVIRTUAL,
+                        daoBinaryName,
+                        "findById",
+                        "(Ljava/lang/Object;)" + entitySignature, false);
+                mv.visitInsn(Opcodes.ARETURN);
+                mv.visitMaxs(0, 0);
+                mv.visitEnd();
+            }
 
             super.visitEnd();
         }