MiFaZhan
62 lines
2.1 KiB
Go
62 lines
2.1 KiB
Go
package packagecustomer
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
|
|
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/consts"
|
|
mq "git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/library/rabbitmq"
|
|
"git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/model"
|
|
)
|
|
|
|
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 = "worker"
|
|
}
|
|
if err := mq.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,
|
|
})
|
|
}
|
|
}
|