pull/1427/head
withchao
2 years ago
parent
1abfc8c805
commit
20158ec3ef
@ -0,0 +1,132 @@
|
||||
package user
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"github.com/OpenIMSDK/protocol/constant"
|
||||
"github.com/OpenIMSDK/protocol/sdkws"
|
||||
"github.com/OpenIMSDK/protocol/user"
|
||||
"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/newmgo"
|
||||
tablerelation "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"
|
||||
"math/rand"
|
||||
"net"
|
||||
"strconv"
|
||||
"testing"
|
||||
)
|
||||
|
||||
var (
|
||||
rdb redis.UniversalClient
|
||||
mgo *unrelation.Mongo
|
||||
ctx context.Context
|
||||
)
|
||||
|
||||
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)
|
||||
}
|
||||
|
||||
users := make([]*tablerelation.UserModel, 0)
|
||||
if len(config.Config.Manager.UserID) != len(config.Config.Manager.Nickname) {
|
||||
return errors.New("len(config.Config.Manager.AppManagerUid) != len(config.Config.Manager.Nickname)")
|
||||
}
|
||||
for k, v := range config.Config.Manager.UserID {
|
||||
users = append(users, &tablerelation.UserModel{UserID: v, Nickname: config.Config.Manager.Nickname[k], AppMangerLevel: constant.AppAdmin})
|
||||
}
|
||||
userDB, err := newmgo.NewUserMongo(mgo.GetDatabase())
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
//var client registry.SvcDiscoveryRegistry
|
||||
//_= client
|
||||
cache := cache.NewUserCacheRedis(rdb, userDB, cache.GetDefaultOpt())
|
||||
userMongoDB := unrelation.NewUserMongoDriver(mgo.GetDatabase())
|
||||
database := controller.NewUserDatabase(userDB, cache, tx, userMongoDB)
|
||||
//friendRpcClient := rpcclient.NewFriendRpcClient(client)
|
||||
//groupRpcClient := rpcclient.NewGroupRpcClient(client)
|
||||
//msgRpcClient := rpcclient.NewMessageRpcClient(client)
|
||||
|
||||
userSrv = &userServer{
|
||||
UserDatabase: database,
|
||||
//RegisterCenter: client,
|
||||
//friendRpcClient: &friendRpcClient,
|
||||
//groupRpcClient: &groupRpcClient,
|
||||
//friendNotificationSender: notification.NewFriendNotificationSender(&msgRpcClient, notification.WithDBFunc(database.FindWithError)),
|
||||
//userNotificationSender: notification.NewUserNotificationSender(&msgRpcClient, notification.WithUserFunc(database.FindWithError)),
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func init() {
|
||||
if err := InitDB(); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
|
||||
var userSrv *userServer
|
||||
|
||||
func TestName(t *testing.T) {
|
||||
userID := strconv.Itoa(int(rand.Uint32()))
|
||||
res, err := userSrv.UserRegister(ctx, &user.UserRegisterReq{
|
||||
Secret: config.Config.Secret,
|
||||
Users: []*sdkws.UserInfo{
|
||||
{
|
||||
UserID: userID,
|
||||
Nickname: userID,
|
||||
FaceURL: "",
|
||||
Ex: "",
|
||||
},
|
||||
},
|
||||
})
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
t.Log(res)
|
||||
|
||||
}
|
||||
@ -0,0 +1,19 @@
|
||||
package tx
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/OpenIMSDK/tools/tx"
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/mongo"
|
||||
)
|
||||
|
||||
func NewAuto(ctx context.Context, cli *mongo.Client) (tx.CtxTx, error) {
|
||||
var res map[string]any
|
||||
if err := cli.Database("admin").RunCommand(ctx, bson.M{"isMaster": 1}).Decode(&res); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if _, ok := res["setName"]; ok {
|
||||
return NewMongoTx(cli), nil
|
||||
}
|
||||
return NewInvalidTx(), nil
|
||||
}
|
||||
@ -0,0 +1,16 @@
|
||||
package tx
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/OpenIMSDK/tools/tx"
|
||||
)
|
||||
|
||||
func NewInvalidTx() tx.CtxTx {
|
||||
return invalid{}
|
||||
}
|
||||
|
||||
type invalid struct{}
|
||||
|
||||
func (m invalid) Transaction(ctx context.Context, fn func(ctx context.Context) error) error {
|
||||
return fn(ctx)
|
||||
}
|
||||
@ -0,0 +1,28 @@
|
||||
package tx
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/OpenIMSDK/tools/tx"
|
||||
"go.mongodb.org/mongo-driver/mongo"
|
||||
)
|
||||
|
||||
func NewMongoTx(client *mongo.Client) tx.CtxTx {
|
||||
return &mongoTx{
|
||||
client: client,
|
||||
}
|
||||
}
|
||||
|
||||
type mongoTx struct {
|
||||
client *mongo.Client
|
||||
}
|
||||
|
||||
func (m *mongoTx) Transaction(ctx context.Context, fn func(ctx context.Context) error) error {
|
||||
sess, err := m.client.StartSession()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
_, err = sess.WithTransaction(ctx, func(ctx mongo.SessionContext) (interface{}, error) {
|
||||
return nil, fn(ctx)
|
||||
})
|
||||
return err
|
||||
}
|
||||
Loading…
Reference in new issue