mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-12-04 03:12:19 +08:00
115 lines
3.1 KiB
Go
115 lines
3.1 KiB
Go
package friend
|
|
|
|
import (
|
|
"context"
|
|
"github.com/OpenIMSDK/protocol/constant"
|
|
"github.com/OpenIMSDK/tools/log"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/config"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/db/cache"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/db/controller"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/db/mgo"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
|
|
tx2 "github.com/openimsdk/open-im-server/v3/pkg/common/db/tx"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/db/unrelation"
|
|
"github.com/redis/go-redis/v9"
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
var (
|
|
rdb redis.UniversalClient
|
|
mgo *unrelation.Mongo
|
|
ctx context.Context
|
|
|
|
friendDB *mgo.FriendMgo
|
|
friendRequestDB *mgo.FriendRequestMgo
|
|
blackDB *mgo.BlackMgo
|
|
|
|
friendDatabase controller.FriendDatabase
|
|
)
|
|
|
|
func InitDB() error {
|
|
addr := "172.16.8.142"
|
|
pwd := "openIM123"
|
|
|
|
config.Config.Redis.Address = []string{net.JoinHostPort(addr, "16379")}
|
|
config.Config.Redis.Password = pwd
|
|
config.Config.Mongo.Address = []string{net.JoinHostPort(addr, "37017")}
|
|
config.Config.Mongo.Database = "openIM_v3"
|
|
config.Config.Mongo.Username = "root"
|
|
config.Config.Mongo.Password = pwd
|
|
config.Config.Mongo.MaxPoolSize = 100
|
|
var err error
|
|
rdb, err = cache.NewRedis()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
mgo, err = unrelation.NewMongo()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
tx, err := tx2.NewAuto(context.Background(), mgo.GetClient())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
config.Config.Object.Enable = "minio"
|
|
config.Config.Object.ApiURL = "http://" + net.JoinHostPort(addr, "10002")
|
|
config.Config.Object.Minio.Bucket = "openim"
|
|
config.Config.Object.Minio.Endpoint = "http://" + net.JoinHostPort(addr, "10005")
|
|
config.Config.Object.Minio.AccessKeyID = "root"
|
|
config.Config.Object.Minio.SecretAccessKey = pwd
|
|
config.Config.Object.Minio.SignEndpoint = config.Config.Object.Minio.Endpoint
|
|
|
|
config.Config.Manager.UserID = []string{"openIM123456"}
|
|
config.Config.Manager.Nickname = []string{"openIM123456"}
|
|
|
|
ctx = context.WithValue(context.Background(), constant.OperationID, "debugOperationID")
|
|
ctx = context.WithValue(context.Background(), constant.OpUserID, config.Config.Manager.UserID[0])
|
|
|
|
if err := log.InitFromConfig("", "", 6, true, false, "", 2, 1); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
a, err := mgo.NewFriendMongo(mgo.GetDatabase())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
b, err := mgo.NewFriendRequestMongo(mgo.GetDatabase())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
c, err := mgo.NewBlackMongo(mgo.GetDatabase())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
friendDB = a.(*mgo.FriendMgo)
|
|
friendRequestDB = b.(*mgo.FriendRequestMgo)
|
|
blackDB = c.(*mgo.BlackMgo)
|
|
|
|
friendDatabase = controller.NewFriendDatabase(
|
|
friendDB,
|
|
friendRequestDB,
|
|
cache.NewFriendCacheRedis(rdb, friendDB, cache.GetDefaultOpt()),
|
|
tx,
|
|
)
|
|
|
|
return nil
|
|
}
|
|
|
|
func init() {
|
|
if err := InitDB(); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func TestName(t *testing.T) {
|
|
err := friendDatabase.AgreeFriendRequest(ctx, &relation.FriendRequestModel{FromUserID: "8449222880", ToUserID: "3136372091", HandleResult: 1, HandleMsg: "ok"})
|
|
t.Log(err)
|
|
}
|
|
|
|
func Test_GroupCreateCount(t *testing.T) {
|
|
|
|
}
|