chore(consumer): remove request timeout parameter.

Signed-off-by: Bo-Yi.Wu <[email protected]>

Author: appleboy

consumer.go

@@ -5,7 +5,6 @@ import (
 	"errors"
 	"sync"
 	"sync/atomic"
-	"time"
 
 	"github.com/golang-queue/queue/core"
 )
@@ -23,8 +22,6 @@ type Consumer struct {
 	logger    Logger
 	stopOnce  sync.Once
 	stopFlag  int32
-
-	requestTimeout time.Duration
 }
 
 // Run to execute new task
@@ -74,7 +71,7 @@ func (s *Consumer) Request() (core.QueuedMessage, error) {
 			return nil, ErrQueueHasBeenClosed
 		}
 		return task, nil
-	case <-time.After(s.requestTimeout):
+	default:
 		return nil, ErrNoTaskInQueue
 	}
 }
@@ -88,8 +85,6 @@ func NewConsumer(opts ...Option) *Consumer {
 		exit:      make(chan struct{}),
 		logger:    o.logger,
 		runFunc:   o.fn,
-
-		requestTimeout: o.requestTimeout,
 	}
 
 	return w

consumer_test.go

@@ -497,12 +497,11 @@ func TestCancelRetryCountWithNewMessage(t *testing.T) {
 	assert.Equal(t, 2, count)
 }
 
-func TestRequestTimeout(t *testing.T) {
+func TestErrNoTaskInQueue(t *testing.T) {
 	w := NewConsumer(
 		WithFn(func(ctx context.Context, m core.QueuedMessage) error {
 			return nil
 		}),
-		WithRequestTimeout(10*time.Millisecond),
 	)
 	task, err := w.Request()
 	assert.Nil(t, task)

options.go

@@ -3,18 +3,16 @@ package queue
 import (
 	"context"
 	"runtime"
-	"time"
 
 	"github.com/golang-queue/queue/core"
 )
 
 var (
-	defaultQueueSize      = 4096
-	defaultWorkerCount    = runtime.NumCPU()
-	defaultNewLogger      = NewLogger()
-	defaultFn             = func(context.Context, core.QueuedMessage) error { return nil }
-	defaultMetric         = NewMetric()
-	defaultRequestTimeout = 5 * time.Second
+	defaultQueueSize   = 4096
+	defaultWorkerCount = runtime.NumCPU()
+	defaultNewLogger   = NewLogger()
+	defaultFn          = func(context.Context, core.QueuedMessage) error { return nil }
+	defaultMetric      = NewMetric()
 )
 
 // An Option configures a mutex.
@@ -47,13 +45,6 @@ func WithQueueSize(num int) Option {
 	})
 }
 
-// WithQueueSize set worker count
-func WithRequestTimeout(timeout time.Duration) Option {
-	return OptionFunc(func(q *Options) {
-		q.requestTimeout = timeout
-	})
-}
-
 // WithLogger set custom logger
 func WithLogger(l Logger) Option {
 	return OptionFunc(func(q *Options) {
@@ -90,9 +81,6 @@ type Options struct {
 	worker      core.Worker
 	fn          func(context.Context, core.QueuedMessage) error
 	metric      Metric
-
-	// timeout for request single task
-	requestTimeout time.Duration
 }
 
 // NewOptions initialize the default value for the options
@@ -104,8 +92,6 @@ func NewOptions(opts ...Option) *Options {
 		worker:      nil,
 		fn:          defaultFn,
 		metric:      defaultMetric,
-
-		requestTimeout: defaultRequestTimeout,
 	}
 
 	// Loop through each option