Refactoring and fixes

[govpp.git] / binapi / teib / teib_rpc.ba.go

diff --git a/binapi/teib/teib_rpc.ba.go b/binapi/teib/teib_rpc.ba.go
index 4803c2d..74d71b4 100644 (file)
--- a/binapi/teib/teib_rpc.ba.go
+++ b/binapi/teib/teib_rpc.ba.go
@@ -5,12 +5,13 @@ package teib
 import (
        "context"
        "fmt"
+       "io"
+
        api "git.fd.io/govpp.git/api"
        vpe "git.fd.io/govpp.git/binapi/vpe"
-       "io"
 )
 
-// RPCService defines RPC service  teib.
+// RPCService defines RPC service teib.
 type RPCService interface {
        TeibDump(ctx context.Context, in *TeibDump) (RPCService_TeibDumpClient, error)
        TeibEntryAddDel(ctx context.Context, in *TeibEntryAddDel) (*TeibEntryAddDelReply, error)
@@ -57,6 +58,10 @@ func (c *serviceClient_TeibDumpClient) Recv() (*TeibDetails, error) {
        case *TeibDetails:
                return m, nil
        case *vpe.ControlPingReply:
+               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)
@@ -69,5 +74,5 @@ func (c *serviceClient) TeibEntryAddDel(ctx context.Context, in *TeibEntryAddDel
        if err != nil {
                return nil, err
        }
-       return out, nil
+       return out, api.RetvalToVPPApiError(out.Retval)
 }