Refactoring and fixes

[govpp.git] / binapi / pnat / pnat_rpc.ba.go

diff --git a/binapi/pnat/pnat_rpc.ba.go b/binapi/pnat/pnat_rpc.ba.go
index 036861a..958bc50 100644 (file)
--- a/binapi/pnat/pnat_rpc.ba.go
+++ b/binapi/pnat/pnat_rpc.ba.go
@@ -94,6 +94,10 @@ func (c *serviceClient_PnatBindingsGetClient) Recv() (*PnatBindingsDetails, erro
        case *PnatBindingsDetails:
                return m, nil
        case *PnatBindingsGetReply:
+               err = c.Stream.Close()
+               if err != nil {
+                       return nil, err
+               }
                return nil, io.EOF
        default:
                return nil, fmt.Errorf("unexpected message: %T %v", m, m)
@@ -130,6 +134,10 @@ func (c *serviceClient_PnatInterfacesGetClient) Recv() (*PnatInterfacesDetails,
        case *PnatInterfacesDetails:
                return m, nil
        case *PnatInterfacesGetReply:
+               err = c.Stream.Close()
+               if err != nil {
+                       return nil, err
+               }
                return nil, io.EOF
        default:
                return nil, fmt.Errorf("unexpected message: %T %v", m, m)