diff --git a/multipool.go b/multipool.go index 1e8b27bd..1de75ae7 100644 --- a/multipool.go +++ b/multipool.go @@ -56,6 +56,9 @@ type MultiPool struct { // NewMultiPool instantiates a MultiPool with a size of the pool list and a size // per pool, and the load-balancing strategy. func NewMultiPool(size, sizePerPool int, lbs LoadBalancingStrategy, options ...Option) (*MultiPool, error) { + if lbs != RoundRobin && lbs != LeastTasks { + return nil, ErrInvalidLoadBalancingStrategy + } pools := make([]*Pool, size) for i := 0; i < size; i++ { pool, err := NewPool(sizePerPool, options...) @@ -64,9 +67,6 @@ func NewMultiPool(size, sizePerPool int, lbs LoadBalancingStrategy, options ...O } pools[i] = pool } - if lbs != RoundRobin && lbs != LeastTasks { - return nil, ErrInvalidLoadBalancingStrategy - } return &MultiPool{pools: pools, lbs: lbs}, nil }