复制项目

2026-01-14 22:35:45 +08:00
parent 305d526110
commit b7f8db7d08
297 changed files with 81784 additions and 0 deletions
--- a/internal/rpc/admin/start.go
+++ b/internal/rpc/admin/start.go
@@ -0,0 +1,110 @@
+package admin
+
+import (
+	"context"
+	"crypto/md5"
+	"encoding/hex"
+	"math/rand"
+	"time"
+
+	"git.imall.cloud/openim/chat/pkg/common/config"
+	"git.imall.cloud/openim/chat/pkg/common/constant"
+	"git.imall.cloud/openim/chat/pkg/common/db/database"
+	"git.imall.cloud/openim/chat/pkg/common/db/dbutil"
+	"git.imall.cloud/openim/chat/pkg/common/db/table/admin"
+	"git.imall.cloud/openim/chat/pkg/common/imapi"
+	"git.imall.cloud/openim/chat/pkg/common/tokenverify"
+	adminpb "git.imall.cloud/openim/chat/pkg/protocol/admin"
+	"git.imall.cloud/openim/chat/pkg/protocol/chat"
+	chatClient "git.imall.cloud/openim/chat/pkg/rpclient/chat"
+	"github.com/openimsdk/tools/db/mongoutil"
+	"github.com/openimsdk/tools/db/redisutil"
+	"github.com/openimsdk/tools/discovery"
+	"github.com/openimsdk/tools/errs"
+	"github.com/openimsdk/tools/mw"
+	"github.com/openimsdk/tools/utils/runtimeenv"
+	"google.golang.org/grpc"
+	"google.golang.org/grpc/credentials/insecure"
+)
+
+type Config struct {
+	RpcConfig     config.Admin
+	RedisConfig   config.Redis
+	MongodbConfig config.Mongo
+	Discovery     config.Discovery
+	Share         config.Share
+
+	RuntimeEnv string
+}
+
+func Start(ctx context.Context, config *Config, client discovery.SvcDiscoveryRegistry, server *grpc.Server) error {
+	config.RuntimeEnv = runtimeenv.PrintRuntimeEnvironment()
+
+	if len(config.Share.ChatAdmin) == 0 {
+		return errs.New("share chat admin not configured")
+	}
+	rand.Seed(time.Now().UnixNano())
+	rdb, err := redisutil.NewRedisClient(ctx, config.RedisConfig.Build())
+	if err != nil {
+		return err
+	}
+	mgocli, err := mongoutil.NewMongoDB(ctx, config.MongodbConfig.Build())
+	if err != nil {
+		return err
+	}
+	var srv adminServer
+	srv.Database, err = database.NewAdminDatabase(mgocli, rdb)
+	if err != nil {
+		return err
+	}
+	srv.ChatDatabase, err = database.NewChatDatabase(mgocli, rdb)
+	if err != nil {
+		return err
+	}
+	conn, err := client.GetConn(ctx, config.Discovery.RpcService.Chat, grpc.WithTransportCredentials(insecure.NewCredentials()), mw.GrpcClient())
+	if err != nil {
+		return err
+	}
+	srv.Chat = chatClient.NewChatClient(chat.NewChatClient(conn))
+	srv.Token = &tokenverify.Token{
+		Expires: time.Duration(config.RpcConfig.TokenPolicy.Expire) * time.Hour * 24,
+		Secret:  config.RpcConfig.Secret,
+	}
+	srv.ImApiCaller = imapi.New(config.Share.OpenIM.ApiURL, config.Share.OpenIM.Secret, config.Share.OpenIM.AdminUserID)
+	if err := srv.initAdmin(ctx, config.Share.ChatAdmin, config.Share.OpenIM.AdminUserID); err != nil {
+		return err
+	}
+	adminpb.RegisterAdminServer(server, &srv)
+	return nil
+}
+
+type adminServer struct {
+	adminpb.UnimplementedAdminServer
+	Database     database.AdminDatabaseInterface
+	ChatDatabase database.ChatDatabaseInterface
+	Chat         *chatClient.ChatClient
+	Token        *tokenverify.Token
+	ImApiCaller  imapi.CallerInterface
+}
+
+func (o *adminServer) initAdmin(ctx context.Context, admins []string, imUserID string) error {
+	for _, account := range admins {
+		if _, err := o.Database.GetAdmin(ctx, account); err == nil {
+			continue
+		} else if !dbutil.IsDBNotFound(err) {
+			return err
+		}
+		sum := md5.Sum([]byte(account))
+		a := admin.Admin{
+			Account:    account,
+			UserID:     imUserID,
+			Password:   hex.EncodeToString(sum[:]),
+			Level:      constant.DefaultAdminLevel,
+			CreateTime: time.Now(),
+		}
+		if err := o.Database.AddAdminAccount(ctx, []*admin.Admin{&a}); err != nil {
+			return err
+		}
+	}
+	return nil
+}