return the native sql in the lables

Author: LinuxSuRen

e2e/compose.yaml

@@ -82,5 +82,7 @@ services:
       interval: 3s
       timeout: 30s
       retries: 10
+    ports:
+      - 5432
 volumes:
   cache:

pkg/data_query.go

@@ -66,6 +66,10 @@ func (s *dbserver) Query(ctx context.Context, query *server.DataQuery) (result *
 
 	query.Sql = dbQuery.GetInnerSQL().ToNativeSQL(query.Sql)
 	result.Meta.Labels = dbQuery.GetLabels(ctx, query.Sql)
+	result.Meta.Labels = append(result.Meta.Labels, &server.Pair{
+		Key:   "_native_sql",
+		Value: query.Sql,
+	})
 
 	var dataResult *server.DataQueryResult
 	now := time.Now()

pkg/inner_sql.go

@@ -23,10 +23,11 @@ type InnerSQL interface {
 }
 
 const (
-	InnerSelectTable_  = "@selectTable_"
-	InnerShowDatabases = "@showDatabases"
-	InnerShowTables    = "@showTables"
-	InnerCurrentDB     = "@currentDB"
+	InnerSelectTable_      = "@selectTable_"
+	InnerSelectTableLimit_ = "@selectTableLImit100_"
+	InnerShowDatabases     = "@showDatabases"
+	InnerShowTables        = "@showTables"
+	InnerCurrentDB         = "@currentDB"
 )
 
 func GetInnerSQL(dialect string) InnerSQL {
@@ -44,6 +45,8 @@ type mysqlDialect struct {
 func (m *mysqlDialect) ToNativeSQL(query string) (sql string) {
 	if strings.HasPrefix(query, InnerSelectTable_) {
 		sql = "SELECT * FROM " + strings.ReplaceAll(query, InnerSelectTable_, "")
+	} else if strings.HasPrefix(query, InnerSelectTableLimit_) {
+		sql = "SELECT * FROM " + strings.ReplaceAll(query, InnerSelectTableLimit_, "") + " LIMIT 100"
 	} else if query == InnerShowDatabases {
 		sql = "SHOW DATABASES"
 	} else if query == InnerShowTables {
@@ -62,6 +65,8 @@ type postgresDialect struct {
 func (p *postgresDialect) ToNativeSQL(query string) (sql string) {
 	if strings.HasPrefix(query, InnerSelectTable_) {
 		sql = `SELECT * FROM "` + strings.ReplaceAll(query, InnerSelectTable_, "") + `"`
+	} else if strings.HasPrefix(query, InnerSelectTableLimit_) {
+		sql = `SELECT * FROM "` + strings.ReplaceAll(query, InnerSelectTableLimit_, "") + `" LIMIT 100`
 	} else if query == InnerShowDatabases {
 		sql = "SELECT table_catalog as name FROM information_schema.tables"
 	} else if query == InnerShowTables {

pkg/inner_sql_test.go

@@ -36,6 +36,7 @@ func TestInnerSQL(t *testing.T) {
 
 	innerSQLs := []string{
 		pkg.InnerSelectTable_,
+		pkg.InnerSelectTableLimit_,
 		pkg.InnerShowDatabases,
 		pkg.InnerShowTables,
 		pkg.InnerCurrentDB,