解决调试发现的问题

This commit is contained in:
Sydonian 2024-06-07 16:18:07 +08:00
parent 3e7fee895f
commit 5209146cf1
2 changed files with 11 additions and 2 deletions

View File

@ -40,6 +40,8 @@ func (t *PCMSubmitTask) Execute(task *task.Task[TaskContext], ctx TaskContext, c
}
func (t *PCMSubmitTask) do(taskID string, ctx TaskContext) error {
log := logger.WithType[PCMSubmitTask]("Task")
pcmCli, err := schglb.PCMPool.Acquire()
if err != nil {
return fmt.Errorf("new pcm client: %w", err)
@ -59,14 +61,21 @@ func (t *PCMSubmitTask) do(taskID string, ctx TaskContext) error {
return err
}
log.Infof("pcm submit task success, task id: %v", resp.TaskID)
var prevStatus pcmsdk.TaskStatus
retryCount := 0
for {
tsResp, err := pcmCli.GetTask(pcmsdk.GetTaskReq{
PartID: t.PCMParticipantID,
TaskID: resp.TaskID,
})
if err != nil {
return err
retryCount++
if retryCount > 3 {
return err
}
continue
}
if tsResp.TaskStatus != prevStatus {

View File

@ -23,7 +23,7 @@ func NewStorageLoadPackage(info *exectsk.StorageLoadPackage) *StorageLoadPackage
func (t *StorageLoadPackage) Execute(task *task.Task[TaskContext], ctx TaskContext, complete CompleteFn) {
log := logger.WithType[StorageLoadPackage]("Task")
log.Debugf("begin with %w", logger.FormatStruct(t.StorageLoadPackage))
log.Debugf("begin with %v", logger.FormatStruct(t.StorageLoadPackage))
defer log.Debugf("end")
packagePath, err := t.do(ctx)