feat: oracle支持服务名、数据库执行超时时间配置等

server/internal/db/api/db.go

@@ -9,6 +9,7 @@ import (
 	"mayfly-go/internal/db/api/form"
 	"mayfly-go/internal/db/api/vo"
 	"mayfly-go/internal/db/application"
+	"mayfly-go/internal/db/config"
 	"mayfly-go/internal/db/dbm/dbi"
 	"mayfly-go/internal/db/domain/entity"
 	msgapp "mayfly-go/internal/msg/application"
@@ -113,7 +114,7 @@ func (d *Db) ExecSql(rc *req.Ctx) {
 	}
 
 	// 比前端超时时间稍微快一点，可以提示到前端
-	ctx, cancel := context.WithTimeout(rc.MetaCtx, 58*time.Second)
+	ctx, cancel := context.WithTimeout(rc.MetaCtx, time.Duration(config.GetDbms().SqlExecTl)*time.Second)
 	defer cancel()
 
 	sqls, err := sqlparser.SplitStatementToPieces(sql, sqlparser.WithDialect(dbConn.Info.Type.Dialect()))

server/internal/db/api/form/instance.go

@@ -6,7 +6,7 @@ type InstanceForm struct {
 	Type               string `binding:"required" json:"type"` // 类型，mysql oracle等
 	Host               string `binding:"required" json:"host"`
 	Port               int    `json:"port"`
-	Sid                string `json:"sid"`
+	Extra              string `json:"extra"`
 	Username           string `json:"username"`
 	Password           string `json:"password"`
 	Params             string `json:"params"`

server/internal/db/api/vo/instance.go

@@ -8,8 +8,8 @@ type InstanceListVO struct {
 	Host       *string    `json:"host"`
 	Port       *int       `json:"port"`
 	Type       *string    `json:"type"`
-	Params     *string    `json:"params"`
-	Sid        *string    `json:"sid"`
+	Params     string     `json:"params"`
+	Extra      string     `json:"extra"`
 	Username   *string    `json:"username"`
 	Remark     *string    `json:"remark"`
 	CreateTime *time.Time `json:"createTime"`

server/internal/db/application/db.go

@@ -199,7 +199,6 @@ func (d *dbAppImpl) GetDbConnByInstanceId(instanceId uint64) (*dbi.DbConn, error
 func toDbInfo(instance *entity.DbInstance, dbId uint64, database string, tagPath ...string) *dbi.DbInfo {
 	di := new(dbi.DbInfo)
 	di.InstanceId = instance.Id
-	di.Sid = instance.Sid
 	di.Id = dbId
 	di.Database = database
 	di.TagPath = tagPath

server/internal/db/application/db_sql_exec.go

@@ -95,7 +95,7 @@ func (d *dbSqlExecAppImpl) Exec(ctx context.Context, execSqlReq *DbSqlExecReq) (
 		isSelect := strings.HasPrefix(lowerSql, "select")
 		if isSelect {
 			// 如果配置为0，则不校验分页参数
-			maxCount := config.GetDbQueryMaxCount()
+			maxCount := config.GetDbms().MaxResultSet
 			if maxCount != 0 {
 				if !strings.Contains(lowerSql, "limit") &&
 					// 兼容oracle rownum分页
@@ -207,7 +207,8 @@ func (d *dbSqlExecAppImpl) saveSqlExecLog(isQuery bool, dbSqlExecRecord *entity.
 		d.dbSqlExecRepo.Insert(context.TODO(), dbSqlExecRecord)
 		return
 	}
-	if config.GetDbSaveQuerySql() {
+
+	if config.GetDbms().QuerySqlSave {
 		dbSqlExecRecord.Table = "-"
 		dbSqlExecRecord.OldValue = "-"
 		dbSqlExecRecord.Type = entity.DbSqlExecTypeQuery
@@ -220,7 +221,7 @@ func (d *dbSqlExecAppImpl) doSelect(ctx context.Context, selectStmt *sqlparser.S
 	if selectExprsStr == "*" || strings.Contains(selectExprsStr, ".*") ||
 		len(strings.Split(selectExprsStr, ",")) > 1 {
 		// 如果配置为0，则不校验分页参数
-		maxCount := config.GetDbQueryMaxCount()
+		maxCount := config.GetDbms().MaxResultSet
 		// 哪些数据库跳过校验
 		skipped := dbi.DbTypeOracle == execSqlReq.DbConn.Info.Type || dbi.DbTypeMssql == execSqlReq.DbConn.Info.Type
 		if maxCount != 0 && !skipped {

server/internal/db/config/config.go

@@ -2,26 +2,33 @@ package config
 
 import (
 	sysapp "mayfly-go/internal/sys/application"
+	"mayfly-go/pkg/utils/conv"
 	"path/filepath"
 	"runtime"
 )
 
 const (
-	ConfigKeyDbSaveQuerySQL  string = "DbSaveQuerySQL"  // 数据库是否记录查询相关sql
-	ConfigKeyDbQueryMaxCount string = "DbQueryMaxCount" // 数据库查询的最大数量
+	ConfigKeyDbms            string = "DbmsConfig"      // dbms相关配置信息
 	ConfigKeyDbBackupRestore string = "DbBackupRestore" // 数据库备份
 	ConfigKeyDbMysqlBin      string = "MysqlBin"        // mysql可执行文件配置
 	ConfigKeyDbMariadbBin    string = "MariadbBin"      // mariadb可执行文件配置
 )
 
-// 获取数据库最大查询数量配置
-func GetDbQueryMaxCount() int {
-	return sysapp.GetConfigApp().GetConfig(ConfigKeyDbQueryMaxCount).IntValue(200)
+type Dbms struct {
+	QuerySqlSave bool // 是否记录查询类sql
+	MaxResultSet int  // 允许sql查询的最大结果集数。注: 0=不限制
+	SqlExecTl    int  // sql执行时间限制，超过该时间（单位：秒），执行将被取消
 }
 
-// 获取数据库是否记录查询相关sql配置
-func GetDbSaveQuerySql() bool {
-	return sysapp.GetConfigApp().GetConfig(ConfigKeyDbSaveQuerySQL).BoolValue(false)
+func GetDbms() *Dbms {
+	c := sysapp.GetConfigApp().GetConfig(ConfigKeyDbms)
+	jm := c.GetJsonMap()
+
+	dbmsConf := new(Dbms)
+	dbmsConf.QuerySqlSave = c.ConvBool(jm["querySqlSave"], false)
+	dbmsConf.MaxResultSet = conv.Str2Int(jm["maxResultSet"], 0)
+	dbmsConf.SqlExecTl = conv.Str2Int(jm["sqlExecTl"], 60)
+	return dbmsConf
 }
 
 type DbBackupRestore struct {

server/internal/db/dbm/dbi/info.go

@@ -16,7 +16,7 @@ type DbInfo struct {
 	Type     DbType // 类型，mysql postgres等
 	Host     string
 	Port     int
-	Sid      string // oracle数据库需要指定sid
+	Extra    string // 连接需要的其他额外参数（json字符串），如oracle数据库需要指定sid
 	Network  string
 	Username string
 	Password string

server/internal/db/dbm/oracle/meta.go

@@ -4,6 +4,8 @@ import (
 	"database/sql"
 	"fmt"
 	"mayfly-go/internal/db/dbm/dbi"
+	"mayfly-go/pkg/utils/anyx"
+	"mayfly-go/pkg/utils/jsonx"
 	"strings"
 
 	go_ora "github.com/sijms/go-ora/v2"
@@ -17,12 +19,11 @@ type OraMeta struct {
 }
 
 func (md *OraMeta) GetSqlDb(d *dbi.DbInfo) (*sql.DB, error) {
-	driverName := "oracle"
-
 	err := d.IfUseSshTunnelChangeIpPort()
 	if err != nil {
 		return nil, err
 	}
+
 	// 参数参考 https://github.com/sijms/go-ora?tab=readme-ov-file#other-connection-options
 	urlOptions := make(map[string]string)
 
@@ -50,9 +51,20 @@ func (md *OraMeta) GetSqlDb(d *dbi.DbInfo) (*sql.DB, error) {
 			}
 		}
 	}
+
+	// 从extra获取sid或serviceName
+	serviceName := ""
+	if d.Extra != "" {
+		extraMap := jsonx.ToMap(d.Extra)
+		serviceName = anyx.ConvString(extraMap["serviceName"])
+		if sid := anyx.ConvString(extraMap["sid"]); sid != "" {
+			urlOptions["SID"] = sid
+		}
+	}
+
 	urlOptions["TIMEOUT"] = "1000"
-	connStr := go_ora.BuildUrl(d.Host, d.Port, d.Sid, d.Username, d.Password, urlOptions)
-	conn, err := sql.Open(driverName, connStr)
+	connStr := go_ora.BuildUrl(d.Host, d.Port, serviceName, d.Username, d.Password, urlOptions)
+	conn, err := sql.Open("oracle", connStr)
 	if err != nil {
 		return nil, err
 	}

server/internal/db/domain/entity/db_instance.go

@@ -15,7 +15,7 @@ type DbInstance struct {
 	Host               string  `json:"host"`
 	Port               int     `json:"port"`
 	Network            string  `json:"network"`
-	Sid                string  `json:"sid"`
+	Extra              *string `json:"extra"` // 连接需要的其他额外参数（json格式）, 如oracle需要sid等
 	Username           string  `json:"username"`
 	Password           string  `json:"-"`
 	Params             *string `json:"params"`