feat: support to get sql explain data info

Author: LinuxSuRen

go.mod

@@ -5,7 +5,7 @@ go 1.22.4
 toolchain go1.22.6
 
 require (
-	github.com/linuxsuren/api-testing v0.0.19-0.20250312132236-e3e7357d6687
+	github.com/linuxsuren/api-testing v0.0.20-0.20250319020913-f5f9383e2948
 	github.com/spf13/cobra v1.8.1
 	github.com/stretchr/testify v1.9.0
 	github.com/taosdata/driver-go/v3 v3.6.0

go.sum

@@ -138,6 +138,8 @@ github.com/linuxsuren/api-testing v0.0.19-0.20250310093028-dd08c128ca0e h1:3dVXB
 github.com/linuxsuren/api-testing v0.0.19-0.20250310093028-dd08c128ca0e/go.mod h1:igcyUJb5Q463tMpdEJxgURLLPR2pTcV7HoRyuvyP/2Y=
 github.com/linuxsuren/api-testing v0.0.19-0.20250312132236-e3e7357d6687 h1:AFF1vWQbxOCqEENxK5F2va3d4Xmu5wK9tl2zZfUgHdY=
 github.com/linuxsuren/api-testing v0.0.19-0.20250312132236-e3e7357d6687/go.mod h1:igcyUJb5Q463tMpdEJxgURLLPR2pTcV7HoRyuvyP/2Y=
+github.com/linuxsuren/api-testing v0.0.20-0.20250319020913-f5f9383e2948 h1:YOyqT+9GQsAhRJ0nfqkFMtFJufAS3X/nKmxs25iPUss=
+github.com/linuxsuren/api-testing v0.0.20-0.20250319020913-f5f9383e2948/go.mod h1:igcyUJb5Q463tMpdEJxgURLLPR2pTcV7HoRyuvyP/2Y=
 github.com/linuxsuren/go-fake-runtime v0.0.4 h1:y+tvBuw6MKTCav8Bo5HWwaXhBx1Z//VAvqI3gpOWqvw=
 github.com/linuxsuren/go-fake-runtime v0.0.4/go.mod h1:zmh6J78hSnWZo68faMA2eKOdaEp8eFbERHi3ZB9xHCQ=
 github.com/linuxsuren/go-service v0.0.0-20231225060426-efabcd3a5161 h1:dSL/ah6zaRGqH3FW0ogtMjP6xCFXX5NsgWJTaNIofI4=

pkg/data_query.go

@@ -64,6 +64,8 @@ func (s *dbserver) Query(ctx context.Context, query *server.DataQuery) (result *
 		return
 	}
 
+	result.Meta.Labels = dbQuery.GetLabels(ctx, query.Sql)
+
 	var dataResult *server.DataQueryResult
 	now := time.Now()
 	if dataResult, err = sqlQuery(ctx, query.Sql, db); err == nil {
@@ -166,6 +168,7 @@ type DataQuery interface {
 	GetDatabases(context.Context) (databases []string, err error)
 	GetTables(ctx context.Context, currentDatabase string) (tables []string, err error)
 	GetCurrentDatabase() (string, error)
+	GetLabels(context.Context, string) map[string]string
 	GetClient() *gorm.DB
 }
 
@@ -236,6 +239,7 @@ func (q *commonDataQuery) GetTables(ctx context.Context, currentDatabase string)
 	}
 	return
 }
+
 func (q *commonDataQuery) GetCurrentDatabase() (current string, err error) {
 	var row *sql.Row
 	if row = q.db.Raw(q.currentDatabase).Row(); row != nil {
@@ -244,6 +248,21 @@ func (q *commonDataQuery) GetCurrentDatabase() (current string, err error) {
 	return
 }
 
+func (q *commonDataQuery) GetLabels(ctx context.Context, sql string) (metadata map[string]string) {
+	if databaseResult, err := sqlQuery(ctx, fmt.Sprintf("explain %s", sql), q.db); err == nil {
+		if len(databaseResult.Items) != 1 {
+			metadata = make(map[string]string)
+			for _, data := range databaseResult.Items[0].Data {
+				switch data.Key {
+				case "type":
+					metadata["sql_type"] = data.Value
+				}
+			}
+		}
+	}
+	return
+}
+
 func (q *commonDataQuery) GetClient() *gorm.DB {
 	return q.db
 }