feat: add im kafka admin integration

2026-08-28 00:05:54 +08:00
parent d53b0c1b62
commit 4779214591
75 changed files with 9053 additions and 185 deletions
@@ -12,9 +12,13 @@ import (
 	"encoding/json"
 	"fmt"
 	"hotgo/internal/consts"
+	"hotgo/internal/library/contexts"
+	"hotgo/internal/library/imkafka"
 	"hotgo/internal/library/imtenant"
 	"hotgo/internal/model/input/sysin"
 	"hotgo/internal/service"
+	"net/url"
+	"strconv"
 	"strings"
 	"time"

@@ -40,36 +44,34 @@ func (s *sSysImUser) List(ctx context.Context, in *sysin.ImUserListInp) (list []
 	if err = in.Filter(ctx); err != nil {
 		return
 	}
-
-	scope, err := resolveImAdminScope(ctx, in.TenantId)
+	tenant, err := resolveImAdminTenant(ctx, in.TenantId)
 	if err != nil {
 		return
 	}
-	db, tenant, err := imtenant.DB(ctx, scope.TenantId)
-	if err != nil {
-		return
+	values := addPageQuery(url.Values{}, in.Page, in.PerPage)
+	if in.UserId != "" {
+		values.Set("user_id", in.UserId)
 	}
-
-	mod := s.listModel(ctx, db)
-	mod = s.applyListWhere(mod, in)
-	mod = mod.Page(in.Page, in.PerPage).OrderDesc("u.id")
-
-	if err = mod.ScanAndCount(&list, &totalCount, false); err != nil {
-		err = gerror.Wrap(err, "获取IM用户列表失败，请稍后重试！")
-		return
+	if in.Nickname != "" {
+		values.Set("nickname", in.Nickname)
 	}
-	if len(list) == 0 {
-		return
+	if in.AuthName != "" {
+		values.Set("auth_name", in.AuthName)
 	}
-
-	for _, item := range list {
-		item.TenantId = tenant.Id
-		item.TenantName = tenant.Name
+	if in.State != nil {
+		values.Set("state", intToString(*in.State))
 	}
-	if err = s.fillAuthAndCredentials(ctx, db, list); err != nil {
-		return
-	}
-	return
+	addTimeRangeQuery(values, "created_at", in.CreatedAt)
+	addTimeRangeQuery(values, "last_seen", in.LastSeen)
+	return imAdminQueryList[sysin.ImUserListModel](ctx, tenant,
+		"/internal/admin-query/tenants/"+url.PathEscape(tenant.Code)+"/users",
+		values,
+		map[string]string{
+			"public":  "profileJson",
+			"trusted": "trustedJson",
+			"access":  "accessJson",
+		},
+	)
 }

 func (s *sSysImUser) listModel(ctx context.Context, db gdb.DB) *gdb.Model {
@@ -120,13 +122,16 @@ func (s *sSysImUser) fillAuthAndCredentials(ctx context.Context, db gdb.DB, list
 	ids := make([]int64, 0, len(list))
 	userMap := make(map[int64]*sysin.ImUserListModel, len(list))
 	for _, item := range list {
-		userId, parseErr := sysin.ParseOptionalInt64(item.UserId)
-		if parseErr != nil {
-			return parseErr
+		userId := item.UserId
+		if userId <= 0 {
+			continue
 		}
 		ids = append(ids, userId)
 		userMap[userId] = item
 	}
+	if len(ids) == 0 {
+		return nil
+	}

 	credentialRows, err := db.Model("im_credentials").Safe().Ctx(ctx).
 		Fields("userid", "method", "value", "done").
@@ -164,41 +169,18 @@ func (s *sSysImUser) View(ctx context.Context, in *sysin.ImUserViewInp) (res *sy
 	if err = in.Filter(ctx); err != nil {
 		return
 	}
-
-	scope, err := resolveImAdminScope(ctx, in.TenantId)
+	tenant, err := resolveImAdminTenant(ctx, in.TenantId)
 	if err != nil {
-		return
-	}
-	db, tenant, err := imtenant.DB(ctx, scope.TenantId)
-	if err != nil {
-		return
-	}
-
-	userId, err := sysin.ParseOptionalInt64(in.UserId)
-	if err != nil {
-		return nil, err
-	}
-	fields := append(s.viewFields(), "u.access access_json", "u.public profile_json", "u.trusted trusted_json")
-	err = db.Model("im_users u").Safe().Ctx(ctx).
-		LeftJoin("im_user_profiles up", "up.user_id = u.id").
-		LeftJoin("im_user_security us", "us.user_id = u.id").
-		Fields(fields...).
-		Where("u.id", userId).
-		Scan(&res)
-	if err != nil {
-		return nil, gerror.Wrap(err, "获取IM用户信息失败，请稍后重试！")
-	}
-	if res == nil {
 		return nil, nil
 	}
-
-	res.TenantId = tenant.Id
-	res.TenantName = tenant.Name
-	list := []*sysin.ImUserListModel{&res.ImUserListModel}
-	if err = s.fillAuthAndCredentials(ctx, db, list); err != nil {
-		return nil, err
-	}
-	return
+	return imAdminQueryOne[sysin.ImUserViewModel](ctx, tenant,
+		"/internal/admin-query/tenants/"+url.PathEscape(tenant.Code)+"/users/"+url.PathEscape(in.UserId),
+		map[string]string{
+			"public":  "profileJson",
+			"trusted": "trustedJson",
+			"access":  "accessJson",
+		},
+	)
 }

 func (s *sSysImUser) viewFields() []interface{} {
@@ -306,45 +288,104 @@ func (s *sSysImUser) Devices(ctx context.Context, in *sysin.ImUserDeviceListInp)
 	if err = in.Filter(ctx); err != nil {
 		return
 	}
-	scope, err := resolveImAdminScope(ctx, in.TenantId)
-	if err != nil {
-		return
-	}
-	userId, err := sysin.ParseOptionalInt64(in.UserId)
-	if err != nil {
-		return
-	}
-	db, tenant, err := imtenant.DB(ctx, scope.TenantId)
+	tenant, err := resolveImAdminTenant(ctx, in.TenantId)
 	if err != nil {
 		return
 	}
+	return imAdminQueryList[sysin.ImUserDeviceListModel](ctx, tenant,
+		"/internal/admin-query/tenants/"+url.PathEscape(tenant.Code)+"/users/"+url.PathEscape(in.UserId)+"/devices",
+		addPageQuery(url.Values{}, in.Page, in.PerPage),
+		nil,
+	)
+}

-	mod := db.Model("im_devices d").Safe().Ctx(ctx).
-		LeftJoin("im_device_push_tokens p", "p.user_id=d.userid AND p.device_id=d.deviceid").
-		Fields(
-			"CAST(d.userid AS CHAR) user_id",
-			"d.deviceid device_id",
-			"COALESCE(d.platform, '') platform",
-			"COALESCE(d.lang, '') lang",
-			"d.lastseen last_seen",
-			"COALESCE(p.provider, '') push_provider",
-			"COALESCE(p.platform, '') push_platform",
-			"COALESCE(p.app_version, '') push_app_version",
-			"(p.id IS NOT NULL) push_token_configured",
-			"p.updated_at push_updated_at",
-		).
-		Where("d.userid", userId).
-		Page(in.Page, in.PerPage).
-		OrderDesc("d.lastseen")
+func (s *sSysImUser) Ban(ctx context.Context, in *sysin.ImUserGovernanceInp) (res *sysin.ImUserCommandModel, err error) {
+	return s.publishGovernanceCommand(ctx, in, consts.ImAdminCommandUserBan)
+}

-	if err = mod.ScanAndCount(&list, &totalCount, false); err != nil {
-		return nil, 0, gerror.Wrap(err, "获取IM用户设备列表失败，请稍后重试！")
+func (s *sSysImUser) Unban(ctx context.Context, in *sysin.ImUserGovernanceInp) (res *sysin.ImUserCommandModel, err error) {
+	return s.publishGovernanceCommand(ctx, in, consts.ImAdminCommandUserUnban)
+}
+
+func (s *sSysImUser) Freeze(ctx context.Context, in *sysin.ImUserGovernanceInp) (res *sysin.ImUserCommandModel, err error) {
+	return s.publishGovernanceCommand(ctx, in, consts.ImAdminCommandUserFreeze)
+}
+
+func (s *sSysImUser) Unfreeze(ctx context.Context, in *sysin.ImUserGovernanceInp) (res *sysin.ImUserCommandModel, err error) {
+	return s.publishGovernanceCommand(ctx, in, consts.ImAdminCommandUserUnfreeze)
+}
+
+func (s *sSysImUser) DeviceKick(ctx context.Context, in *sysin.ImUserGovernanceInp) (res *sysin.ImUserCommandModel, err error) {
+	if err = in.FilterDevice(ctx); err != nil {
+		return nil, err
 	}
-	for _, item := range list {
-		item.TenantId = tenant.Id
-		item.TenantName = tenant.Name
+	return s.publishGovernanceCommand(ctx, in, consts.ImAdminCommandUserDeviceKick)
+}
+
+func (s *sSysImUser) LogoutAll(ctx context.Context, in *sysin.ImUserGovernanceInp) (res *sysin.ImUserCommandModel, err error) {
+	return s.publishGovernanceCommand(ctx, in, consts.ImAdminCommandUserLogoutAll)
+}
+
+func (s *sSysImUser) AccountCancel(ctx context.Context, in *sysin.ImUserGovernanceInp) (res *sysin.ImUserCommandModel, err error) {
+	return s.publishGovernanceCommand(ctx, in, consts.ImAdminCommandUserCancel)
+}
+
+func (s *sSysImUser) AccountDelete(ctx context.Context, in *sysin.ImUserGovernanceInp) (res *sysin.ImUserCommandModel, err error) {
+	in.Hard = true
+	return s.publishGovernanceCommand(ctx, in, consts.ImAdminCommandUserDelete)
+}
+
+func (s *sSysImUser) publishGovernanceCommand(ctx context.Context, in *sysin.ImUserGovernanceInp, eventType string) (*sysin.ImUserCommandModel, error) {
+	if err := in.Filter(ctx); err != nil {
+		return nil, err
 	}
-	return
+	_, tenant, err := imtenant.DB(ctx, in.TenantId)
+	if err != nil {
+		return nil, err
+	}
+	user := contexts.GetUser(ctx)
+	actorName := ""
+	if user != nil {
+		actorName = user.RealName
+	}
+	payload := map[string]any{
+		"actor_user_id":  strconv.FormatInt(in.ActorUserId, 10),
+		"user_id":        strconv.FormatInt(in.UserId, 10),
+		"target_user_id": strconv.FormatInt(in.UserId, 10),
+		"reason":         in.Reason,
+	}
+	if in.DeviceId != "" {
+		payload["device_id"] = in.DeviceId
+	}
+	if eventType == consts.ImAdminCommandUserCancel || eventType == consts.ImAdminCommandUserDelete {
+		payload["hard"] = in.Hard
+	}
+	command, err := imkafka.PublishAdminCommand(ctx, imkafka.AdminCommandInput{
+		TenantID:   tenant.Id,
+		TenantCode: tenant.Code,
+		EventType:  eventType,
+		BizType:    "user",
+		BizID:      strconv.FormatInt(in.UserId, 10),
+		Actor: &imkafka.Actor{
+			Type:        "admin",
+			ID:          strconv.FormatInt(contexts.GetUserId(ctx), 10),
+			DisplayName: actorName,
+		},
+		Payload:   payload,
+		CreatedBy: contexts.GetUserId(ctx),
+	})
+	if err != nil {
+		return nil, err
+	}
+	return &sysin.ImUserCommandModel{
+		CommandID:      command.CommandID,
+		EventID:        command.EventID,
+		Status:         command.Status,
+		TenantID:       command.TenantID,
+		TenantCode:     command.TenantCode,
+		EventType:      command.EventType,
+		IdempotencyKey: command.IdempotencyKey,
+	}, nil
 }

 func updateIMUserStatus(ctx context.Context, db gdb.DB, scope imAdminScope, userId int64, state int) error {