feat: add QueryContext to RegionRequestHeader

GreptimeTeam · Apr 27, 2024 · 2426058 · 2426058
1 parent 2c14c6e
commit 2426058
Show file tree

Hide file tree

Showing 9 changed files with 11,009 additions and 4,253 deletions.
diff --git a/c++/greptime/v1/common.pb.cc b/c++/greptime/v1/common.pb.cc
diff --git a/c++/greptime/v1/common.pb.h b/c++/greptime/v1/common.pb.h
diff --git a/c++/greptime/v1/region/server.pb.cc b/c++/greptime/v1/region/server.pb.cc
diff --git a/c++/greptime/v1/region/server.pb.h b/c++/greptime/v1/region/server.pb.h
diff --git a/go/greptime/v1/common.pb.go b/go/greptime/v1/common.pb.go
diff --git a/java/src/main/java/io/greptime/v1/Common.java b/java/src/main/java/io/greptime/v1/Common.java
diff --git a/java/src/main/java/io/greptime/v1/region/Server.java b/java/src/main/java/io/greptime/v1/region/Server.java
diff --git a/proto/greptime/v1/common.proto b/proto/greptime/v1/common.proto
@@ -113,6 +113,40 @@ enum ColumnDataType {
   DECIMAL128 = 30;
 }
 
+enum SqlDialect {
+  MYSQL = 0;
+  POSTGRES = 1;
+  GREPTIMEDB = 2;
+}
+
+enum PostgresByteaOutputValue {
+  HEX = 0;
+  ESCAPE = 1;
+}
+
+enum PostgresDateTimeStyle {
+  ISO = 0;
+  SQL = 1;
+  Postgres = 2;
+  German = 3;
+}
+
+enum PostgresDateOrder {
+  MDY = 0;
+  DMY = 1;
+  YMD = 2;
+}
+
+message PostgresDateStyleFormat {
+  PostgresDateTimeStyle date_style = 1;
+  PostgresDateOrder date_order = 2;
+}
+
+message ConfigurationVariables {
+  PostgresByteaOutputValue bytea_output = 1; 
+  PostgresDateStyleFormat datestyle_format = 2;
+}
+
 message IntervalMonthDayNano {
   int32 months = 1;
   int32 days = 2;

diff --git a/proto/greptime/v1/region/server.proto b/proto/greptime/v1/region/server.proto
@@ -26,6 +26,16 @@ import "greptime/v1/ddl.proto";
 
 service Region { rpc Handle(RegionRequest) returns (RegionResponse); }
 
+message QueryContext {
+  string current_catalog = 1;
+  string current_schema = 2;
+  string current_user = 3;
+  string timezone = 4;
+  map<string, string> extension = 5;
+  SqlDialect sql_dialect = 6;
+  ConfigurationVariables configuration_parameter = 7;
+}
+
 message RegionRequestHeader {
   // Encoded trace_id & span_id, follow the w3c Trace Context
   // https://www.w3.org/TR/trace-context/#header-name
@@ -34,6 +44,8 @@ message RegionRequestHeader {
   string dbname = 3;
   // The `timezone` for the request
   string timezone = 4;
+  // The contextual information of the query
+  QueryContext query_context = 6;
 }
 
 message RegionRequest {