merch-tracker-v2/task-processor
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

method fix

Browse source
This commit is contained in:
nquidox 2026-04-02 18:01:12 +03:00
parent 792ce2d08b
commit 77753da0d3
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
internal/taskAgent/service.go
View file

@ -32,8 +32,8 @@ func (s *service) FetchTasks(ctx context.Context) (received []structs.Task, err
return nil, err return nil, err
} }
log.Infof("%v Receiving tasks", pkgLogHeader)
for { for {
log.Infof("%v Receiving tasks", pkgLogHeader)
task, err := stream.Recv() task, err := stream.Recv()
if err == io.EOF { if err == io.EOF {
break break
@ -47,8 +47,8 @@ func (s *service) FetchTasks(ctx context.Context) (received []structs.Task, err
MerchUuid: task.MerchUuid, MerchUuid: task.MerchUuid,
Origins: task.Origins, Origins: task.Origins,
}) })
log.WithField("Count", len(received)).Infof("%v End receiving tasks", pkgLogHeader)
} }
log.WithField("Count", len(received)).Infof("%v End receiving tasks", pkgLogHeader)
return received, nil return received, nil
} }
@ -81,5 +81,11 @@ func (s *service) SendResults(ctx context.Context, results []structs.Result) err
return err return err
} }
if _, err = stream.CloseAndRecv(); err != nil {
log.WithError(err).Errorf("%v Failed to receive server response", pkgLogHeader)
return err
}
log.Debugf("%v End send results", pkgLogHeader)
return nil return nil
} }