X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=binapi%2Fwireguard%2Fwireguard_rpc.ba.go;h=9d3a02435eaaba4c5cb56ce04330e4b71bc28bb5;hb=c09ee3241377aae2530a73d48c4e20641d76d0ee;hp=77b9097f77e50c6d0b334700fbdbeaeab0dcc271;hpb=debc52dea8a81417bb08ca5bb934c7876b6d65e0;p=govpp.git

diff --git a/binapi/wireguard/wireguard_rpc.ba.go b/binapi/wireguard/wireguard_rpc.ba.go
index 77b9097..9d3a024 100644
--- a/binapi/wireguard/wireguard_rpc.ba.go
+++ b/binapi/wireguard/wireguard_rpc.ba.go
@@ -80,6 +80,10 @@ func (c *serviceClient_WireguardInterfaceDumpClient) Recv() (*WireguardInterface
 	case *WireguardInterfaceDetails:
 		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)
@@ -137,6 +141,10 @@ func (c *serviceClient_WireguardPeersDumpClient) Recv() (*WireguardPeersDetails,
 	case *WireguardPeersDetails:
 		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)