pgwire: fix memory account leak

Want unique_ptr. Release note: None
cockroachdb · Feb 7, 2018 · f86d95e · f86d95e
1 parent 533e45e
commit f86d95e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkg/sql/pgwire/v3.go b/pkg/sql/pgwire/v3.go
@@ -337,10 +337,12 @@ func (c *v3Conn) serve(ctx context.Context, draining func() bool, reserved mon.B
 		c.writeBuf.writeTerminatedString(key)
 		c.writeBuf.writeTerminatedString(value)
 		if err := c.writeBuf.finishMsg(c.wr); err != nil {
+			reserved.Close(ctx)
 			return err
 		}
 	}
 	if err := c.wr.Flush(); err != nil {
+		reserved.Close(ctx)
 		return err
 	}