chore: support to show version (#34)

Co-authored-by: rick <[email protected]>

Author: LinuxSuRen

pkg/data_query.go

@@ -23,6 +23,7 @@ import (
 	"reflect"
 	"sort"
 	"strings"
+	"sync"
 	"time"
 
 	"github.com/linuxsuren/api-testing/pkg/server"
@@ -44,32 +45,48 @@ func (s *dbserver) Query(ctx context.Context, query *server.DataQuery) (result *
 		Meta:  &server.DataMeta{},
 	}
 
-	// query database and tables
-	if result.Meta.Databases, err = dbQuery.GetDatabases(ctx); err != nil {
-		log.Printf("failed to query databases: %v\n", err)
-	}
+	wg := sync.WaitGroup{}
 
-	if result.Meta.CurrentDatabase = query.Key; query.Key == "" {
-		if result.Meta.CurrentDatabase, err = dbQuery.GetCurrentDatabase(); err != nil {
-			log.Printf("failed to query current database: %v\n", err)
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		// query database and tables
+		if result.Meta.Databases, err = dbQuery.GetDatabases(ctx); err != nil {
+			log.Printf("failed to query databases: %v\n", err)
 		}
-	}
+	}()
 
-	if result.Meta.Tables, err = dbQuery.GetTables(ctx, result.Meta.CurrentDatabase); err != nil {
-		log.Printf("failed to query tables: %v\n", err)
-	}
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		if result.Meta.CurrentDatabase = query.Key; query.Key == "" {
+			if result.Meta.CurrentDatabase, err = dbQuery.GetCurrentDatabase(); err != nil {
+				log.Printf("failed to query current database: %v\n", err)
+			}
+		}
 
+		if result.Meta.Tables, err = dbQuery.GetTables(ctx, result.Meta.CurrentDatabase); err != nil {
+			log.Printf("failed to query tables: %v\n", err)
+		}
+	}()
+
+	defer wg.Wait()
 	// query data
 	if query.Sql == "" {
 		return
 	}
 
 	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,
-	})
+
+	wg.Add(1)
+	go func() {
+		wg.Done()
+		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()
@@ -254,10 +271,7 @@ func (q *commonDataQuery) GetCurrentDatabase() (current string, err error) {
 
 func (q *commonDataQuery) GetLabels(ctx context.Context, sql string) (metadata []*server.Pair) {
 	metadata = make([]*server.Pair, 0)
-	if databaseResult, err := sqlQuery(ctx, fmt.Sprintf("explain %s", sql), q.db); err == nil {
-		if len(databaseResult.Items) != 1 {
-			return
-		}
+	if databaseResult, err := sqlQuery(ctx, fmt.Sprintf("explain %s", sql), q.db); err == nil && len(databaseResult.Items) != 1 {
 		for _, data := range databaseResult.Items[0].Data {
 			switch data.Key {
 			case "type":
@@ -268,6 +282,18 @@ func (q *commonDataQuery) GetLabels(ctx context.Context, sql string) (metadata [
 			}
 		}
 	}
+
+	if databaseResult, err := sqlQuery(ctx, `show variables like 'version'`, q.db); err == nil && len(databaseResult.Items) >= 1 {
+		for _, data := range databaseResult.Items[0].Data {
+			switch data.Key {
+			case "version":
+				metadata = append(metadata, &server.Pair{
+					Key:   "version",
+					Value: data.Value,
+				})
+			}
+		}
+	}
 	return
 }