75 lines
2.5 KiB
Go
75 lines
2.5 KiB
Go
package query
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
|
|
"git.5iot.com/iotlcm-public/iotlcm-openapi/internal/consts"
|
|
"git.5iot.com/iotlcm-public/iotlcm-openapi/internal/dao"
|
|
"git.5iot.com/iotlcm-public/iotlcm-openapi/internal/library/contexts"
|
|
"git.5iot.com/iotlcm-public/iotlcm-openapi/internal/model"
|
|
)
|
|
|
|
type batchStatusRecord struct {
|
|
Id int64 `orm:"id"`
|
|
BatchNo string `orm:"batch_no"`
|
|
BatchAsyncStatus int `orm:"batch_async_status"`
|
|
SourceChannel string `orm:"source_channel"`
|
|
ExecutionMode int `orm:"execution_mode"`
|
|
SourceType int `orm:"source_type"`
|
|
TotalCount int `orm:"total_count"`
|
|
ValidCount int `orm:"valid_count"`
|
|
SuccessCount int `orm:"success_count"`
|
|
FailCount int `orm:"fail_count"`
|
|
RunResult int `orm:"run_result"`
|
|
FailReason string `orm:"fail_reason"`
|
|
}
|
|
|
|
func (s *sQuerySim) QueryBatchStatus(ctx context.Context, in *model.QueryBatchStatusParam) (*model.QueryBatchStatusOutput, error) {
|
|
if in == nil {
|
|
return nil, gerror.New("查询参数不能为空")
|
|
}
|
|
if strings.TrimSpace(in.AppId) == "" {
|
|
return nil, gerror.New("appId不能为空")
|
|
}
|
|
batchNo := strings.TrimSpace(in.BatchNo)
|
|
if batchNo == "" {
|
|
return nil, gerror.New("batchNo不能为空")
|
|
}
|
|
tenantID := strings.TrimSpace(contexts.GetTenantId(ctx))
|
|
deptID := contexts.GetAgentDeptId(ctx)
|
|
if tenantID == "" || deptID <= 0 {
|
|
return nil, gerror.New("当前接口调用方信息不完整")
|
|
}
|
|
columns := dao.BatchMain.Columns()
|
|
var batch batchStatusRecord
|
|
err := dao.BatchMain.Ctx(ctx).
|
|
Where(columns.TenantId, tenantID).
|
|
Where(columns.DeptId, deptID).
|
|
Where(columns.BatchNo, batchNo).
|
|
Where(fmt.Sprintf("NOT (%s = ? AND COALESCE(%s, '') <> '' AND %s <> %s)", columns.BatchType, columns.SourceBatchNo, columns.SourceBatchNo, columns.BatchNo), consts.BatchTypeAllocate).
|
|
Scan(&batch)
|
|
if err != nil {
|
|
return nil, gerror.Wrap(err, "查询批次状态失败")
|
|
}
|
|
if batch.Id == 0 {
|
|
return nil, gerror.New("批次不存在或无权限")
|
|
}
|
|
return &model.QueryBatchStatusOutput{
|
|
BatchNo: batch.BatchNo,
|
|
BatchAsyncStatus: batch.BatchAsyncStatus,
|
|
SourceChannel: batch.SourceChannel,
|
|
ExecutionMode: batch.ExecutionMode,
|
|
SourceType: batch.SourceType,
|
|
TotalCount: batch.TotalCount,
|
|
ValidCount: batch.ValidCount,
|
|
SuccessCount: batch.SuccessCount,
|
|
FailCount: batch.FailCount,
|
|
RunResult: batch.RunResult,
|
|
FailReason: batch.FailReason,
|
|
}, nil
|
|
}
|