forked from TrueCloudLab/frostfs-node
f24daa10ff
Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
package response
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/nspcc-dev/neofs-node/pkg/services/util"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// ServerMessageStreamer represents server-side message streamer
|
|
// that sets meta values to all response messages.
|
|
type ServerMessageStreamer struct {
|
|
cfg *cfg
|
|
|
|
recv util.ResponseMessageReader
|
|
}
|
|
|
|
// Recv calls Recv method of internal streamer, sets response meta
|
|
// values and returns the response.
|
|
func (s *ServerMessageStreamer) Recv() (util.ResponseMessage, error) {
|
|
m, err := s.recv()
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "could not receive response message for signing")
|
|
}
|
|
|
|
setMeta(m, s.cfg)
|
|
|
|
return m, nil
|
|
}
|
|
|
|
// HandleServerStreamRequest builds internal streamer via handlers, wraps it to ServerMessageStreamer and returns the result.
|
|
func (s *Service) HandleServerStreamRequest(ctx context.Context, req interface{}, handler util.ServerStreamHandler) (*ServerMessageStreamer, error) {
|
|
msgRdr, err := handler(ctx, req)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "could not create message reader")
|
|
}
|
|
|
|
return &ServerMessageStreamer{
|
|
cfg: s.cfg,
|
|
recv: msgRdr,
|
|
}, nil
|
|
}
|
|
|
|
func (s *Service) HandleServerStreamRequest_(respWriter util.ResponseMessageWriter) util.ResponseMessageWriter {
|
|
return func(resp util.ResponseMessage) error {
|
|
setMeta(resp, s.cfg)
|
|
|
|
return respWriter(resp)
|
|
}
|
|
}
|