diff options
| author |   mhalaj1 <matus.halaj@pantheon.tech> | 2021-08-26 20:15:08 +0200 |
|---|---|---|
| committer | mhalaj1 <matus.halaj@pantheon.tech> | 2021-09-07 15:24:53 +0200 |
| commit | c09ee3241377aae2530a73d48c4e20641d76d0ee (patch) | |
| tree | 219fd1f7d9ba595f0f4c8dac9796bc76ce3556fc /binapi/igmp/igmp_rpc.ba.go | |
| parent | debc52dea8a81417bb08ca5bb934c7876b6d65e0 (diff) | |
Refactoring and fixes
* refactor creation of new channel
* add missing closing of created streams
* correct documentation regarding thread safety of stream
Signed-off-by: mhalaj1 <matus.halaj@pantheon.tech>
Change-Id: Ic601efff298fcbdecaafab83fa236253af69de21
Diffstat (limited to 'binapi/igmp/igmp_rpc.ba.go')
| -rw-r--r-- | binapi/igmp/igmp_rpc.ba.go | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/binapi/igmp/igmp_rpc.ba.go b/binapi/igmp/igmp_rpc.ba.go index 3cda04c..641e2db 100644 --- a/binapi/igmp/igmp_rpc.ba.go +++ b/binapi/igmp/igmp_rpc.ba.go @@ -74,6 +74,10 @@ func (c *serviceClient_IgmpDumpClient) Recv() (*IgmpDetails, error) { case *IgmpDetails: 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) @@ -122,6 +126,10 @@ func (c *serviceClient_IgmpGroupPrefixDumpClient) Recv() (*IgmpGroupPrefixDetail case *IgmpGroupPrefixDetails: 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) |