TrueCloudLab/frostfs-node
21
1
Fork 27
Code Issues 108 Pull requests 3 Projects Releases 1 Activity Actions

[#13] transport/object: Fix io.EOF check in Object gRPC handlers

Browse source 
Signed-off-by: Leonard Lyubich <leonard@nspcc.ru>
This commit is contained in:
Leonard Lyubich 2020-08-25 16:35:36 +03:00 committed by Alex Vanin
parent da1961cdf9
commit a0aa175fcc
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
pkg/network/transport/object/grpc/service.go
View file

@ -6,6 +6,7 @@ import (
"github.com/nspcc-dev/neofs-api-go/v2/object"
objectGRPC "github.com/nspcc-dev/neofs-api-go/v2/object/grpc"
"github.com/pkg/errors"
)
// Server wraps NeoFS API Object service and
@ -33,7 +34,7 @@ func (s *Server) Get(req *objectGRPC.GetRequest, gStream objectGRPC.ObjectServic
for {
r, err := stream.Recv()
if err != nil {
if err == io.EOF {
if errors.Is(errors.Cause(err), io.EOF) {
return nil
}
@ -62,7 +63,7 @@ func (s *Server) Put(gStream objectGRPC.ObjectService_PutServer) error {
}
}
if err == io.EOF {
if errors.Is(errors.Cause(err), io.EOF) {
resp, err := stream.CloseAndRecv()
if err != nil {
return err
@ -109,7 +110,7 @@ func (s *Server) Search(req *objectGRPC.SearchRequest, gStream objectGRPC.Object
for {
r, err := stream.Recv()
if err != nil {
if err == io.EOF {
if errors.Is(errors.Cause(err), io.EOF) {
return nil
}
@ -134,7 +135,7 @@ func (s *Server) GetRange(req *objectGRPC.GetRangeRequest, gStream objectGRPC.Ob
for {
r, err := stream.Recv()
if err != nil {
if err == io.EOF {
if errors.Is(errors.Cause(err), io.EOF) {
return nil
}