im_manager/server/internal/logic/sys/im_outbox.go

// Package sys
// @Link  https://github.com/bufanyun/hotgo
// @Copyright  Copyright (c) 2026 HotGo
// @Author  Ms <133814250@qq.com>
// @License  https://github.com/bufanyun/hotgo/blob/master/LICENSE
package sys

import (
	"context"
	"hotgo/internal/library/imtenant"
	"hotgo/internal/model/entity"
	"hotgo/internal/model/input/sysin"
	"hotgo/internal/service"

	"github.com/gogf/gf/v2/database/gdb"
	"github.com/gogf/gf/v2/errors/gerror"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/gtime"
)

type sSysImOutbox struct{}

func NewSysImOutbox() *sSysImOutbox {
	return &sSysImOutbox{}
}

func init() {
	service.RegisterSysImOutbox(NewSysImOutbox())
}

func (s *sSysImOutbox) List(ctx context.Context, in *sysin.ImOutboxListInp) (list []*sysin.ImOutboxListModel, totalCount int, err error) {
	if err = in.Filter(ctx); err != nil {
		return
	}
	db, tenant, err := s.tenantDB(ctx, in.TenantId)
	if err != nil {
		return
	}

	mod := s.model(ctx, db, in.Source)
	mod = s.applyWhere(mod, in)
	mod = mod.Page(in.Page, in.PerPage).OrderDesc("id")
	if err = mod.ScanAndCount(&list, &totalCount, false); err != nil {
		err = gerror.Wrap(err, "获取IM Outbox列表失败，请稍后重试！")
		return
	}
	for _, item := range list {
		item.TenantId = tenant.Id
		item.TenantName = tenant.Name
	}
	return
}

func (s *sSysImOutbox) View(ctx context.Context, in *sysin.ImOutboxViewInp) (res *sysin.ImOutboxViewModel, err error) {
	if err = in.Filter(ctx); err != nil {
		return
	}
	db, tenant, err := s.tenantDB(ctx, in.TenantId)
	if err != nil {
		return nil, err
	}
	if err = s.model(ctx, db, in.Source).Where("id", in.Id).Scan(&res); err != nil {
		return nil, gerror.Wrap(err, "获取IM Outbox详情失败，请稍后重试！")
	}
	if res == nil {
		return nil, nil
	}
	res.TenantId = tenant.Id
	res.TenantName = tenant.Name
	return
}

func (s *sSysImOutbox) Retry(ctx context.Context, in *sysin.ImOutboxActionInp) (err error) {
	if err = in.Filter(ctx); err != nil {
		return
	}
	return s.updateStatus(ctx, in, sysin.ImOutboxStatusPending)
}

func (s *sSysImOutbox) Dead(ctx context.Context, in *sysin.ImOutboxActionInp) (err error) {
	if err = in.Filter(ctx); err != nil {
		return
	}
	return s.updateStatus(ctx, in, sysin.ImOutboxStatusDead)
}

func (s *sSysImOutbox) tenantDB(ctx context.Context, selectedTenantId int64) (db gdb.DB, tenant *entity.ImTenant, err error) {
	scope, err := resolveImAdminScope(ctx, selectedTenantId)
	if err != nil {
		return nil, nil, err
	}
	return imtenant.DB(ctx, scope.TenantId)
}

func (s *sSysImOutbox) model(ctx context.Context, db gdb.DB, source string) *gdb.Model {
	if source == sysin.ImOutboxSourceGroup {
		return db.Model("im_group_events_outbox").Safe().Ctx(ctx).Fields(
			"'group' source",
			"id",
			"event_id",
			"'topic' target_type",
			"topic target",
			"actor_user_id",
			"event_type",
			"CAST(payload AS CHAR) payload",
			"status",
			"attempts",
			"next_retry_at",
			"delivered_at",
			"created_at",
			"updated_at",
		)
	}
	return db.Model("im_business_events_outbox").Safe().Ctx(ctx).Fields(
		"'business' source",
		"id",
		"event_id",
		"target_type",
		"target",
		"actor_user_id",
		"event_type",
		"CAST(payload AS CHAR) payload",
		"status",
		"attempts",
		"next_retry_at",
		"delivered_at",
		"created_at",
		"updated_at",
	)
}

func (s *sSysImOutbox) applyWhere(mod *gdb.Model, in *sysin.ImOutboxListInp) *gdb.Model {
	if in.EventId != "" {
		mod = mod.WhereLike("event_id", "%"+in.EventId+"%")
	}
	if in.EventType != "" {
		mod = mod.Where("event_type", in.EventType)
	}
	if in.TargetType != "" && in.Source == sysin.ImOutboxSourceBusiness {
		mod = mod.Where("target_type", in.TargetType)
	}
	if in.Target != "" {
		if in.Source == sysin.ImOutboxSourceGroup {
			mod = mod.WhereLike("topic", "%"+in.Target+"%")
		} else {
			mod = mod.WhereLike("target", "%"+in.Target+"%")
		}
	}
	if in.Status != "" {
		mod = mod.Where("status", in.Status)
	}
	if len(in.CreatedAt) == 2 {
		mod = mod.WhereBetween("created_at", in.CreatedAt[0], in.CreatedAt[1])
	}
	return mod
}

func (s *sSysImOutbox) updateStatus(ctx context.Context, in *sysin.ImOutboxActionInp, status string) (err error) {
	db, _, err := s.tenantDB(ctx, in.TenantId)
	if err != nil {
		return err
	}
	table := "im_business_events_outbox"
	if in.Source == sysin.ImOutboxSourceGroup {
		table = "im_group_events_outbox"
	}
	current, err := db.Model(table).Safe().Ctx(ctx).Fields("status").Where("id", in.Id).Value()
	if err != nil {
		return gerror.Wrap(err, "获取IM Outbox事件失败，请稍后重试！")
	}
	if current.IsNil() {
		return gerror.New("IM Outbox事件不存在")
	}
	if current.String() == sysin.ImOutboxStatusDelivered {
		return gerror.New("已投递事件不能重新处置")
	}

	data := g.Map{
		"status":     status,
		"updated_at": gtime.Now(),
	}
	if status == sysin.ImOutboxStatusPending {
		data["next_retry_at"] = nil
		data["delivered_at"] = nil
	}
	if _, err = db.Model(table).Safe().Ctx(ctx).Where("id", in.Id).Data(data).Update(); err != nil {
		return gerror.Wrap(err, "更新IM Outbox事件失败，请稍后重试！")
	}
	return
}