2020-08-24 14:07:08 +00:00
|
|
|
package container
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"crypto/ecdsa"
|
|
|
|
|
2023-03-07 13:38:26 +00:00
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/util"
|
2024-11-07 14:32:10 +00:00
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/container"
|
2020-08-24 14:07:08 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type signService struct {
|
2020-08-26 09:43:39 +00:00
|
|
|
sigSvc *util.SignService
|
2020-08-26 09:20:26 +00:00
|
|
|
|
2021-03-15 10:53:08 +00:00
|
|
|
svc Server
|
2020-08-24 14:07:08 +00:00
|
|
|
}
|
|
|
|
|
2021-03-15 10:53:08 +00:00
|
|
|
func NewSignService(key *ecdsa.PrivateKey, svc Server) Server {
|
2020-08-24 14:07:08 +00:00
|
|
|
return &signService{
|
2020-08-26 09:43:39 +00:00
|
|
|
sigSvc: util.NewUnarySignService(key),
|
|
|
|
svc: svc,
|
2020-08-24 14:07:08 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *signService) Put(ctx context.Context, req *container.PutRequest) (*container.PutResponse, error) {
|
2022-12-30 15:26:43 +00:00
|
|
|
if err := s.sigSvc.VerifyRequest(req); err != nil {
|
|
|
|
resp := new(container.PutResponse)
|
2023-07-26 12:47:32 +00:00
|
|
|
return resp, s.sigSvc.SignResponse(resp, err)
|
2020-08-24 14:07:08 +00:00
|
|
|
}
|
2022-12-30 15:26:43 +00:00
|
|
|
resp, err := util.EnsureNonNilResponse(s.svc.Put(ctx, req))
|
2023-07-26 12:47:32 +00:00
|
|
|
return resp, s.sigSvc.SignResponse(resp, err)
|
2020-08-24 14:07:08 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (s *signService) Delete(ctx context.Context, req *container.DeleteRequest) (*container.DeleteResponse, error) {
|
2022-12-30 15:26:43 +00:00
|
|
|
if err := s.sigSvc.VerifyRequest(req); err != nil {
|
|
|
|
resp := new(container.DeleteResponse)
|
2023-07-26 12:47:32 +00:00
|
|
|
return resp, s.sigSvc.SignResponse(resp, err)
|
2020-08-24 14:07:08 +00:00
|
|
|
}
|
2022-12-30 15:26:43 +00:00
|
|
|
resp, err := util.EnsureNonNilResponse(s.svc.Delete(ctx, req))
|
2023-07-26 12:47:32 +00:00
|
|
|
return resp, s.sigSvc.SignResponse(resp, err)
|
2020-08-24 14:07:08 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (s *signService) Get(ctx context.Context, req *container.GetRequest) (*container.GetResponse, error) {
|
2022-12-30 15:26:43 +00:00
|
|
|
if err := s.sigSvc.VerifyRequest(req); err != nil {
|
|
|
|
resp := new(container.GetResponse)
|
2023-07-26 12:47:32 +00:00
|
|
|
return resp, s.sigSvc.SignResponse(resp, err)
|
2020-08-24 14:07:08 +00:00
|
|
|
}
|
2022-12-30 15:26:43 +00:00
|
|
|
resp, err := util.EnsureNonNilResponse(s.svc.Get(ctx, req))
|
2023-07-26 12:47:32 +00:00
|
|
|
return resp, s.sigSvc.SignResponse(resp, err)
|
2020-08-24 14:07:08 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func (s *signService) List(ctx context.Context, req *container.ListRequest) (*container.ListResponse, error) {
|
2022-12-30 15:26:43 +00:00
|
|
|
if err := s.sigSvc.VerifyRequest(req); err != nil {
|
|
|
|
resp := new(container.ListResponse)
|
2023-07-26 12:47:32 +00:00
|
|
|
return resp, s.sigSvc.SignResponse(resp, err)
|
2020-08-24 14:07:08 +00:00
|
|
|
}
|
2022-12-30 15:26:43 +00:00
|
|
|
resp, err := util.EnsureNonNilResponse(s.svc.List(ctx, req))
|
2023-07-26 12:47:32 +00:00
|
|
|
return resp, s.sigSvc.SignResponse(resp, err)
|
2020-08-24 14:07:08 +00:00
|
|
|
}
|
2024-10-28 15:10:07 +00:00
|
|
|
|
|
|
|
func (s *signService) ListStream(req *container.ListStreamRequest, stream ListStream) error {
|
|
|
|
if err := s.sigSvc.VerifyRequest(req); err != nil {
|
|
|
|
resp := new(container.ListStreamResponse)
|
|
|
|
_ = s.sigSvc.SignResponse(resp, err)
|
|
|
|
return stream.Send(resp)
|
|
|
|
}
|
|
|
|
|
|
|
|
ss := &listStreamSigner{
|
|
|
|
ListStream: stream,
|
|
|
|
sigSvc: s.sigSvc,
|
|
|
|
}
|
|
|
|
err := s.svc.ListStream(req, ss)
|
|
|
|
if err != nil || !ss.nonEmptyResp {
|
|
|
|
return ss.send(new(container.ListStreamResponse), err)
|
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
type listStreamSigner struct {
|
|
|
|
ListStream
|
|
|
|
sigSvc *util.SignService
|
|
|
|
|
|
|
|
nonEmptyResp bool // set on first Send call
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *listStreamSigner) Send(resp *container.ListStreamResponse) error {
|
|
|
|
s.nonEmptyResp = true
|
|
|
|
return s.send(resp, nil)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (s *listStreamSigner) send(resp *container.ListStreamResponse, err error) error {
|
|
|
|
if err := s.sigSvc.SignResponse(resp, err); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
return s.ListStream.Send(resp)
|
|
|
|
}
|