updated package name

Author: andectionsharechat

bigcache.go

@@ -1,4 +1,4 @@
-package fastcache
+package turbocache
 
 import (
 	xxhash "github.com/cespare/xxhash/v2"

bigcache_test.go

@@ -1,4 +1,4 @@
-package fastcache
+package turbocache
 
 import (
 	"bytes"

bigcache_timing_test.go

@@ -1,4 +1,4 @@
-package fastcache
+package turbocache
 
 import (
 	"testing"

fastcache.go

@@ -1,7 +1,7 @@
 // Package fastcache implements fast in-memory cache.
 //
 // The package has been extracted from https://victoriametrics.com/
-package fastcache
+package turbocache
 
 import (
 	"fmt"
@@ -254,35 +254,44 @@ func (b *bucket) Init(maxBytes uint64) {
 	b.m = make(map[uint64]uint64)
 	b.Reset()
 	b.setBuf = make(chan *insertValue, setBufSize)
-	go func() {
-		t := time.Tick(maxDelayMillis * time.Millisecond)
-		var firstTimeTimestamp int64
-		keys := make([][]byte, 0, 64)
-		values := make([][]byte, 0, 64)
-		for {
-			select {
-			case i := <-b.setBuf:
-				if firstTimeTimestamp == 0 {
-					firstTimeTimestamp = time.Now().UnixMilli()
-				}
-				keys = append(keys, i.K)
-				values = append(values, i.V)
-				if len(keys) >= writeSizeThreshold || time.Since(time.UnixMilli(firstTimeTimestamp)).Milliseconds() >= maxDelayMillis {
-					b.setBatch(keys, values)
-					firstTimeTimestamp = 0
-					keys = make([][]byte, 0, 64)
-					values = make([][]byte, 0, 64)
-				}
-			case _ = <-t:
-				if firstTimeTimestamp != 0 && (len(keys) >= writeSizeThreshold || time.Since(time.UnixMilli(firstTimeTimestamp)).Milliseconds() >= maxDelayMillis) {
-					b.setBatch(keys, values)
-					firstTimeTimestamp = 0
-					keys = make([][]byte, 0, 64)
-					values = make([][]byte, 0, 64)
+	go b.processWriteQueue()
+}
+
+func (b *bucket) processWriteQueue() {
+	t := time.Tick(maxDelayMillis * time.Millisecond)
+	var firstTimeTimestamp int64
+	keys := make([][]byte, 0, 64)
+	values := make([][]byte, 0, 64)
+	waitGroups := make([]*sync.WaitGroup, 0, 64)
+	for {
+		select {
+		case i := <-b.setBuf:
+			if firstTimeTimestamp == 0 {
+				firstTimeTimestamp = time.Now().UnixMilli()
+			}
+			keys = append(keys, i.K)
+			values = append(values, i.V)
+			waitGroups = append(waitGroups, i.waitGroup)
+			if len(keys) >= writeSizeThreshold || time.Since(time.UnixMilli(firstTimeTimestamp)).Milliseconds() >= maxDelayMillis {
+				b.setBatch(keys, values)
+				firstTimeTimestamp = 0
+				keys = make([][]byte, 0, 64)
+				values = make([][]byte, 0, 64)
+				waitGroups = make([]*sync.WaitGroup, 0, 64)
+			}
+		case _ = <-t:
+			if firstTimeTimestamp != 0 && (len(keys) >= writeSizeThreshold || time.Since(time.UnixMilli(firstTimeTimestamp)).Milliseconds() >= maxDelayMillis) {
+				b.setBatch(keys, values)
+				for _, group := range waitGroups {
+					group.Done()
 				}
+				firstTimeTimestamp = 0
+				keys = make([][]byte, 0, 64)
+				values = make([][]byte, 0, 64)
+				waitGroups = make([]*sync.WaitGroup, 0, 64)
 			}
 		}
-	}()
+	}
 }
 
 func (b *bucket) Reset() {
@@ -394,9 +403,12 @@ func (b *bucket) set(k, v []byte, h uint64) {
 }
 
 func (b *bucket) Set(k, v []byte, h uint64) {
+	var wg sync.WaitGroup
+	wg.Add(1)
 	b.setBuf <- &insertValue{
-		K: k,
-		V: v,
+		K:         k,
+		V:         v,
+		waitGroup: &wg,
 	}
 }
 
@@ -467,5 +479,6 @@ func (b *bucket) Del(h uint64) {
 }
 
 type insertValue struct {
-	K, V []byte
+	K, V      []byte
+	waitGroup *sync.WaitGroup
 }

fastcache_gen_test.go

@@ -1,4 +1,4 @@
-package fastcache
+package turbocache
 
 import (
 	"bytes"

fastcache_test.go

@@ -1,4 +1,4 @@
-package fastcache
+package turbocache
 
 import (
 	"fmt"

fastcache_timing_test.go

@@ -1,4 +1,4 @@
-package fastcache
+package turbocache
 
 import (
 	"fmt"

file.go

@@ -1,4 +1,4 @@
-package fastcache
+package turbocache
 
 import (
 	"encoding/binary"

file_test.go

@@ -1,4 +1,4 @@
-package fastcache
+package turbocache
 
 import (
 	"fmt"

file_timing_test.go

@@ -1,4 +1,4 @@
-package fastcache
+package turbocache
 
 import (
 	"fmt"