feat: expand IM admin controls

2026-08-21 15:09:39 +08:00
parent c3676f9744
commit 543c7b624e
29 changed files with 1308 additions and 39 deletions
@@ -7,10 +7,13 @@ package sys

 import (
 	"context"
+	"fmt"
+	"hotgo/internal/consts"
 	"hotgo/internal/dao"
 	"hotgo/internal/library/contexts"
 	"hotgo/internal/library/hgorm/handler"
 	"hotgo/internal/library/hgorm/hook"
+	"hotgo/internal/library/imtenant"
 	"hotgo/internal/model/entity"
 	"hotgo/internal/model/input/sysin"
 	"hotgo/internal/service"
@@ -272,3 +275,137 @@ func (s *sSysImTenant) Status(ctx context.Context, in *sysin.ImTenantStatusInp)
 	}
 	return
 }
+
+func (s *sSysImTenant) Usage(ctx context.Context, in *sysin.ImTenantUsageInp) (res *sysin.ImTenantUsageModel, err error) {
+	if err = in.Filter(ctx); err != nil {
+		return
+	}
+
+	var tenant *entity.ImTenant
+	if err = s.Model(ctx).WherePri(in.Id).Scan(&tenant); err != nil {
+		return nil, gerror.Wrap(err, "获取IM租户信息失败，请稍后重试！")
+	}
+	if tenant == nil || tenant.Id <= 0 {
+		return nil, gerror.New("IM租户不存在")
+	}
+
+	res = &sysin.ImTenantUsageModel{
+		Id:              tenant.Id,
+		Code:            tenant.Code,
+		Name:            tenant.Name,
+		DbName:          tenant.DbName,
+		DbDsnConfigured: tenant.DbDsn != "",
+		Warnings:        []string{},
+	}
+
+	db, _, dbErr := imtenant.DB(ctx, tenant.Id)
+	if dbErr != nil {
+		res.DbReachable = false
+		res.DbCheckError = dbErr.Error()
+		return res, nil
+	}
+	if _, pingErr := db.GetValue(ctx, "SELECT 1"); pingErr != nil {
+		res.DbReachable = false
+		res.DbCheckError = pingErr.Error()
+		return res, nil
+	}
+	res.DbReachable = true
+
+	res.SchemaVersion = s.queryScalarString(ctx, db, "kvmeta.version", "SELECT value FROM kvmeta WHERE key_='version' LIMIT 1", &res.Warnings)
+	res.UserCount = s.queryScalarInt64(ctx, db, "im_users.total", "SELECT COUNT(*) FROM im_users", &res.Warnings)
+	res.ActiveUserCount = s.queryScalarInt64(ctx, db, "im_users.active", "SELECT COUNT(*) FROM im_users WHERE state=0", &res.Warnings)
+	res.GroupCount = s.queryScalarInt64(ctx, db, "im_group_settings", "SELECT COUNT(*) FROM im_group_settings WHERE group_status<>'dissolved'", &res.Warnings)
+	res.DeviceCount = s.queryScalarInt64(ctx, db, "im_devices", "SELECT COUNT(*) FROM im_devices", &res.Warnings)
+	res.PushTokenCount = s.queryScalarInt64(ctx, db, "im_device_push_tokens", "SELECT COUNT(*) FROM im_device_push_tokens", &res.Warnings)
+	res.InviteUseCount = s.queryScalarInt64(ctx, db, "im_registration_invite_uses", "SELECT COUNT(*) FROM im_registration_invite_uses", &res.Warnings)
+	res.OutboxPendingCount = s.queryScalarInt64(ctx, db, "im_business_events_outbox", "SELECT COUNT(*) FROM im_business_events_outbox WHERE status='pending'", &res.Warnings)
+	return
+}
+
+func (s *sSysImTenant) queryScalarInt64(ctx context.Context, db gdb.DB, label, sql string, warnings *[]string) int64 {
+	value, err := db.GetValue(ctx, sql)
+	if err != nil {
+		*warnings = append(*warnings, label+": "+err.Error())
+		return 0
+	}
+	return value.Int64()
+}
+
+func (s *sSysImTenant) queryScalarString(ctx context.Context, db gdb.DB, label, sql string, warnings *[]string) string {
+	value, err := db.GetValue(ctx, sql)
+	if err != nil {
+		*warnings = append(*warnings, label+": "+err.Error())
+		return ""
+	}
+	return value.String()
+}
+
+func (s *sSysImTenant) DeletionCheck(ctx context.Context, in *sysin.ImTenantDeletionCheckInp) (res *sysin.ImTenantDeletionCheckModel, err error) {
+	if err = in.Filter(ctx); err != nil {
+		return
+	}
+	usage, err := s.Usage(ctx, &sysin.ImTenantUsageInp{Id: in.Id})
+	if err != nil {
+		return nil, err
+	}
+	res = &sysin.ImTenantDeletionCheckModel{
+		Id:            usage.Id,
+		Code:          usage.Code,
+		Name:          usage.Name,
+		Blockers:      []string{},
+		Warnings:      usage.Warnings,
+		UserCount:     usage.UserCount,
+		GroupCount:    usage.GroupCount,
+		PendingOutbox: usage.OutboxPendingCount,
+	}
+	if !usage.DbReachable {
+		res.Blockers = append(res.Blockers, "租户业务库不可达："+usage.DbCheckError)
+	}
+	if usage.UserCount > 0 {
+		res.Blockers = append(res.Blockers, fmt.Sprintf("租户仍有 %d 个用户", usage.UserCount))
+	}
+	if usage.GroupCount > 0 {
+		res.Blockers = append(res.Blockers, fmt.Sprintf("租户仍有 %d 个群", usage.GroupCount))
+	}
+	if usage.OutboxPendingCount > 0 {
+		res.Blockers = append(res.Blockers, fmt.Sprintf("租户仍有 %d 条待处理 Outbox", usage.OutboxPendingCount))
+	}
+	res.CanEnterDeleting = len(res.Blockers) == 0
+	return
+}
+
+func (s *sSysImTenant) EnterDeleting(ctx context.Context, in *sysin.ImTenantLifecycleInp) (err error) {
+	if err = in.Filter(ctx); err != nil {
+		return
+	}
+	check, err := s.DeletionCheck(ctx, &sysin.ImTenantDeletionCheckInp{Id: in.Id})
+	if err != nil {
+		return err
+	}
+	if !check.CanEnterDeleting {
+		return gerror.New("租户未通过删除前检查，不能进入删除流程")
+	}
+	if _, err = s.Model(ctx).WherePri(in.Id).Data(g.Map{
+		dao.ImTenant.Columns().State:     consts.ImTenantStateDeleting,
+		dao.ImTenant.Columns().UpdatedBy: contexts.GetUserId(ctx),
+	}).Update(); err != nil {
+		return gerror.Wrap(err, "设置IM租户删除流程失败，请稍后重试！")
+	}
+	return
+}
+
+func (s *sSysImTenant) Restore(ctx context.Context, in *sysin.ImTenantLifecycleInp) (err error) {
+	if err = in.Filter(ctx); err != nil {
+		return
+	}
+	if _, err = s.Model(ctx).
+		WherePri(in.Id).
+		WhereIn(dao.ImTenant.Columns().State, []int{consts.ImTenantStateSuspended, consts.ImTenantStateDeleting}).
+		Data(g.Map{
+			dao.ImTenant.Columns().State:     consts.ImTenantStateActive,
+			dao.ImTenant.Columns().UpdatedBy: contexts.GetUserId(ctx),
+		}).Update(); err != nil {
+		return gerror.Wrap(err, "恢复IM租户失败，请稍后重试！")
+	}
+	return
+}