Fix channel buffer size in namespace datastore to use KeysOnlyBufSize

0353c0ca · Kevin Atkinson · dfb94c6d · 0353c0ca
Commit 0353c0ca authored Nov 13, 2016 by Kevin Atkinson
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 1 deletion

namespace/namespace.go namespace/namespace.go +7 -1

No files found.
--- a/namespace/namespace.go
+++ b/namespace/namespace.go
@@ -58,7 +58,13 @@ func (d *datastore) Query(q dsq.Query) (dsq.Results, error) {
 		return nil, err
 	}

-	ch := make(chan dsq.Result)
+	var ch chan dsq.Result
+	if q.KeysOnly {
+		ch = make(chan dsq.Result, dsq.KeysOnlyBufSize)
+	} else {
+		ch = make(chan dsq.Result)
+	}
+
 	go func() {
 		defer close(ch)
 		defer qr.Close()