Refactoring and fixes

[govpp.git] / binapigen / gen_rpc.go

diff --git a/binapigen/gen_rpc.go b/binapigen/gen_rpc.go
index a5974c3..fa123f0 100644 (file)
--- a/binapigen/gen_rpc.go
+++ b/binapigen/gen_rpc.go
@@ -152,6 +152,8 @@ func genService(g *GenFile, svc *Service) {
                        g.P("   case *", msgDetails.GoIdent, ":")
                        g.P("           return m, nil")
                        g.P("   case *", msgReply.GoIdent, ":")
+                       g.P("           err = c.Stream.Close()")
+                       g.P("           if err != nil { return nil, err }")
                        g.P("           return nil, ", ioPkg.Ident("EOF"))
                        g.P("   default:")
                        g.P("           return nil, ", fmtPkg.Ident("Errorf"), "(\"unexpected message: %T %v\", m, m)")
@@ -174,6 +176,8 @@ func genService(g *GenFile, svc *Service) {
                        g.P("if err != nil { return err }")
                        g.P("err = stream.SendMsg(in)")
                        g.P("if err != nil { return err }")
+                       g.P("err = stream.Close()")
+                       g.P("if err != nil { return err }")
                        g.P("return nil")
                }
                g.P("}")