Add optional responseData to TSStatus

iotdb-client/service-rpc/src/main/java/org/apache/iotdb/rpc/RpcUtils.java

@@ -30,6 +30,7 @@
 import org.slf4j.LoggerFactory;
 
 import java.lang.reflect.Proxy;
+import java.nio.ByteBuffer;
 import java.sql.Timestamp;
 import java.text.SimpleDateFormat;
 import java.time.Instant;
@@ -221,6 +222,38 @@ public static TSStatus getStatus(int code, String message) {
     return status;
   }
 
+  /**
+   * Build a {@link TSStatus} with code, optional message and optional binary payload (Thrift {@code
+   * binary} maps to {@link ByteBuffer}).
+   *
+   * @param tsStatusCode status code
+   * @param message optional message; if null, message field is left unset
+   * @param responseData optional serialized payload; if null, responseData field is left unset
+   */
+  public static TSStatus getStatus(
+      final TSStatusCode tsStatusCode, final String message, final ByteBuffer responseData) {
+    return getStatus(tsStatusCode.getStatusCode(), message, responseData);
+  }
+
+  /**
+   * Build a {@link TSStatus} with code, optional message and optional binary payload.
+   *
+   * @param code status code
+   * @param message optional message; if null, message field is left unset
+   * @param responseData optional serialized payload; if null, responseData field is left unset
+   */
+  public static TSStatus getStatus(
+      final int code, final String message, final ByteBuffer responseData) {
+    final TSStatus status = new TSStatus(code);
+    if (message != null) {
+      status.setMessage(message);
+    }
+    if (responseData != null) {
+      status.setResponseData(responseData);
+    }
+    return status;
+  }
+
   public static TSExecuteStatementResp getTSExecuteStatementResp(TSStatusCode tsStatusCode) {
     TSStatus status = getStatus(tsStatusCode);
     return getTSExecuteStatementResp(status);

iotdb-protocol/thrift-commons/src/main/thrift/common.thrift

@@ -34,6 +34,7 @@ struct TSStatus {
   3: optional list<TSStatus> subStatus
   4: optional TEndPoint redirectNode
   5: optional bool needRetry
+  6: optional binary responseData
 }
 
 enum TConsensusGroupType {