StreamWriter: Close head writer (#1347)

The head writer was not being closed which would leak the handleRequests go routine. This PR fixes it by closing the head writer.
dgraph-io · Oct 2, 2020 · a958294 · a958294
1 parent 90daf43
commit a958294
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/stream_writer.go b/stream_writer.go
@@ -223,6 +223,9 @@ func (sw *StreamWriter) Flush() error {
 		y.ValueStruct{Value: data}); err != nil {
 		return err
 	}
+
+	headWriter.closer.SignalAndWait()
+
 	if err := headWriter.Done(); err != nil {
 		return err
 	}