MiFaZhan
88dfe5c640
iotlcm-openapi：对外 OpenAPI 网关（V3 接口 + 旧版兼容接口）。
- Go module：iotlcm-openapi → git.5iot.com/iotlcm-public/iotlcm-openapi（178 文件）
- 凭据与地址脱敏：JWT signingKey、MySQL DSN、Redis 口令、
RabbitMQ DSN 全部改为 ${*} 占位；私网 IP → 文档地址段
- 7 处真实手机号 → 合法 11 位编造号
- 启动 banner 统一 IOTLCM 方块字 + 社区惯例署名
- 移除内部资产：CI 触发器
- Dockerfile 改用公共基础镜像，移除 GOPRIVATE 内网设置
- README 重写（原为 GoFrame 模板占位文案），补 Apache-2.0 LICENSE，
LF 行尾 + .gitattributes
67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
package common
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"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/model/entity"
|
|
)
|
|
|
|
const (
|
|
singleBatchWaitTimeout = 60 * time.Second
|
|
singleBatchPollInterval = 500 * time.Millisecond
|
|
)
|
|
|
|
func (s *sCommon) waitBatchCompletion(ctx context.Context, tenantID string, deptID int64, batchNo string) (int, error) {
|
|
deadline := time.Now().Add(singleBatchWaitTimeout)
|
|
for {
|
|
status, err := s.loadBatchAsyncStatus(ctx, tenantID, deptID, batchNo)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
switch status {
|
|
case consts.BatchAsyncStatusSuccess, consts.BatchAsyncStatusFailed, consts.BatchAsyncStatusPartialSuccess:
|
|
return status, nil
|
|
}
|
|
|
|
remaining := time.Until(deadline)
|
|
if remaining <= 0 {
|
|
return 0, gerror.Newf("batch %s execution timed out; query batch status for the result", batchNo)
|
|
}
|
|
wait := singleBatchPollInterval
|
|
if remaining < wait {
|
|
wait = remaining
|
|
}
|
|
timer := time.NewTimer(wait)
|
|
select {
|
|
case <-ctx.Done():
|
|
if !timer.Stop() {
|
|
<-timer.C
|
|
}
|
|
return 0, gerror.Wrap(ctx.Err(), "request canceled while waiting for batch completion")
|
|
case <-timer.C:
|
|
}
|
|
}
|
|
}
|
|
|
|
func (s *sCommon) loadBatchAsyncStatus(ctx context.Context, tenantID string, deptID int64, batchNo string) (int, error) {
|
|
columns := dao.BatchMain.Columns()
|
|
var batch entity.BatchMain
|
|
err := dao.BatchMain.Ctx(ctx).
|
|
Where(columns.TenantId, tenantID).
|
|
Where(columns.DeptId, deptID).
|
|
Where(columns.BatchNo, batchNo).
|
|
Scan(&batch)
|
|
if err != nil {
|
|
return 0, gerror.Wrap(err, "failed to query batch execution status")
|
|
}
|
|
if batch.Id == 0 {
|
|
return 0, gerror.Newf("batch %s does not exist or is not accessible", batchNo)
|
|
}
|
|
return batch.BatchAsyncStatus, nil
|
|
}
|