zhangqian
2024-04-26 d0175e436b5987511dfedb713abd19cba6093fab
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
50
51
52
53
54
55
package initialize
 
import (
    "os"
    models "srm/model/outsourcing"
    "srm/model/purchase"
 
    "go.uber.org/zap"
    "gorm.io/gorm"
    "srm/global"
    "srm/model/test"
)
 
func Gorm() *gorm.DB {
    switch global.GVA_CONFIG.System.DbType {
    case "mysql":
        return GormMysql()
    case "pgsql":
        return GormPgSql()
    case "oracle":
        return GormOracle()
    case "mssql":
        return GormMssql()
    case "sqlite":
        return GormSqlite()
    default:
        return GormMysql()
    }
}
 
func RegisterTables() {
    db := global.GVA_DB
    err := db.AutoMigrate(
        test.Industry{},
        test.SupplierType{},
        test.Supplier{},
        test.Contract{},
        test.SupplierMaterial{},
        purchase.Purchase{},
        purchase.PurchaseProducts{},
        purchase.PurchaseProductConfirm{},
        purchase.PurchaseQualityInspection{},
        models.OutsourcingEnterprise{},
        models.OutsourcingMaterialApply{},
        models.OutsourcingOrder{},
        models.OutsourcingOrderDelivery{},
        models.OutsourcingOrderDeliveryDetails{},
        models.OutsourcingOrderProduct{},
    )
    if err != nil {
        global.GVA_LOG.Error("register table failed", zap.Error(err))
        os.Exit(0)
    }
    global.GVA_LOG.Info("register table success")
}