add inner sql support to remove dialect

Author: LinuxSuRen

pkg/data_query.go

@@ -64,6 +64,7 @@ func (s *dbserver) Query(ctx context.Context, query *server.DataQuery) (result *
 		return
 	}
 
+	query.Sql = dbQuery.GetInnerSQL().ToNativeSQL(query.Sql)
 	result.Meta.Labels = dbQuery.GetLabels(ctx, query.Sql)
 
 	var dataResult *server.DataQueryResult
@@ -170,27 +171,27 @@ type DataQuery interface {
 	GetCurrentDatabase() (string, error)
 	GetLabels(context.Context, string) map[string]string
 	GetClient() *gorm.DB
+	GetInnerSQL() InnerSQL
 }
 
 type commonDataQuery struct {
-	showDatabases, showTables, currentDatabase string
-	db                                         *gorm.DB
+	showTables string
+	db         *gorm.DB
+	innerSQL   InnerSQL
 }
 
 var _ DataQuery = &commonDataQuery{}
 
-func NewCommonDataQuery(showDatabases, showTables, currentDatabase string, db *gorm.DB) DataQuery {
+func NewCommonDataQuery(innerSQL InnerSQL, db *gorm.DB) DataQuery {
 	return &commonDataQuery{
-		showDatabases:   showDatabases,
-		showTables:      showTables,
-		currentDatabase: currentDatabase,
-		db:              db,
+		innerSQL: innerSQL,
+		db:       db,
 	}
 }
 
 func (q *commonDataQuery) GetDatabases(ctx context.Context) (databases []string, err error) {
 	var databaseResult *server.DataQueryResult
-	if databaseResult, err = sqlQuery(ctx, q.showDatabases, q.db); err == nil {
+	if databaseResult, err = sqlQuery(ctx, q.GetInnerSQL().ToNativeSQL(innerShowDatabases), q.db); err == nil {
 		for _, table := range databaseResult.Items {
 			for _, item := range table.GetData() {
 				if item.Key == "Database" || item.Key == "name" {
@@ -212,7 +213,7 @@ func (q *commonDataQuery) GetDatabases(ctx context.Context) (databases []string,
 }
 
 func (q *commonDataQuery) GetTables(ctx context.Context, currentDatabase string) (tables []string, err error) {
-	showTables := q.showTables
+	showTables := q.GetInnerSQL().ToNativeSQL(innerShowTables)
 	if strings.Contains(showTables, "%s") {
 		showTables = fmt.Sprintf(showTables, currentDatabase)
 	}
@@ -242,7 +243,7 @@ func (q *commonDataQuery) GetTables(ctx context.Context, currentDatabase string)
 
 func (q *commonDataQuery) GetCurrentDatabase() (current string, err error) {
 	var row *sql.Row
-	if row = q.db.Raw(q.currentDatabase).Row(); row != nil {
+	if row = q.db.Raw(q.GetInnerSQL().ToNativeSQL(innerCurrentDB)).Row(); row != nil {
 		err = row.Scan(&current)
 	}
 	return
@@ -266,3 +267,7 @@ func (q *commonDataQuery) GetLabels(ctx context.Context, sql string) (metadata m
 func (q *commonDataQuery) GetClient() *gorm.DB {
 	return q.db
 }
+
+func (q *commonDataQuery) GetInnerSQL() InnerSQL {
+	return q.innerSQL
+}

pkg/inner_sql.go

@@ -0,0 +1,75 @@
+/*
+Copyright 2025 API Testing Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+	http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package pkg
+
+import "strings"
+
+type InnerSQL interface {
+	ToNativeSQL(query string) string
+}
+
+const (
+	innerSelectTable_  = "@selectTable_"
+	innerShowDatabases = "@showDatabases"
+	innerShowTables    = "@showTables"
+	innerCurrentDB     = "@currentDB"
+)
+
+func GetInnerSQL(dialect string) InnerSQL {
+	switch dialect {
+	case "postgres":
+		return &postgresDialect{}
+	default:
+		return &mysqlDialect{}
+	}
+}
+
+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 query == innerShowDatabases {
+		sql = "SHOW DATABASES"
+	} else if query == innerShowTables {
+		sql = "SHOW TABLES"
+	} else if query == innerCurrentDB {
+		sql = "SELECT DATABASE() as name"
+	} else {
+		sql = query
+	}
+	return
+}
+
+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 query == innerShowDatabases {
+		sql = "SELECT table_catalog as name FROM information_schema.tables"
+	} else if query == innerShowTables {
+		sql = `SELECT table_name FROM information_schema.tables WHERE table_catalog = '%s' and table_schema != 'pg_catalog' and table_schema != 'information_schema'`
+	} else if query == innerCurrentDB {
+		sql = "SELECT current_database() as name"
+	} else {
+		sql = query
+	}
+	return
+}

pkg/server.go

@@ -128,13 +128,7 @@ func (s *dbserver) getClientWithDatabase(ctx context.Context, dbName string) (db
 			}
 		}
 
-		switch driver {
-		case "postgres":
-			dbQuery = NewCommonDataQuery("select table_catalog as name from information_schema.tables",
-				`SELECT table_name FROM information_schema.tables WHERE table_catalog = '%s' and table_schema != 'pg_catalog' and table_schema != 'information_schema'`, "SELECT current_database() as name", db)
-		default:
-			dbQuery = NewCommonDataQuery("show databases", "show tables", "SELECT DATABASE() as name", db)
-		}
+		dbQuery = NewCommonDataQuery(GetInnerSQL(driver), db)
 	}
 	return
 }