method fix
This commit is contained in:
parent
792ce2d08b
commit
77753da0d3
1 changed files with 8 additions and 2 deletions
|
|
@ -32,8 +32,8 @@ func (s *service) FetchTasks(ctx context.Context) (received []structs.Task, err
|
|||
return nil, err
|
||||
}
|
||||
|
||||
for {
|
||||
log.Infof("%v Receiving tasks", pkgLogHeader)
|
||||
for {
|
||||
task, err := stream.Recv()
|
||||
if err == io.EOF {
|
||||
break
|
||||
|
|
@ -47,8 +47,8 @@ func (s *service) FetchTasks(ctx context.Context) (received []structs.Task, err
|
|||
MerchUuid: task.MerchUuid,
|
||||
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
|
||||
}
|
||||
|
|
@ -81,5 +81,11 @@ func (s *service) SendResults(ctx context.Context, results []structs.Result) 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
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue