fix

wangpengfei

2023-08-24 2128d417cbb9136ad5d8448fef7ce62a082922be

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package system
 
import (
    "github.com/flipped-aurora/gin-vue-admin/server/global"
    "github.com/flipped-aurora/gin-vue-admin/server/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
    }
 
}