Improve BufferedChannelQueue efficiency

Author: johnteee

queue.go

@@ -584,7 +584,16 @@ func (q *BufferedChannelQueue) Put(val interface{}) error {
 	// 	return ErrQueueIsClosed
 	// }
 	//
-	// return q.blockingQueue.Put(val)
+	// q.lock.Lock()
+	// poolCount := q.pool.Count()
+	//
+	// // If appearing nothing in the pool
+	// if poolCount == 0 {
+	// 	defer q.lock.Unlock()
+	// 	// Try channel
+	// 	return q.blockingQueue.Put(val)
+	// }
+	// q.lock.Unlock()
 
 	return q.Offer(val)
 }
@@ -638,23 +647,31 @@ func (q *BufferedChannelQueue) Offer(val interface{}) error {
 		return ErrQueueIsClosed
 	}
 
+	poolCount := q.pool.Count()
+
+	// If appearing nothing in the pool
+	if poolCount == 0 {
+		// Try channel
+		err := q.blockingQueue.Offer(val)
+		if err == nil {
+			// Success
+			return nil
+		} else if err == ErrQueueIsFull {
+			// Do nothing and let pool.Offer(val)
+		} else {
+			// Other
+			return err
+		}
+	}
+
 	// Before +1: >=, After +1: >
-	if q.pool.Count() >= q.bufferSizeMaximum {
+	if poolCount >= q.bufferSizeMaximum {
 		return ErrQueueIsFull
 	}
 
 	q.pool.Offer(val)
 	q.loadWorkerCh.Offer(1)
 	return nil
-
-	// err := q.blockingQueue.Offer(val)
-	// if err == ErrQueueIsFull {
-	// 	q.pool.Offer(val)
-	// 	q.loadWorkerCh.Offer(1)
-	// 	return nil
-	// }
-	//
-	// return err
 }
 
 // Poll Poll the val(non-blocking)

queue_test.go

@@ -275,13 +275,10 @@ func TestNewBufferedChannelQueue(t *testing.T) {
 
 	err = queue.Offer(1)
 	assert.Equal(t, nil, err)
-	time.Sleep(1 * timeout)
 	err = queue.Offer(2)
 	assert.Equal(t, nil, err)
-	time.Sleep(1 * timeout)
 	err = queue.Offer(3)
 	assert.Equal(t, nil, err)
-	time.Sleep(1 * timeout)
 	// Channel: only 3 positions & Buffer: 1 position, now `4` is inserted into the buffer(buffer size： 1)
 	err = queue.Offer(4)
 	assert.Equal(t, nil, err)
@@ -319,6 +316,7 @@ func TestNewBufferedChannelQueue(t *testing.T) {
 	go func() {
 		for i := 1; i <= 10000; i++ {
 			// err := bufferedChannelQueue.PutWithTimeout(i, timeout)
+			// err := bufferedChannelQueue.Put(i)
 			err := bufferedChannelQueue.Offer(i)
 			assert.Equal(t, nil, err)
 		}