From 5af01e4da18b2fe93105c3c1244838d5d4bbe6f7 Mon Sep 17 00:00:00 2001
From: Gordon <1432970085@qq.com>
Date: Sat, 28 Jan 2023 11:28:11 +0800
Subject: [PATCH] group update

---
 internal/rpc/group/group.go       | 23 ++++++++++++++++++++---
 pkg/common/db/model/group.go      |  4 ++--
 pkg/common/db/mysql/init_mysql.go | 31 +++++++++++++++++++++++--------
 3 files changed, 45 insertions(+), 13 deletions(-)

diff --git a/internal/rpc/group/group.go b/internal/rpc/group/group.go
index 2bdb3ea56..a8c1db503 100644
--- a/internal/rpc/group/group.go
+++ b/internal/rpc/group/group.go
@@ -41,18 +41,34 @@ type groupServer struct {
 	rpcRegisterName string
 	etcdSchema      string
 	etcdAddr        []string
-	               imdb.GroupInterface
-
+	group           model.GroupInterface
 }
 
 func NewGroupServer(port int) *groupServer {
 	log.NewPrivateLog(constant.LogFileName)
-	return &groupServer{
+	g := groupServer{
 		rpcPort:         port,
 		rpcRegisterName: config.Config.RpcRegisterName.OpenImGroupName,
 		etcdSchema:      config.Config.Etcd.EtcdSchema,
 		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() {
@@ -70,6 +86,7 @@ func (s *groupServer) Run() {
 		panic("listening err:" + err.Error() + s.rpcRegisterName)
 	}
 	log.NewInfo("", "listen network success, ", address, listener)
+
 	defer listener.Close()
 	//grpc server
 	recvSize := 1024 * 1024 * constant.GroupRPCRecvSize
diff --git a/pkg/common/db/model/group.go b/pkg/common/db/model/group.go
index 0f765683e..eb061bf98 100644
--- a/pkg/common/db/model/group.go
+++ b/pkg/common/db/model/group.go
@@ -36,14 +36,14 @@ type MySqlDatabase struct {
 }
 
 func (m *MySqlDatabase) Delete(ctx context.Context, groupIDs []string) error {
-	panic("implement me")
+	return m.Delete(ctx, groupIDs)
 }
 
 func NewMySqlDatabase(db mysql.GroupModelInterface) DataBase {
 	return &MySqlDatabase{db}
 }
 func (m *MySqlDatabase) DeleteTx(ctx context.Context, groupIDs []string) error {
-	return nil
+	return m.Delete(ctx, groupIDs)
 }
 
 func NewGroupController(groupModel mysql.GroupModelInterface, rdb redis.UniversalClient, mdb *mongo.Client) *GroupController {
diff --git a/pkg/common/db/mysql/init_mysql.go b/pkg/common/db/mysql/init_mysql.go
index 839902dfb..91e76d4c4 100644
--- a/pkg/common/db/mysql/init_mysql.go
+++ b/pkg/common/db/mysql/init_mysql.go
@@ -10,7 +10,18 @@ import (
 	"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",
 		config.Config.Mysql.DBUserName, config.Config.Mysql.DBPassword, config.Config.Mysql.DBAddress[0], "mysql")
 	var db *gorm.DB
@@ -51,18 +62,22 @@ func ConnectToDB() *gorm.DB {
 	sqlDB.SetConnMaxLifetime(time.Second * time.Duration(config.Config.Mysql.DBMaxLifeTime))
 	sqlDB.SetMaxOpenConns(config.Config.Mysql.DBMaxOpenConns)
 	sqlDB.SetMaxIdleConns(config.Config.Mysql.DBMaxIdleConns)
-	return db
+	m.SetGormConn(db)
+	return m
 }
 
 //models := []interface{}{&Friend{}, &FriendRequest{}, &Group{}, &GroupMember{}, &GroupRequest{},
 //	&User{}, &Black{}, &ChatLog{}, &Conversation{}, &AppVersion{}}
 
-func initModel(db *gorm.DB, model interface{}) *gorm.DB {
-	db.AutoMigrate(model)
-	db.Set("gorm:table_options", "CHARSET=utf8")
-	db.Set("gorm:table_options", "collation=utf8_unicode_ci")
-	_ = db.Migrator().CreateTable(model)
-	return db.Model(model)
+func (m *Mysql) AutoMigrateModel(model interface{}) error {
+	err := m.gormConn.AutoMigrate(model)
+	if err != nil {
+		return err
+	}
+	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{}