wiring for bucketized pool

src/dbnode/client/session.go

@@ -1005,7 +1005,7 @@ func (s *session) setTopologyWithLock(topoMap topology.Map, queues []hostQueue,
 		s.pools.multiReaderIteratorArray = encoding.NewMultiReaderIteratorArrayPool([]pool.Bucket{
 			{
 				Capacity: replicas,
-				Count:    s.opts.SeriesIteratorPoolSize(),
+				Count:    pool.Size(s.opts.SeriesIteratorPoolSize()),
 			},
 		})
 		s.pools.multiReaderIteratorArray.Init()

src/dbnode/encoding/multi_reader_iterator_array_pool.go

@@ -66,7 +66,7 @@ func (p *multiReaderIteratorArrayPool) Init() {
 	for i := range p.sizesAsc {
 		buckets[i].capacity = p.sizesAsc[i].Capacity
 		buckets[i].values = make(chan []MultiReaderIterator, p.sizesAsc[i].Count)
-		for j := 0; j < p.sizesAsc[i].Count; j++ {
+		for j := 0; pool.Size(j) < p.sizesAsc[i].Count; j++ {
 			buckets[i].values <- p.alloc(p.sizesAsc[i].Capacity)
 		}
 	}

src/dbnode/encoding/mutable_series_iterators_pool.go

@@ -59,7 +59,7 @@ func (p *seriesIteratorsPool) Init() {
 	for i := range p.sizesAsc {
 		buckets[i].capacity = p.sizesAsc[i].Capacity
 		buckets[i].values = make(chan MutableSeriesIterators, p.sizesAsc[i].Count)
-		for j := 0; j < p.sizesAsc[i].Count; j++ {
+		for j := 0; pool.Size(j) < p.sizesAsc[i].Count; j++ {
 			buckets[i].values <- p.alloc(p.sizesAsc[i].Capacity)
 		}
 	}

src/query/ts/m3db/options.go

@@ -70,7 +70,7 @@ type encodedBlockOptions struct {
 func NewOptions() Options {
 	bytesPool := pool.NewCheckedBytesPool([]pool.Bucket{{
 		Capacity: defaultCapacity,
-		Count:    defaultCount,
+		Count:    pool.Size(defaultCount),
 	}}, nil, func(s []pool.Bucket) pool.BytesPool {
 		return pool.NewBytesPool(s, nil)
 	})

src/x/pool/bucketized.go

@@ -77,8 +77,10 @@ func (p *bucketizedObjectPool) Init(alloc BucketizedAllocator) {
 			opts = perBucketOpts
 		}
 
-		if !opts.Dynamic() {
-			opts = opts.SetSize(size)
+		if size > 0 {
+			opts = opts.SetSize(int(size))
+		} else if size == _dynamicPoolSize {
+			opts.SetDynamic(true)
 		}
 		iopts := opts.InstrumentOptions()
 

src/x/pool/bytes_test.go

@@ -89,7 +89,7 @@ func getBytesPool(bucketSizes int, bucketCaps []int) *bytesPool {
 	buckets := make([]Bucket, len(bucketCaps))
 	for i, cap := range bucketCaps {
 		buckets[i] = Bucket{
-			Count:    bucketSizes,
+			Count:    Size(bucketSizes),
 			Capacity: cap,
 		}
 	}

src/x/pool/checked_bytes_test.go

@@ -109,7 +109,7 @@ func getCheckedBytesPool(
 	buckets := make([]Bucket, len(bucketCaps))
 	for i, cap := range bucketCaps {
 		buckets[i] = Bucket{
-			Count:    bucketSizes,
+			Count:    Size(bucketSizes),
 			Capacity: cap,
 		}
 	}

src/x/pool/config.go

@@ -92,7 +92,7 @@ func (c *BucketizedPoolConfiguration) NewBuckets() []Bucket {
 // BucketConfiguration contains configuration for a pool bucket.
 type BucketConfiguration struct {
 	// The count of the items in the bucket.
-	Count int `yaml:"count"`
+	Count Size `yaml:"count"`
 
 	// The capacity of each item in the bucket.
 	Capacity int `yaml:"capacity"`

src/x/pool/floats_test.go

@@ -53,7 +53,7 @@ func getFloatsPool(bucketSizes int, bucketCaps []int) *floatsPool {
 	buckets := make([]Bucket, len(bucketCaps))
 	for i, cap := range bucketCaps {
 		buckets[i] = Bucket{
-			Count:    bucketSizes,
+			Count:    Size(bucketSizes),
 			Capacity: cap,
 		}
 	}

src/x/pool/types.go

@@ -96,7 +96,7 @@ type Bucket struct {
 	Capacity int
 
 	// Count is the number of fixed elements in the bucket.
-	Count int
+	Count Size
 
 	// Options is an optional override to specify options to use for a bucket,
 	// specify nil to use the options specified to the bucketized pool