-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
gorm.go
62 lines (54 loc) · 1.44 KB
/
gorm.go
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
56
57
58
59
60
61
62
package initialize
import (
"os"
"github.com/edufriendchen/hertz-vue-admin/server/global"
"github.com/edufriendchen/hertz-vue-admin/server/model/example"
"github.com/edufriendchen/hertz-vue-admin/server/model/system"
"go.uber.org/zap"
"gorm.io/gorm"
)
// Gorm 初始化数据库并产生数据库全局变量
// Author SliverHorn
func Gorm() *gorm.DB {
switch global.CONFIG.System.DbType {
case "mysql":
return GormMysql()
case "pgsql":
return GormPgSql()
default:
return GormMysql()
}
}
// RegisterTables 注册数据库表专用
// Author SliverHorn
func RegisterTables(db *gorm.DB) {
err := db.AutoMigrate(
// 系统模块表
system.SysApi{},
system.SysUser{},
system.SysBaseMenu{},
system.JwtBlacklist{},
system.SysAuthority{},
system.SysDictionary{},
system.SysOperationRecord{},
system.SysAutoCodeHistory{},
system.SysDictionaryDetail{},
system.SysBaseMenuParameter{},
system.SysBaseMenuBtn{},
system.SysAuthorityBtn{},
system.SysAutoCode{},
// 示例模块表
example.ExaFile{},
example.ExaCustomer{},
example.ExaFileChunk{},
example.ExaFileUploadAndDownload{},
// 自动化模块表
// Code generated by github.com/edufriendchen/hertz-vue-admin/server Begin; DO NOT EDIT.
// Code generated by github.com/edufriendchen/hertz-vue-admin/server End; DO NOT EDIT.
)
if err != nil {
global.LOG.Error("register table failed", zap.Error(err))
os.Exit(0)
}
global.LOG.Info("register table success")
}