MiFaZhan
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
|
|
}
|