GetN returns an error if the queue is closed (#16)

gammazero · web-flow · commit 845aae39e779 · 2025-09-13T01:21:06.000-07:00
diff --git a/dsqueue.go b/dsqueue.go
@@ -133,10 +133,17 @@ func (q *DSQueue) GetN(n int) ([][]byte, error) {
 		return nil, nil
 	}
 	rsp := make(chan getResponse)
-	q.getn <- getRequest{
+	req := getRequest{
 		n:   n,
 		rsp: rsp,
 	}
+
+	select {
+	case q.getn <- req:
+	case <-q.closed:
+		return nil, fmt.Errorf("%s queue closed", q.name)
+	}
+
 	getRsp := <-rsp
 	return getRsp.items, getRsp.err
 }

Original file line number	Diff line number	Diff line change
`@@ -133,10 +133,17 @@ func (q *DSQueue) GetN(n int) ([][]byte, error) {`
`133`	`133`	`return nil, nil`
`134`	`134`	`}`
`135`	`135`	`rsp := make(chan getResponse)`
`136`		`- q.getn <- getRequest{`
	`136`	`+ req := getRequest{`
`137`	`137`	`n: n,`
`138`	`138`	`rsp: rsp,`
`139`	`139`	`}`
	`140`	`+`
	`141`	`+ select {`
	`142`	`+ case q.getn <- req:`
	`143`	`+ case <-q.closed:`
	`144`	`+ return nil, fmt.Errorf("%s queue closed", q.name)`
	`145`	`+ }`
	`146`	`+`
`140`	`147`	`getRsp := <-rsp`
`141`	`148`	`return getRsp.items, getRsp.err`
`142`	`149`	`}`