Merge branch 'errcode' of github.com:OpenIMSDK/Open-IM-Server into errcode

# Conflicts:
#	internal/rpc/group/group.go
#	pkg/common/db/model/group.go
This commit is contained in:
wangchuxiao 2023-01-28 13:23:34 +08:00
commit 36e82d0a6d
2 changed files with 44 additions and 10 deletions

View File

@ -5,6 +5,7 @@ import (
chat "Open_IM/internal/rpc/msg" chat "Open_IM/internal/rpc/msg"
"Open_IM/pkg/common/config" "Open_IM/pkg/common/config"
"Open_IM/pkg/common/constant" "Open_IM/pkg/common/constant"
"Open_IM/pkg/common/db/controller"
"Open_IM/pkg/common/db/model" "Open_IM/pkg/common/db/model"
imdb "Open_IM/pkg/common/db/mysql" imdb "Open_IM/pkg/common/db/mysql"
"Open_IM/pkg/common/log" "Open_IM/pkg/common/log"
@ -41,17 +42,34 @@ type groupServer struct {
rpcRegisterName string rpcRegisterName string
etcdSchema string etcdSchema string
etcdAddr []string etcdAddr []string
imdb.GroupInterface controller.GroupInterface
} }
func NewGroupServer(port int) *groupServer { func NewGroupServer(port int) *groupServer {
log.NewPrivateLog(constant.LogFileName) log.NewPrivateLog(constant.LogFileName)
return &groupServer{ g := groupServer{
rpcPort: port, rpcPort: port,
rpcRegisterName: config.Config.RpcRegisterName.OpenImGroupName, rpcRegisterName: config.Config.RpcRegisterName.OpenImGroupName,
etcdSchema: config.Config.Etcd.EtcdSchema, etcdSchema: config.Config.Etcd.EtcdSchema,
etcdAddr: config.Config.Etcd.EtcdAddr, etcdAddr: config.Config.Etcd.EtcdAddr,
} }
//mysql init
var mysql imdb.Mysql
var groupModel imdb.Group
err := mysql.InitConn().AutoMigrateModel(&groupModel)
if err != nil {
panic("db init err:" + err.Error())
}
if mysql.GormConn() != nil {
groupModel.DB = mysql.GormConn()
} else {
panic("db init err:" + "conn is nil")
}
//redis
//mongo
g.group = model.NewGroupController(groupModel)
return &g
} }
func (s *groupServer) Run() { func (s *groupServer) Run() {
@ -69,6 +87,7 @@ func (s *groupServer) Run() {
panic("listening err:" + err.Error() + s.rpcRegisterName) panic("listening err:" + err.Error() + s.rpcRegisterName)
} }
log.NewInfo("", "listen network success, ", address, listener) log.NewInfo("", "listen network success, ", address, listener)
defer listener.Close() defer listener.Close()
//grpc server //grpc server
recvSize := 1024 * 1024 * constant.GroupRPCRecvSize recvSize := 1024 * 1024 * constant.GroupRPCRecvSize

View File

@ -10,7 +10,18 @@ import (
"gorm.io/gorm/logger" "gorm.io/gorm/logger"
) )
func ConnectToDB() *gorm.DB { type Mysql struct {
gormConn *gorm.DB
}
func (m *Mysql) GormConn() *gorm.DB {
return m.gormConn
}
func (m *Mysql) SetGormConn(gormConn *gorm.DB) {
m.gormConn = gormConn
}
func (m *Mysql) InitConn() *Mysql {
dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4&parseTime=true&loc=Local", dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s?charset=utf8mb4&parseTime=true&loc=Local",
config.Config.Mysql.DBUserName, config.Config.Mysql.DBPassword, config.Config.Mysql.DBAddress[0], "mysql") config.Config.Mysql.DBUserName, config.Config.Mysql.DBPassword, config.Config.Mysql.DBAddress[0], "mysql")
var db *gorm.DB var db *gorm.DB
@ -51,18 +62,22 @@ func ConnectToDB() *gorm.DB {
sqlDB.SetConnMaxLifetime(time.Second * time.Duration(config.Config.Mysql.DBMaxLifeTime)) sqlDB.SetConnMaxLifetime(time.Second * time.Duration(config.Config.Mysql.DBMaxLifeTime))
sqlDB.SetMaxOpenConns(config.Config.Mysql.DBMaxOpenConns) sqlDB.SetMaxOpenConns(config.Config.Mysql.DBMaxOpenConns)
sqlDB.SetMaxIdleConns(config.Config.Mysql.DBMaxIdleConns) sqlDB.SetMaxIdleConns(config.Config.Mysql.DBMaxIdleConns)
return db m.SetGormConn(db)
return m
} }
//models := []interface{}{&Friend{}, &FriendRequest{}, &Group{}, &GroupMember{}, &GroupRequest{}, //models := []interface{}{&Friend{}, &FriendRequest{}, &Group{}, &GroupMember{}, &GroupRequest{},
// &User{}, &Black{}, &ChatLog{}, &Conversation{}, &AppVersion{}} // &User{}, &Black{}, &ChatLog{}, &Conversation{}, &AppVersion{}}
func initModel(db *gorm.DB, model interface{}) *gorm.DB { func (m *Mysql) AutoMigrateModel(model interface{}) error {
db.AutoMigrate(model) err := m.gormConn.AutoMigrate(model)
db.Set("gorm:table_options", "CHARSET=utf8") if err != nil {
db.Set("gorm:table_options", "collation=utf8_unicode_ci") return err
_ = db.Migrator().CreateTable(model) }
return db.Model(model) m.gormConn.Set("gorm:table_options", "CHARSET=utf8")
m.gormConn.Set("gorm:table_options", "collation=utf8_unicode_ci")
_ = m.gormConn.Migrator().CreateTable(model)
return nil
} }
type Writer struct{} type Writer struct{}