byte cache

byte_cache/byte_cache.go

@@ -0,0 +1,36 @@
+package bytecache
+
+type BytePool struct {
+	c    chan []byte
+	w    int
+	wcap int
+}
+
+func NewBytePool(poolSize, size, cap int) *BytePool {
+	return &BytePool{
+		c:    make(chan []byte, poolSize),
+		w:    size,
+		wcap: cap,
+	}
+}
+
+func (bp *BytePool) Get() (b []byte) {
+	select {
+	case b = <-bp.c:
+	default:
+		if bp.wcap > 0 {
+			b = make([]byte, bp.w, bp.wcap)
+		} else {
+			b = make([]byte, bp.w)
+		}
+	}
+
+	return
+}
+
+func (bp *BytePool) Put(b []byte) {
+	select {
+	case bp.c <- b:
+	default:
+	}
+}

byte_cache/byte_cache_test.go

@@ -0,0 +1,28 @@
+package bytecache
+
+import "testing"
+
+func TestByteCache(t *testing.T) {
+	bp := NewBytePool(512, 1024, 1024)
+	buffer := bp.Get()
+	defer bp.Put(buffer)
+
+	t.Log(len(buffer))
+}
+
+func BenchmarkByteCache(b *testing.B) {
+	bp := NewBytePool(512, 1024, 1024)
+	b.Run("aaaaa", func(b *testing.B) {
+		for i := 0; i < b.N; i++ {
+			bp.Put(bp.Get())
+		}
+	})
+
+	b.Run("bbbb", func(b *testing.B) {
+		for i := 0; i < b.N; i++ {
+			a := make([]byte, 1024)
+
+			a[0] = 1
+		}
+	})
+}