forked from TrueCloudLab/frostfs-node
Dmitrii Stepanov
27bdddc48f
Resolve containedctx linter for streamer and remote target Signed-off-by: Dmitrii Stepanov <d.stepanov@yadro.com>
48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
package response
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/util"
|
|
)
|
|
|
|
// ClientMessageStreamer represents client-side message streamer
|
|
// that sets meta values to the response.
|
|
type ClientMessageStreamer struct {
|
|
cfg *cfg
|
|
|
|
send util.RequestMessageWriter
|
|
|
|
close util.ClientStreamCloser
|
|
}
|
|
|
|
// Send calls send method of internal streamer.
|
|
func (s *ClientMessageStreamer) Send(ctx context.Context, req any) error {
|
|
if err := s.send(ctx, req); err != nil {
|
|
return fmt.Errorf("(%T) could not send the request: %w", s, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// CloseAndRecv closes internal stream, receivers the response,
|
|
// sets meta values and returns the result.
|
|
func (s *ClientMessageStreamer) CloseAndRecv(ctx context.Context) (util.ResponseMessage, error) {
|
|
resp, err := s.close(ctx)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("(%T) could not close stream and receive response: %w", s, err)
|
|
}
|
|
|
|
setMeta(resp, s.cfg)
|
|
|
|
return resp, nil
|
|
}
|
|
|
|
// CreateRequestStreamer wraps stream methods and returns ClientMessageStreamer instance.
|
|
func (s *Service) CreateRequestStreamer(sender util.RequestMessageWriter, closer util.ClientStreamCloser) *ClientMessageStreamer {
|
|
return &ClientMessageStreamer{
|
|
cfg: s.cfg,
|
|
send: sender,
|
|
close: closer,
|
|
}
|
|
}
|