X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=binapigen%2Fgen_rpc.go;h=fa123f085b6faa9d5c7b319ae3b35487012c071c;hb=c09ee3241377aae2530a73d48c4e20641d76d0ee;hp=a5974c398878850a691dc00b4b9eae086cf35d08;hpb=debc52dea8a81417bb08ca5bb934c7876b6d65e0;p=govpp.git

diff --git a/binapigen/gen_rpc.go b/binapigen/gen_rpc.go
index a5974c3..fa123f0 100644
--- 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("}")