Merge pull request #12 from mgurevin/master

fix blocking Depth() call on the closed queue

Author: ploxiln

diskqueue.go

@@ -138,7 +138,12 @@ func New(name string, dataPath string, maxBytesPerFile int64,
 
 // Depth returns the depth of the queue
 func (d *diskQueue) Depth() int64 {
-	return <-d.depthChan
+	depth, ok := <-d.depthChan
+	if !ok {
+		// ioLoop exited
+		depth = d.depth
+	}
+	return depth
 }
 
 // ReadChan returns the receive-only []byte channel for reading data
@@ -188,6 +193,8 @@ func (d *diskQueue) exit(deleted bool) error {
 	// ensure that ioLoop has exited
 	<-d.exitSyncChan
 
+	close(d.depthChan)
+
 	if d.readFile != nil {
 		d.readFile.Close()
 		d.readFile = nil