50 lines
1.1 KiB
Go
50 lines
1.1 KiB
Go
|
package system
|
||
|
|
|
||
|
|
import (
|
||
|
|
"miniapp/global"
|
||
|
|
"miniapp/model/system/response"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Database interface {
|
||
|
|
GetDB(businessDB string) (data []response.Db, err error)
|
||
|
|
GetTables(businessDB string, dbName string) (data []response.Table, err error)
|
||
|
|
GetColumn(businessDB string, tableName string, dbName string) (data []response.Column, err error)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (autoCodeService *AutoCodeService) Database(businessDB string) Database {
|
||
|
|
|
||
|
|
if businessDB == "" {
|
||
|
|
switch global.GVA_CONFIG.System.DbType {
|
||
|
|
case "mysql":
|
||
|
|
return AutoCodeMysql
|
||
|
|
case "pgsql":
|
||
|
|
return AutoCodePgsql
|
||
|
|
case "sqlite":
|
||
|
|
return AutoCodeSqlite
|
||
|
|
default:
|
||
|
|
return AutoCodeMysql
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
for _, info := range global.GVA_CONFIG.DBList {
|
||
|
|
if info.AliasName == businessDB {
|
||
|
|
switch info.Type {
|
||
|
|
case "mysql":
|
||
|
|
return AutoCodeMysql
|
||
|
|
case "mssql":
|
||
|
|
return AutoCodeMssql
|
||
|
|
case "pgsql":
|
||
|
|
return AutoCodePgsql
|
||
|
|
case "oracle":
|
||
|
|
return AutoCodeOracle
|
||
|
|
case "sqlite":
|
||
|
|
return AutoCodeSqlite
|
||
|
|
default:
|
||
|
|
return AutoCodeMysql
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return AutoCodeMysql
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|