[feat][sdk] Add document to the grpc generation.

wt0530 · Aug 19, 2024 · 6acd090 · 6acd090
1 parent d13bc8d
commit 6acd090
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/java/dingo-grpc-tools/src/main/java/io/dingodb/grpc/RpcMethodAnnotationProcessor.java b/java/dingo-grpc-tools/src/main/java/io/dingodb/grpc/RpcMethodAnnotationProcessor.java
@@ -156,6 +156,9 @@ public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment
                 if (elements.getPackageOf(reqTypeElement).getSimpleName().toString().equals("index")) {
                     messageGenerateProcessor.messages.get(reqTypeName).addSuperinterface(Constant.STORE_REQ);
                 }
+                if (elements.getPackageOf(reqTypeElement).getSimpleName().toString().equals("document")) {
+                    messageGenerateProcessor.messages.get(reqTypeName).addSuperinterface(Constant.STORE_REQ);
+                }
                 serviceElements.add((TypeElement) element.getEnclosingElement());
                 entityElements.add((TypeElement) reqTypeElement.getEnclosingElement());
             } catch (Exception e) {