feat: support to select database (#27)

* update dep

* support to get databases and tables

* feat(dbserver): support dynamic database selection

- Introduce getClientWithDatabase method to allow querying specific databases
- Implement dbNameCache to store and retrieve database names
- Update Query method to use getClientWithDatabase with query-specific key
- Modify getClient to use getClientWithDatabase with default database

---------

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

Author: LinuxSuRen

pkg/data_query.go

@@ -27,7 +27,7 @@ import (
 
 func (s *dbserver) Query(ctx context.Context, query *server.DataQuery) (result *server.DataQueryResult, err error) {
 	var db *gorm.DB
-	if db, err = s.getClient(ctx); err != nil {
+	if db, err = s.getClientWithDatabase(ctx, query.Key); err != nil {
 		return
 	}
 

pkg/server.go

@@ -94,32 +94,41 @@ func createDB(user, password, address, database, driver string) (db *gorm.DB, er
 }
 
 var dbCache map[string]*gorm.DB = make(map[string]*gorm.DB)
+var dbNameCache map[string]string = make(map[string]string)
 
-func (s *dbserver) getClient(ctx context.Context) (db *gorm.DB, err error) {
+func (s *dbserver) getClientWithDatabase(ctx context.Context, dbName string) (db *gorm.DB, err error) {
 	store := remote.GetStoreFromContext(ctx)
 	if store == nil {
 		err = errors.New("no connect to database")
 	} else {
-		var ok bool
-		if db, ok = dbCache[store.Name]; ok && db != nil {
-			return
+		database := dbName
+		if database == "" {
+			if v, ok := store.Properties["database"]; ok && v != "" {
+				database = v
+			}
 		}
 
-		database := "atest"
 		driver := "mysql"
-		if v, ok := store.Properties["database"]; ok && v != "" {
-			database = v
-		}
 		if v, ok := store.Properties["driver"]; ok && v != "" {
 			driver = v
 		}
 
+		var ok bool
+		if db, ok = dbCache[store.Name]; ok && db != nil && dbNameCache[store.Name] == database {
+			return
+		}
+
 		if db, err = createDB(store.Username, store.Password, store.URL, database, driver); err == nil {
 			dbCache[store.Name] = db
+			dbNameCache[store.Name] = database
 		}
 	}
 	return
 }
+func (s *dbserver) getClient(ctx context.Context) (db *gorm.DB, err error) {
+	db, err = s.getClientWithDatabase(ctx, "")
+	return
+}
 
 func (s *dbserver) ListTestSuite(ctx context.Context, _ *server.Empty) (suites *remote.TestSuites, err error) {
 	items := make([]*TestSuite, 0)