From 4fff1731313d36dd9ee250b15e5192a0fd112dbf Mon Sep 17 00:00:00 2001
From: Tao Wang <18234030530@163.com>
Date: Mon, 19 Aug 2024 17:05:02 +0800
Subject: [PATCH] [feat][sdk] Add document to the grpc generation.

---
 .../java/io/dingodb/grpc/RpcMethodAnnotationProcessor.java     | 3 +++
 1 file changed, 3 insertions(+)

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
index bbe5a0181..eada94a19 100644
--- 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) {