Files
iotlcm/admin-svr/server/internal/logic/packagecustomer/publisher.go
T

62 lines
2.1 KiB
Go

package packagecustomer
import (
"context"
"strings"
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/consts"
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/library/rabbitmq"
"git.5iot.com/iotlcm-public/iotlcm-admin-svr/internal/model"
"github.com/gogf/gf/v2/errors/gerror"
"github.com/gogf/gf/v2/frame/g"
)
func PublishSync(ctx context.Context, message model.PackageCustomerSyncMessage) error {
message.TenantId = strings.TrimSpace(message.TenantId)
message.Trigger = strings.TrimSpace(message.Trigger)
message.SourceBatchNo = strings.TrimSpace(message.SourceBatchNo)
if message.TenantId == "" {
return gerror.New("package customer sync tenantId is empty")
}
if message.DeptId <= 0 {
return gerror.New("package customer sync deptId is empty")
}
if message.Trigger == "" {
message.Trigger = "manual"
}
if err := rabbitmq.Publish(ctx, consts.QueuePackageCustomerSync, message); err != nil {
return gerror.Wrapf(err, "publish package customer sync failed, queue:%s", consts.QueuePackageCustomerSync)
}
g.Log().Infof(ctx, "package customer sync published, tenantId:%s, deptId:%d, trigger:%s, sourceBatchNo:%s",
message.TenantId, message.DeptId, message.Trigger, message.SourceBatchNo)
return nil
}
func PublishSyncSafe(ctx context.Context, message model.PackageCustomerSyncMessage) {
if err := PublishSync(ctx, message); err != nil {
g.Log().Warningf(ctx, "package customer sync publish skipped, tenantId:%s, deptId:%d, trigger:%s, sourceBatchNo:%s, err:%+v",
message.TenantId, message.DeptId, message.Trigger, message.SourceBatchNo, err)
}
}
func PublishSyncForDeptsSafe(ctx context.Context, tenantId string, userId int64, deptIds []int64, trigger, sourceBatchNo string) {
seen := make(map[int64]struct{}, len(deptIds))
for _, deptId := range deptIds {
if deptId <= 0 {
continue
}
if _, ok := seen[deptId]; ok {
continue
}
seen[deptId] = struct{}{}
PublishSyncSafe(ctx, model.PackageCustomerSyncMessage{
TenantId: tenantId,
DeptId: deptId,
UserId: userId,
Trigger: trigger,
SourceBatchNo: sourceBatchNo,
})
}
}