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 · Jun 1, 2020 · a7e239e · a7e239e
1 parent da80eb9
commit a7e239e
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
 	}