mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-05-09 20:49:34 +08:00
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:
commit
36e82d0a6d
@ -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
|
||||||
|
@ -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{}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user