Refactor code

This commit is contained in:
wenxu12345 2021-12-31 11:08:16 +08:00
parent 9782802bd8
commit 82c0f8a863

View File

@ -53,13 +53,48 @@ func initMysqlDB() {
log.NewInfo("open db ok ", dsn) log.NewInfo("open db ok ", dsn)
db.AutoMigrate(&Friend{}, //db.AutoMigrate(&Friend{},
&FriendRequest{}, // &FriendRequest{},
&Group{}, // &Group{},
&GroupMember{}, // &GroupMember{},
&GroupRequest{}, // &GroupRequest{},
&User{}, // &User{},
&Black{}) // &Black{})
if !db.HasTable(&Friend{}) {
log.NewInfo("CreateTable Friend")
db.CreateTable(&Friend{})
}
if !db.HasTable(&FriendRequest{}) {
log.NewInfo("CreateTable FriendRequest")
db.CreateTable(&FriendRequest{})
}
if !db.HasTable(&Group{}) {
log.NewInfo("CreateTable Group")
db.CreateTable(&Group{})
}
if !db.HasTable(&GroupMember{}) {
log.NewInfo("CreateTable GroupMember")
db.CreateTable(&GroupMember{})
}
if !db.HasTable(&GroupRequest{}) {
log.NewInfo("CreateTable GroupRequest")
db.CreateTable(&GroupRequest{})
}
if !db.HasTable(&User{}) {
log.NewInfo("CreateTable User")
db.CreateTable(&User{})
}
if !db.HasTable(&Black{}) {
log.NewInfo("CreateTable Black")
db.CreateTable(&Black{})
}
return return