feat: add TraceID and SpanID in request header (#59)

* feat: add TraceID and SpanID in request header

Signed-off-by: Ruihang Xia <waynestxia@gmail.com>

* chore: gen go and java

---------

Signed-off-by: Ruihang Xia <waynestxia@gmail.com>
Co-authored-by: evenyag <realevenyag@gmail.com>

go/greptime/v1/common.pb.go

@@ -85,6 +85,10 @@ type RequestHeader struct {
 	Authorization *AuthHeader `protobuf:"bytes,3,opt,name=authorization,proto3" json:"authorization,omitempty"`
 	// The `dbname` for the request
 	Dbname string `protobuf:"bytes,4,opt,name=dbname,proto3" json:"dbname,omitempty"`
+	// TraceID of request
+	TraceId *uint64 `protobuf:"varint,5,opt,name=trace_id,json=traceId,proto3,oneof" json:"trace_id,omitempty"`
+	// SpanID of request
+	SpanId *uint64 `protobuf:"varint,6,opt,name=span_id,json=spanId,proto3,oneof" json:"span_id,omitempty"`
 }
 
 func (x *RequestHeader) Reset() {
@@ -147,6 +151,20 @@ func (x *RequestHeader) GetDbname() string {
 	return ""
 }
 
+func (x *RequestHeader) GetTraceId() uint64 {
+	if x != nil && x.TraceId != nil {
+		return *x.TraceId
+	}
+	return 0
+}
+
+func (x *RequestHeader) GetSpanId() uint64 {
+	if x != nil && x.SpanId != nil {
+		return *x.SpanId
+	}
+	return 0
+}
+
 type AuthHeader struct {
 	state         protoimpl.MessageState
 	sizeCache     protoimpl.SizeCache
@@ -430,7 +448,7 @@ var file_greptime_v1_common_proto_rawDesc = []byte{
 	0x0a, 0x18, 0x67, 0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x2f, 0x76, 0x31, 0x2f, 0x63, 0x6f,
 	0x6d, 0x6d, 0x6f, 0x6e, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x0b, 0x67, 0x72, 0x65, 0x70,
 	0x74, 0x69, 0x6d, 0x65, 0x2e, 0x76, 0x31, 0x22, 0x10, 0x0a, 0x0e, 0x52, 0x65, 0x73, 0x70, 0x6f,
-	0x6e, 0x73, 0x65, 0x48, 0x65, 0x61, 0x64, 0x65, 0x72, 0x22, 0x98, 0x01, 0x0a, 0x0d, 0x52, 0x65,
+	0x6e, 0x73, 0x65, 0x48, 0x65, 0x61, 0x64, 0x65, 0x72, 0x22, 0xef, 0x01, 0x0a, 0x0d, 0x52, 0x65,
 	0x71, 0x75, 0x65, 0x73, 0x74, 0x48, 0x65, 0x61, 0x64, 0x65, 0x72, 0x12, 0x18, 0x0a, 0x07, 0x63,
 	0x61, 0x74, 0x61, 0x6c, 0x6f, 0x67, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x07, 0x63, 0x61,
 	0x74, 0x61, 0x6c, 0x6f, 0x67, 0x12, 0x16, 0x0a, 0x06, 0x73, 0x63, 0x68, 0x65, 0x6d, 0x61, 0x18,
@@ -440,33 +458,39 @@ var file_greptime_v1_common_proto_rawDesc = []byte{
 	0x76, 0x31, 0x2e, 0x41, 0x75, 0x74, 0x68, 0x48, 0x65, 0x61, 0x64, 0x65, 0x72, 0x52, 0x0d, 0x61,
 	0x75, 0x74, 0x68, 0x6f, 0x72, 0x69, 0x7a, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x12, 0x16, 0x0a, 0x06,
 	0x64, 0x62, 0x6e, 0x61, 0x6d, 0x65, 0x18, 0x04, 0x20, 0x01, 0x28, 0x09, 0x52, 0x06, 0x64, 0x62,
-	0x6e, 0x61, 0x6d, 0x65, 0x22, 0x73, 0x0a, 0x0a, 0x41, 0x75, 0x74, 0x68, 0x48, 0x65, 0x61, 0x64,
+	0x6e, 0x61, 0x6d, 0x65, 0x12, 0x1e, 0x0a, 0x08, 0x74, 0x72, 0x61, 0x63, 0x65, 0x5f, 0x69, 0x64,
-	0x65, 0x72, 0x12, 0x2a, 0x0a, 0x05, 0x62, 0x61, 0x73, 0x69, 0x63, 0x18, 0x01, 0x20, 0x01, 0x28,
+	0x18, 0x05, 0x20, 0x01, 0x28, 0x04, 0x48, 0x00, 0x52, 0x07, 0x74, 0x72, 0x61, 0x63, 0x65, 0x49,
-	0x0b, 0x32, 0x12, 0x2e, 0x67, 0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x2e, 0x76, 0x31, 0x2e,
+	0x64, 0x88, 0x01, 0x01, 0x12, 0x1c, 0x0a, 0x07, 0x73, 0x70, 0x61, 0x6e, 0x5f, 0x69, 0x64, 0x18,
-	0x42, 0x61, 0x73, 0x69, 0x63, 0x48, 0x00, 0x52, 0x05, 0x62, 0x61, 0x73, 0x69, 0x63, 0x12, 0x2a,
+	0x06, 0x20, 0x01, 0x28, 0x04, 0x48, 0x01, 0x52, 0x06, 0x73, 0x70, 0x61, 0x6e, 0x49, 0x64, 0x88,
-	0x0a, 0x05, 0x74, 0x6f, 0x6b, 0x65, 0x6e, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x12, 0x2e,
+	0x01, 0x01, 0x42, 0x0b, 0x0a, 0x09, 0x5f, 0x74, 0x72, 0x61, 0x63, 0x65, 0x5f, 0x69, 0x64, 0x42,
-	0x67, 0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x2e, 0x76, 0x31, 0x2e, 0x54, 0x6f, 0x6b, 0x65,
+	0x0a, 0x0a, 0x08, 0x5f, 0x73, 0x70, 0x61, 0x6e, 0x5f, 0x69, 0x64, 0x22, 0x73, 0x0a, 0x0a, 0x41,
-	0x6e, 0x48, 0x00, 0x52, 0x05, 0x74, 0x6f, 0x6b, 0x65, 0x6e, 0x42, 0x0d, 0x0a, 0x0b, 0x61, 0x75,
+	0x75, 0x74, 0x68, 0x48, 0x65, 0x61, 0x64, 0x65, 0x72, 0x12, 0x2a, 0x0a, 0x05, 0x62, 0x61, 0x73,
-	0x74, 0x68, 0x5f, 0x73, 0x63, 0x68, 0x65, 0x6d, 0x65, 0x22, 0x3f, 0x0a, 0x05, 0x42, 0x61, 0x73,
+	0x69, 0x63, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x12, 0x2e, 0x67, 0x72, 0x65, 0x70, 0x74,
-	0x69, 0x63, 0x12, 0x1a, 0x0a, 0x08, 0x75, 0x73, 0x65, 0x72, 0x6e, 0x61, 0x6d, 0x65, 0x18, 0x01,
+	0x69, 0x6d, 0x65, 0x2e, 0x76, 0x31, 0x2e, 0x42, 0x61, 0x73, 0x69, 0x63, 0x48, 0x00, 0x52, 0x05,
-	0x20, 0x01, 0x28, 0x09, 0x52, 0x08, 0x75, 0x73, 0x65, 0x72, 0x6e, 0x61, 0x6d, 0x65, 0x12, 0x1a,
+	0x62, 0x61, 0x73, 0x69, 0x63, 0x12, 0x2a, 0x0a, 0x05, 0x74, 0x6f, 0x6b, 0x65, 0x6e, 0x18, 0x02,
-	0x0a, 0x08, 0x70, 0x61, 0x73, 0x73, 0x77, 0x6f, 0x72, 0x64, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09,
+	0x20, 0x01, 0x28, 0x0b, 0x32, 0x12, 0x2e, 0x67, 0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x2e,
-	0x52, 0x08, 0x70, 0x61, 0x73, 0x73, 0x77, 0x6f, 0x72, 0x64, 0x22, 0x1d, 0x0a, 0x05, 0x54, 0x6f,
+	0x76, 0x31, 0x2e, 0x54, 0x6f, 0x6b, 0x65, 0x6e, 0x48, 0x00, 0x52, 0x05, 0x74, 0x6f, 0x6b, 0x65,
-	0x6b, 0x65, 0x6e, 0x12, 0x14, 0x0a, 0x05, 0x74, 0x6f, 0x6b, 0x65, 0x6e, 0x18, 0x01, 0x20, 0x01,
+	0x6e, 0x42, 0x0d, 0x0a, 0x0b, 0x61, 0x75, 0x74, 0x68, 0x5f, 0x73, 0x63, 0x68, 0x65, 0x6d, 0x65,
-	0x28, 0x09, 0x52, 0x05, 0x74, 0x6f, 0x6b, 0x65, 0x6e, 0x22, 0x24, 0x0a, 0x0c, 0x41, 0x66, 0x66,
+	0x22, 0x3f, 0x0a, 0x05, 0x42, 0x61, 0x73, 0x69, 0x63, 0x12, 0x1a, 0x0a, 0x08, 0x75, 0x73, 0x65,
-	0x65, 0x63, 0x74, 0x65, 0x64, 0x52, 0x6f, 0x77, 0x73, 0x12, 0x14, 0x0a, 0x05, 0x76, 0x61, 0x6c,
+	0x72, 0x6e, 0x61, 0x6d, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x08, 0x75, 0x73, 0x65,
-	0x75, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0d, 0x52, 0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x22,
+	0x72, 0x6e, 0x61, 0x6d, 0x65, 0x12, 0x1a, 0x0a, 0x08, 0x70, 0x61, 0x73, 0x73, 0x77, 0x6f, 0x72,
-	0x50, 0x0a, 0x0e, 0x46, 0x6c, 0x69, 0x67, 0x68, 0x74, 0x4d, 0x65, 0x74, 0x61, 0x64, 0x61, 0x74,
+	0x64, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x08, 0x70, 0x61, 0x73, 0x73, 0x77, 0x6f, 0x72,
-	0x61, 0x12, 0x3e, 0x0a, 0x0d, 0x61, 0x66, 0x66, 0x65, 0x63, 0x74, 0x65, 0x64, 0x5f, 0x72, 0x6f,
+	0x64, 0x22, 0x1d, 0x0a, 0x05, 0x54, 0x6f, 0x6b, 0x65, 0x6e, 0x12, 0x14, 0x0a, 0x05, 0x74, 0x6f,
-	0x77, 0x73, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x19, 0x2e, 0x67, 0x72, 0x65, 0x70, 0x74,
+	0x6b, 0x65, 0x6e, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x05, 0x74, 0x6f, 0x6b, 0x65, 0x6e,
-	0x69, 0x6d, 0x65, 0x2e, 0x76, 0x31, 0x2e, 0x41, 0x66, 0x66, 0x65, 0x63, 0x74, 0x65, 0x64, 0x52,
+	0x22, 0x24, 0x0a, 0x0c, 0x41, 0x66, 0x66, 0x65, 0x63, 0x74, 0x65, 0x64, 0x52, 0x6f, 0x77, 0x73,
-	0x6f, 0x77, 0x73, 0x52, 0x0c, 0x61, 0x66, 0x66, 0x65, 0x63, 0x74, 0x65, 0x64, 0x52, 0x6f, 0x77,
+	0x12, 0x14, 0x0a, 0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0d, 0x52,
-	0x73, 0x42, 0x4f, 0x0a, 0x0e, 0x69, 0x6f, 0x2e, 0x67, 0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65,
+	0x05, 0x76, 0x61, 0x6c, 0x75, 0x65, 0x22, 0x50, 0x0a, 0x0e, 0x46, 0x6c, 0x69, 0x67, 0x68, 0x74,
-	0x2e, 0x76, 0x31, 0x42, 0x06, 0x43, 0x6f, 0x6d, 0x6d, 0x6f, 0x6e, 0x5a, 0x35, 0x67, 0x69, 0x74,
+	0x4d, 0x65, 0x74, 0x61, 0x64, 0x61, 0x74, 0x61, 0x12, 0x3e, 0x0a, 0x0d, 0x61, 0x66, 0x66, 0x65,
-	0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x47, 0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65,
+	0x63, 0x74, 0x65, 0x64, 0x5f, 0x72, 0x6f, 0x77, 0x73, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, 0x32,
-	0x54, 0x65, 0x61, 0x6d, 0x2f, 0x67, 0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x2d, 0x70, 0x72,
+	0x19, 0x2e, 0x67, 0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x2e, 0x76, 0x31, 0x2e, 0x41, 0x66,
-	0x6f, 0x74, 0x6f, 0x2f, 0x67, 0x6f, 0x2f, 0x67, 0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x2f,
+	0x66, 0x65, 0x63, 0x74, 0x65, 0x64, 0x52, 0x6f, 0x77, 0x73, 0x52, 0x0c, 0x61, 0x66, 0x66, 0x65,
-	0x76, 0x31, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
+	0x63, 0x74, 0x65, 0x64, 0x52, 0x6f, 0x77, 0x73, 0x42, 0x4f, 0x0a, 0x0e, 0x69, 0x6f, 0x2e, 0x67,
+	0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x2e, 0x76, 0x31, 0x42, 0x06, 0x43, 0x6f, 0x6d, 0x6d,
+	0x6f, 0x6e, 0x5a, 0x35, 0x67, 0x69, 0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x47,
+	0x72, 0x65, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x54, 0x65, 0x61, 0x6d, 0x2f, 0x67, 0x72, 0x65, 0x70,
+	0x74, 0x69, 0x6d, 0x65, 0x2d, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x2f, 0x67, 0x6f, 0x2f, 0x67, 0x72,
+	0x65, 0x70, 0x74, 0x69, 0x6d, 0x65, 0x2f, 0x76, 0x31, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f,
+	0x33,
 }
 
 var (
@@ -594,6 +618,7 @@ func file_greptime_v1_common_proto_init() {
 			}
 		}
 	}
+	file_greptime_v1_common_proto_msgTypes[1].OneofWrappers = []interface{}{}
 	file_greptime_v1_common_proto_msgTypes[2].OneofWrappers = []interface{}{
 		(*AuthHeader_Basic)(nil),
 		(*AuthHeader_Token)(nil),

java/src/main/java/io/greptime/v1/Common.java

@@ -524,6 +524,44 @@ public final class Common {
      */
     com.google.protobuf.ByteString
         getDbnameBytes();
+
+    /**
+     * <pre>
+     * TraceID of request
+     * </pre>
+     *
+     * <code>optional uint64 trace_id = 5;</code>
+     * @return Whether the traceId field is set.
+     */
+    boolean hasTraceId();
+    /**
+     * <pre>
+     * TraceID of request
+     * </pre>
+     *
+     * <code>optional uint64 trace_id = 5;</code>
+     * @return The traceId.
+     */
+    long getTraceId();
+
+    /**
+     * <pre>
+     * SpanID of request
+     * </pre>
+     *
+     * <code>optional uint64 span_id = 6;</code>
+     * @return Whether the spanId field is set.
+     */
+    boolean hasSpanId();
+    /**
+     * <pre>
+     * SpanID of request
+     * </pre>
+     *
+     * <code>optional uint64 span_id = 6;</code>
+     * @return The spanId.
+     */
+    long getSpanId();
   }
   /**
    * Protobuf type {@code greptime.v1.RequestHeader}
@@ -563,6 +601,7 @@ public final class Common {
       if (extensionRegistry == null) {
         throw new java.lang.NullPointerException();
       }
+      int mutable_bitField0_ = 0;
       com.google.protobuf.UnknownFieldSet.Builder unknownFields =
           com.google.protobuf.UnknownFieldSet.newBuilder();
       try {
@@ -604,6 +643,16 @@ public final class Common {
               dbname_ = s;
               break;
             }
+            case 40: {
+              bitField0_ |= 0x00000001;
+              traceId_ = input.readUInt64();
+              break;
+            }
+            case 48: {
+              bitField0_ |= 0x00000002;
+              spanId_ = input.readUInt64();
+              break;
+            }
             default: {
               if (!parseUnknownField(
                   input, unknownFields, extensionRegistry, tag)) {
@@ -638,6 +687,7 @@ public final class Common {
               io.greptime.v1.Common.RequestHeader.class, io.greptime.v1.Common.RequestHeader.Builder.class);
     }
 
+    private int bitField0_;
     public static final int CATALOG_FIELD_NUMBER = 1;
     private volatile java.lang.Object catalog_;
     /**
@@ -814,6 +864,60 @@ public final class Common {
       }
     }
 
+    public static final int TRACE_ID_FIELD_NUMBER = 5;
+    private long traceId_;
+    /**
+     * <pre>
+     * TraceID of request
+     * </pre>
+     *
+     * <code>optional uint64 trace_id = 5;</code>
+     * @return Whether the traceId field is set.
+     */
+    @java.lang.Override
+    public boolean hasTraceId() {
+      return ((bitField0_ & 0x00000001) != 0);
+    }
+    /**
+     * <pre>
+     * TraceID of request
+     * </pre>
+     *
+     * <code>optional uint64 trace_id = 5;</code>
+     * @return The traceId.
+     */
+    @java.lang.Override
+    public long getTraceId() {
+      return traceId_;
+    }
+
+    public static final int SPAN_ID_FIELD_NUMBER = 6;
+    private long spanId_;
+    /**
+     * <pre>
+     * SpanID of request
+     * </pre>
+     *
+     * <code>optional uint64 span_id = 6;</code>
+     * @return Whether the spanId field is set.
+     */
+    @java.lang.Override
+    public boolean hasSpanId() {
+      return ((bitField0_ & 0x00000002) != 0);
+    }
+    /**
+     * <pre>
+     * SpanID of request
+     * </pre>
+     *
+     * <code>optional uint64 span_id = 6;</code>
+     * @return The spanId.
+     */
+    @java.lang.Override
+    public long getSpanId() {
+      return spanId_;
+    }
+
     private byte memoizedIsInitialized = -1;
     @java.lang.Override
     public final boolean isInitialized() {
@@ -840,6 +944,12 @@ public final class Common {
       if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(dbname_)) {
         com.google.protobuf.GeneratedMessageV3.writeString(output, 4, dbname_);
       }
+      if (((bitField0_ & 0x00000001) != 0)) {
+        output.writeUInt64(5, traceId_);
+      }
+      if (((bitField0_ & 0x00000002) != 0)) {
+        output.writeUInt64(6, spanId_);
+      }
       unknownFields.writeTo(output);
     }
 
@@ -862,6 +972,14 @@ public final class Common {
       if (!com.google.protobuf.GeneratedMessageV3.isStringEmpty(dbname_)) {
         size += com.google.protobuf.GeneratedMessageV3.computeStringSize(4, dbname_);
       }
+      if (((bitField0_ & 0x00000001) != 0)) {
+        size += com.google.protobuf.CodedOutputStream
+          .computeUInt64Size(5, traceId_);
+      }
+      if (((bitField0_ & 0x00000002) != 0)) {
+        size += com.google.protobuf.CodedOutputStream
+          .computeUInt64Size(6, spanId_);
+      }
       size += unknownFields.getSerializedSize();
       memoizedSize = size;
       return size;
@@ -888,6 +1006,16 @@ public final class Common {
       }
       if (!getDbname()
           .equals(other.getDbname())) return false;
+      if (hasTraceId() != other.hasTraceId()) return false;
+      if (hasTraceId()) {
+        if (getTraceId()
+            != other.getTraceId()) return false;
+      }
+      if (hasSpanId() != other.hasSpanId()) return false;
+      if (hasSpanId()) {
+        if (getSpanId()
+            != other.getSpanId()) return false;
+      }
       if (!unknownFields.equals(other.unknownFields)) return false;
       return true;
     }
@@ -909,6 +1037,16 @@ public final class Common {
       }
       hash = (37 * hash) + DBNAME_FIELD_NUMBER;
       hash = (53 * hash) + getDbname().hashCode();
+      if (hasTraceId()) {
+        hash = (37 * hash) + TRACE_ID_FIELD_NUMBER;
+        hash = (53 * hash) + com.google.protobuf.Internal.hashLong(
+            getTraceId());
+      }
+      if (hasSpanId()) {
+        hash = (37 * hash) + SPAN_ID_FIELD_NUMBER;
+        hash = (53 * hash) + com.google.protobuf.Internal.hashLong(
+            getSpanId());
+      }
       hash = (29 * hash) + unknownFields.hashCode();
       memoizedHashCode = hash;
       return hash;
@@ -1054,6 +1192,10 @@ public final class Common {
         }
         dbname_ = "";
 
+        traceId_ = 0L;
+        bitField0_ = (bitField0_ & ~0x00000001);
+        spanId_ = 0L;
+        bitField0_ = (bitField0_ & ~0x00000002);
         return this;
       }
 
@@ -1080,6 +1222,8 @@ public final class Common {
       @java.lang.Override
       public io.greptime.v1.Common.RequestHeader buildPartial() {
         io.greptime.v1.Common.RequestHeader result = new io.greptime.v1.Common.RequestHeader(this);
+        int from_bitField0_ = bitField0_;
+        int to_bitField0_ = 0;
         result.catalog_ = catalog_;
         result.schema_ = schema_;
         if (authorizationBuilder_ == null) {
@@ -1088,6 +1232,15 @@ public final class Common {
           result.authorization_ = authorizationBuilder_.build();
         }
         result.dbname_ = dbname_;
+        if (((from_bitField0_ & 0x00000001) != 0)) {
+          result.traceId_ = traceId_;
+          to_bitField0_ |= 0x00000001;
+        }
+        if (((from_bitField0_ & 0x00000002) != 0)) {
+          result.spanId_ = spanId_;
+          to_bitField0_ |= 0x00000002;
+        }
+        result.bitField0_ = to_bitField0_;
         onBuilt();
         return result;
       }
@@ -1151,6 +1304,12 @@ public final class Common {
           dbname_ = other.dbname_;
           onChanged();
         }
+        if (other.hasTraceId()) {
+          setTraceId(other.getTraceId());
+        }
+        if (other.hasSpanId()) {
+          setSpanId(other.getSpanId());
+        }
         this.mergeUnknownFields(other.unknownFields);
         onChanged();
         return this;
@@ -1179,6 +1338,7 @@ public final class Common {
         }
         return this;
       }
+      private int bitField0_;
 
       private java.lang.Object catalog_ = "";
       /**
@@ -1622,6 +1782,116 @@ public final class Common {
         onChanged();
         return this;
       }
+
+      private long traceId_ ;
+      /**
+       * <pre>
+       * TraceID of request
+       * </pre>
+       *
+       * <code>optional uint64 trace_id = 5;</code>
+       * @return Whether the traceId field is set.
+       */
+      @java.lang.Override
+      public boolean hasTraceId() {
+        return ((bitField0_ & 0x00000001) != 0);
+      }
+      /**
+       * <pre>
+       * TraceID of request
+       * </pre>
+       *
+       * <code>optional uint64 trace_id = 5;</code>
+       * @return The traceId.
+       */
+      @java.lang.Override
+      public long getTraceId() {
+        return traceId_;
+      }
+      /**
+       * <pre>
+       * TraceID of request
+       * </pre>
+       *
+       * <code>optional uint64 trace_id = 5;</code>
+       * @param value The traceId to set.
+       * @return This builder for chaining.
+       */
+      public Builder setTraceId(long value) {
+        bitField0_ |= 0x00000001;
+        traceId_ = value;
+        onChanged();
+        return this;
+      }
+      /**
+       * <pre>
+       * TraceID of request
+       * </pre>
+       *
+       * <code>optional uint64 trace_id = 5;</code>
+       * @return This builder for chaining.
+       */
+      public Builder clearTraceId() {
+        bitField0_ = (bitField0_ & ~0x00000001);
+        traceId_ = 0L;
+        onChanged();
+        return this;
+      }
+
+      private long spanId_ ;
+      /**
+       * <pre>
+       * SpanID of request
+       * </pre>
+       *
+       * <code>optional uint64 span_id = 6;</code>
+       * @return Whether the spanId field is set.
+       */
+      @java.lang.Override
+      public boolean hasSpanId() {
+        return ((bitField0_ & 0x00000002) != 0);
+      }
+      /**
+       * <pre>
+       * SpanID of request
+       * </pre>
+       *
+       * <code>optional uint64 span_id = 6;</code>
+       * @return The spanId.
+       */
+      @java.lang.Override
+      public long getSpanId() {
+        return spanId_;
+      }
+      /**
+       * <pre>
+       * SpanID of request
+       * </pre>
+       *
+       * <code>optional uint64 span_id = 6;</code>
+       * @param value The spanId to set.
+       * @return This builder for chaining.
+       */
+      public Builder setSpanId(long value) {
+        bitField0_ |= 0x00000002;
+        spanId_ = value;
+        onChanged();
+        return this;
+      }
+      /**
+       * <pre>
+       * SpanID of request
+       * </pre>
+       *
+       * <code>optional uint64 span_id = 6;</code>
+       * @return This builder for chaining.
+       */
+      public Builder clearSpanId() {
+        bitField0_ = (bitField0_ & ~0x00000002);
+        spanId_ = 0L;
+        onChanged();
+        return this;
+      }
       @java.lang.Override
       public final Builder setUnknownFields(
           final com.google.protobuf.UnknownFieldSet unknownFields) {
@@ -5074,19 +5344,21 @@ public final class Common {
   static {
     java.lang.String[] descriptorData = {
       "\n\030greptime/v1/common.proto\022\013greptime.v1\"" +
-      "\020\n\016ResponseHeader\"p\n\rRequestHeader\022\017\n\007ca" +
+      "\020\n\016ResponseHeader\"\266\001\n\rRequestHeader\022\017\n\007c" +
-      "talog\030\001 \001(\t\022\016\n\006schema\030\002 \001(\t\022.\n\rauthoriza" +
+      "atalog\030\001 \001(\t\022\016\n\006schema\030\002 \001(\t\022.\n\rauthoriz" +
-      "tion\030\003 \001(\0132\027.greptime.v1.AuthHeader\022\016\n\006d" +
+      "ation\030\003 \001(\0132\027.greptime.v1.AuthHeader\022\016\n\006" +
-      "bname\030\004 \001(\t\"e\n\nAuthHeader\022#\n\005basic\030\001 \001(\013" +
+      "dbname\030\004 \001(\t\022\025\n\010trace_id\030\005 \001(\004H\000\210\001\001\022\024\n\007s" +
-      "2\022.greptime.v1.BasicH\000\022#\n\005token\030\002 \001(\0132\022." +
+      "pan_id\030\006 \001(\004H\001\210\001\001B\013\n\t_trace_idB\n\n\010_span_" +
-      "greptime.v1.TokenH\000B\r\n\013auth_scheme\"+\n\005Ba" +
+      "id\"e\n\nAuthHeader\022#\n\005basic\030\001 \001(\0132\022.grepti" +
-      "sic\022\020\n\010username\030\001 \001(\t\022\020\n\010password\030\002 \001(\t\"" +
+      "me.v1.BasicH\000\022#\n\005token\030\002 \001(\0132\022.greptime." +
-      "\026\n\005Token\022\r\n\005token\030\001 \001(\t\"\035\n\014AffectedRows\022" +
+      "v1.TokenH\000B\r\n\013auth_scheme\"+\n\005Basic\022\020\n\010us" +
-      "\r\n\005value\030\001 \001(\r\"B\n\016FlightMetadata\0220\n\raffe" +
+      "ername\030\001 \001(\t\022\020\n\010password\030\002 \001(\t\"\026\n\005Token\022" +
-      "cted_rows\030\001 \001(\0132\031.greptime.v1.AffectedRo" +
+      "\r\n\005token\030\001 \001(\t\"\035\n\014AffectedRows\022\r\n\005value\030" +
-      "wsBO\n\016io.greptime.v1B\006CommonZ5github.com" +
+      "\001 \001(\r\"B\n\016FlightMetadata\0220\n\raffected_rows" +
-      "/GreptimeTeam/greptime-proto/go/greptime" +
+      "\030\001 \001(\0132\031.greptime.v1.AffectedRowsBO\n\016io." +
-      "/v1b\006proto3"
+      "greptime.v1B\006CommonZ5github.com/Greptime" +
+      "Team/greptime-proto/go/greptime/v1b\006prot" +
+      "o3"
     };
     descriptor = com.google.protobuf.Descriptors.FileDescriptor
       .internalBuildGeneratedFileFrom(descriptorData,
@@ -5103,7 +5375,7 @@ public final class Common {
     internal_static_greptime_v1_RequestHeader_fieldAccessorTable = new
       com.google.protobuf.GeneratedMessageV3.FieldAccessorTable(
         internal_static_greptime_v1_RequestHeader_descriptor,
-        new java.lang.String[] { "Catalog", "Schema", "Authorization", "Dbname", });
+        new java.lang.String[] { "Catalog", "Schema", "Authorization", "Dbname", "TraceId", "SpanId", "TraceId", "SpanId", });
     internal_static_greptime_v1_AuthHeader_descriptor =
       getDescriptor().getMessageTypes().get(2);
     internal_static_greptime_v1_AuthHeader_fieldAccessorTable = new

proto/greptime/v1/common.proto

@@ -31,6 +31,10 @@ message RequestHeader {
   AuthHeader authorization = 3;
   // The `dbname` for the request
   string dbname = 4;
+  // TraceID of request
+  optional uint64 trace_id = 5;
+  // SpanID of request
+  optional uint64 span_id = 6;
 }
 
 message AuthHeader {