Function TestPoolMaxBufferMemory

lib/pool/pool_test.go:259–316

(t *testing.T)

}

func TestPoolMaxBufferMemory(t *testing.T) {
	ctx := context.Background()
	ci := fs.GetConfig(ctx)
	ci.MaxBufferMemory = 4 * 4096
	defer func() {
		ci.MaxBufferMemory = 0
		totalMemory = nil
	}()
	totalMemoryInit = sync.Once{} // reset the sync.Once as it likely has been used
	totalMemory = nil
	bp := New(60*time.Second, 4096, 2, true)
	assert.NotNil(t, totalMemory)

	assert.Equal(t, bp.alloced, 0)
	buf := bp.Get()
	bp.Put(buf)
	assert.Equal(t, bp.alloced, 1)

	var (
		wg       sync.WaitGroup
		mu       sync.Mutex
		bufs     int
		maxBufs  int
		countBuf = func(i int) {
			mu.Lock()
			defer mu.Unlock()
			bufs += i
			if bufs > maxBufs {
				maxBufs = bufs
			}
		}
	)
	const trials = 50
	for i := range trials {
		wg.Go(func() {
			if i < trials/2 {
				n := i%4 + 1
				buf := bp.GetN(n)
				countBuf(n)
				time.Sleep(1 * time.Millisecond)
				countBuf(-n)
				bp.PutN(buf)
			} else {
				buf := bp.Get()
				countBuf(1)
				time.Sleep(1 * time.Millisecond)
				countBuf(-1)
				bp.Put(buf)
			}
		})
	}

	wg.Wait()

	assert.Equal(t, bufs, 0)
	assert.Equal(t, maxBufs, 4)
	assert.Equal(t, bp.alloced, 2)
}