From 3f739459db20e64f6812d1d0f578d00427ec14d5 Mon Sep 17 00:00:00 2001
From: withchao <993506633@qq.com>
Date: Mon, 16 Jan 2023 16:07:42 +0800
Subject: [PATCH] 1

---
 internal/rpc/friend/friend.go | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/internal/rpc/friend/friend.go b/internal/rpc/friend/friend.go
index 92d454745..183bdd0ca 100644
--- a/internal/rpc/friend/friend.go
+++ b/internal/rpc/friend/friend.go
@@ -4,6 +4,7 @@ import (
 	chat "Open_IM/internal/rpc/msg"
 	"Open_IM/pkg/common/config"
 	"Open_IM/pkg/common/constant"
+	"Open_IM/pkg/common/db"
 	imdb "Open_IM/pkg/common/db/mysql_model/im_mysql_model"
 	rocksCache "Open_IM/pkg/common/db/rocks_cache"
 	"Open_IM/pkg/common/log"
@@ -43,14 +44,15 @@ type friendServer struct {
 
 func NewFriendServer(port int) *friendServer {
 	log.NewPrivateLog(constant.LogFileName)
+	DB := db.DB.MysqlDB.DefaultGormDB()
 	return &friendServer{
 		rpcPort:            port,
 		rpcRegisterName:    config.Config.RpcRegisterName.OpenImFriendName,
 		etcdSchema:         config.Config.Etcd.EtcdSchema,
 		etcdAddr:           config.Config.Etcd.EtcdAddr,
-		friendModel:        imdb.NewFriend(nil), // TODO
-		friendRequestModel: imdb.NewFriendRequest(nil),
-		blackModel:         imdb.NewBlack(nil),
+		friendModel:        imdb.NewFriend(DB),
+		friendRequestModel: imdb.NewFriendRequest(DB),
+		blackModel:         imdb.NewBlack(DB),
 	}
 }