From bff71ed368ab29d94039c889dbf1fc3484cc68e0 Mon Sep 17 00:00:00 2001
From: gobwas <gobwas@gmail.com>
Date: Mon, 22 Feb 2016 22:17:07 +0300
Subject: [PATCH] test parallel

---
 match/btree_test.go     |  9 +++--
 match/segements_test.go | 56 +++++++++++++++++++--------
 match/segments.go       | 84 +++++++++++++++++++++++++++--------------
 3 files changed, 102 insertions(+), 47 deletions(-)

diff --git a/match/btree_test.go b/match/btree_test.go
index 4a05bcf..dced6e5 100644
--- a/match/btree_test.go
+++ b/match/btree_test.go
@@ -75,8 +75,9 @@ func BenchmarkMatchBTree(b *testing.B) {
 
 	bt := NewBTree(v, l, r)
 
-	b.SetParallelism(1)
-	for i := 0; i < b.N; i++ {
-		bt.Match(fixture)
-	}
+	b.RunParallel(func(pb *testing.PB) {
+		for pb.Next() {
+			bt.Match(fixture)
+		}
+	})
 }
diff --git a/match/segements_test.go b/match/segements_test.go
index d3cf89d..9eaf49c 100644
--- a/match/segements_test.go
+++ b/match/segements_test.go
@@ -5,31 +5,57 @@ import (
 )
 
 func BenchmarkPerfPoolSequenced(b *testing.B) {
-	pool := NewPoolSequenced(32, func() []int {
+	pool := NewPoolSequenced(512, func() []int {
 		return make([]int, 0, 16)
 	})
 
-	for i := 0; i < b.N; i++ {
-		s := pool.Get()
-		pool.Put(s)
-	}
+	b.SetParallelism(32)
+	b.RunParallel(func(pb *testing.PB) {
+		for pb.Next() {
+			s := pool.Get()
+			pool.Put(s)
+		}
+	})
 }
 
 func BenchmarkPerfPoolSynced(b *testing.B) {
 	pool := NewPoolSynced(32)
 
-	for i := 0; i < b.N; i++ {
-		s := pool.Get()
-		pool.Put(s)
-	}
+	b.SetParallelism(32)
+	b.RunParallel(func(pb *testing.PB) {
+		for pb.Next() {
+			s := pool.Get()
+			pool.Put(s)
+		}
+	})
 }
-func BenchmarkPerfPoolPoolNative(b *testing.B) {
-	pool := NewPoolNative(32)
 
-	for i := 0; i < b.N; i++ {
-		s := pool.Get()
-		pool.Put(s)
-	}
+func BenchmarkPerfPoolNative(b *testing.B) {
+	pool := NewPoolNative(func() []int {
+		return make([]int, 0, 16)
+	})
+
+	b.SetParallelism(32)
+	b.RunParallel(func(pb *testing.PB) {
+		for pb.Next() {
+			s := pool.Get()
+			pool.Put(s)
+		}
+	})
+}
+
+func BenchmarkPerfPoolStatic(b *testing.B) {
+	pool := NewPoolStatic(32, func() []int {
+		return make([]int, 0, 16)
+	})
+
+	b.SetParallelism(32)
+	b.RunParallel(func(pb *testing.PB) {
+		for pb.Next() {
+			i, v := pool.Get()
+			pool.Put(i, v)
+		}
+	})
 }
 
 func BenchmarkPerfMake(b *testing.B) {
diff --git a/match/segments.go b/match/segments.go
index f714034..d5df10d 100644
--- a/match/segments.go
+++ b/match/segments.go
@@ -1,8 +1,11 @@
 package match
 
-import "sync"
+import (
+	"sync"
+	"sync/atomic"
+)
 
-var segmentsPools [1024]*PoolSequenced
+var segmentsPools [1024]*PoolNative
 
 func toPowerOfTwo(v int) int {
 	v--
@@ -26,18 +29,9 @@ const (
 func init() {
 	for i := maxSegment; i >= minSegment; i >>= 1 {
 		func(i int) {
-			//			pool := sync.Pool{
-			//				New: func() interface{} {
-			//					//					fmt.Printf("N%d;", i)
-			//					return make([]int, 0, i)
-			//				},
-			//			}
-
-			pool := NewPoolSequenced(64, func() []int {
+			segmentsPools[i-1] = NewPoolNative(func() []int {
 				return make([]int, 0, i)
 			})
-
-			segmentsPools[i-1] = pool
 		}(i)
 	}
 }
@@ -54,17 +48,11 @@ func getIdx(c int) int {
 	}
 }
 
-//var p = make([]int, 0, 128)
-
 func acquireSegments(c int) []int {
-	//	return p
-	//	fmt.Printf("a%d;", getIdx(c))
 	return segmentsPools[getIdx(c)].Get()
 }
 
 func releaseSegments(s []int) {
-	//	p = s
-	//		fmt.Printf("r%d;", getIdx(cap(s)))
 	segmentsPools[getIdx(cap(s))].Put(s)
 }
 
@@ -141,25 +129,65 @@ func (p *PoolSynced) Put(s []int) {
 }
 
 type PoolNative struct {
-	size int
-	pool sync.Pool
+	pool *sync.Pool
 }
 
-func NewPoolNative(size int) *PoolNative {
+func NewPoolNative(f newSegmentsFunc) *PoolNative {
 	return &PoolNative{
-		size: size,
+		pool: &sync.Pool{New: func() interface{} {
+			return f()
+		}},
 	}
 }
 
 func (p *PoolNative) Get() []int {
-	s := p.pool.Get()
-	if s == nil {
-		return make([]int, 0, p.size)
-	}
-
-	return s.([]int)
+	return p.pool.Get().([]int)[:0]
 }
 
 func (p *PoolNative) Put(s []int) {
 	p.pool.Put(s)
 }
+
+type segments struct {
+	data   []int
+	locked int32
+}
+
+type PoolStatic struct {
+	f    newSegmentsFunc
+	pool []*segments
+}
+
+func NewPoolStatic(size int, f newSegmentsFunc) *PoolStatic {
+	p := &PoolStatic{
+		f:    f,
+		pool: make([]*segments, 0, size),
+	}
+
+	for i := 0; i < size; i++ {
+		p.pool = append(p.pool, &segments{
+			data: f(),
+		})
+	}
+
+	return p
+}
+
+func (p *PoolStatic) Get() (int, []int) {
+	for i, s := range p.pool {
+		if atomic.CompareAndSwapInt32(&s.locked, 0, 1) {
+			return i, s.data
+		}
+	}
+
+	return -1, p.f()
+}
+
+func (p *PoolStatic) Put(i int, s []int) {
+	if i < 0 {
+		return
+	}
+
+	p.pool[i].data = s
+	atomic.CompareAndSwapInt32(&(p.pool[i].locked), 1, 0)
+}