add cache generic

Author: markus621

cache/cache_time.go

@@ -0,0 +1,102 @@
+package cache
+
+import (
+	"context"
+	"sync"
+	"time"
+
+	"go.osspkg.com/routine"
+)
+
+type (
+	withTTL[K comparable, T interface{}] struct {
+		ttl  time.Duration
+		list map[K]*ttlItem[T]
+		mux  sync.RWMutex
+	}
+
+	ttlItem[T interface{}] struct {
+		link T
+		ts   int64
+	}
+)
+
+func NewWithTTL[K comparable, T interface{}](ctx context.Context, ttl time.Duration) TCache[K, T] {
+	cache := &withTTL[K, T]{
+		ttl:  ttl,
+		list: make(map[K]*ttlItem[T], 1000),
+	}
+	go cache.cleaner(ctx)
+	return cache
+}
+
+func (v *withTTL[K, T]) cleaner(ctx context.Context) {
+	routine.Interval(ctx, v.ttl, func(ctx context.Context) {
+		curr := time.Now().Unix()
+
+		for k, t := range v.list {
+			if t.ts < curr {
+				delete(v.list, k)
+			}
+		}
+	})
+}
+
+func (v *withTTL[K, T]) Has(key K) bool {
+	v.mux.RLock()
+	defer v.mux.RUnlock()
+
+	_, ok := v.list[key]
+
+	return ok
+}
+
+func (v *withTTL[K, T]) Get(key K) (T, bool) {
+	v.mux.RLock()
+	defer v.mux.RUnlock()
+
+	item, ok := v.list[key]
+	if !ok {
+		return *(new(T)), false
+	}
+
+	return item.link, true
+}
+
+func (v *withTTL[K, T]) Set(key K, value T) {
+	v.mux.Lock()
+	defer v.mux.Unlock()
+
+	v.list[key] = &ttlItem[T]{
+		link: value,
+		ts:   time.Now().Add(v.ttl).Unix(),
+	}
+}
+
+func (v *withTTL[K, T]) Del(key K) {
+	v.mux.Lock()
+	defer v.mux.Unlock()
+
+	delete(v.list, key)
+}
+
+func (v *withTTL[K, T]) Keys() []K {
+	v.mux.RLock()
+	defer v.mux.RUnlock()
+
+	result := make([]K, 0, len(v.list))
+	for k := range v.list {
+		result = append(result, k)
+	}
+
+	return result
+}
+
+func (v *withTTL[K, T]) Flush() {
+	v.mux.Lock()
+	defer v.mux.Unlock()
+
+	for k := range v.list {
+		delete(v.list, k)
+	}
+}

cache/cache_time_test.go

@@ -0,0 +1,21 @@
+package cache
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"go.osspkg.com/casecheck"
+)
+
+func TestUnit_WithTTL_Pointer(t *testing.T) {
+	type A struct {
+		Data uint64
+	}
+
+	ctx, _ := context.WithTimeout(context.TODO(), time.Second)
+	c := NewWithTTL[string, *A](ctx, time.Minute)
+	v, ok := c.Get("a")
+	casecheck.False(t, ok)
+	casecheck.Nil(t, v)
+}

cache/common.go

@@ -0,0 +1,10 @@
+package cache
+
+type TCache[K comparable, T interface{}] interface {
+	Has(key K) bool
+	Get(key K) (T, bool)
+	Set(key K, value T)
+	Del(key K)
+	Keys() []K
+	Flush()
+}

go.mod

@@ -5,6 +5,7 @@ go 1.21
 require (
 	go.osspkg.com/casecheck v0.3.0
 	go.osspkg.com/errors v0.3.1
+	go.osspkg.com/routine v0.3.1
 	go.osspkg.com/syncing v0.3.0
 	gopkg.in/yaml.v3 v3.0.1
 )

go.sum

@@ -2,6 +2,8 @@ go.osspkg.com/casecheck v0.3.0 h1:x15blEszElbrHrEH5H02JIIhGIg/lGZzIt1kQlD3pwM=
 go.osspkg.com/casecheck v0.3.0/go.mod h1:TRFXDMFJEOtnlp3ET2Hix3osbxwPWhvaiT/HfD3+gBA=
 go.osspkg.com/errors v0.3.1 h1:F9m/EEd/Ot2jba/TV7tvVRIpWXzIpNLc7vRJKcBD86A=
 go.osspkg.com/errors v0.3.1/go.mod h1:dKXe6Rt07nzY7OyKQNZ8HGBicZ2uQ5TKEoVFnVFOK44=
+go.osspkg.com/routine v0.3.1 h1:R0o4P0Ml5eoeHc2DiHjRvHBo/XXrW5nJNqIj3ToRzjg=
+go.osspkg.com/routine v0.3.1/go.mod h1:z5AvvTbB19/tt1E5JOb4POhK1tOPgmejajgao/IWn4s=
 go.osspkg.com/syncing v0.3.0 h1:yBkCsDPEt12a+qagInFFt7+ZongfT+GjSQl7nBmcybI=
 go.osspkg.com/syncing v0.3.0/go.mod h1:Dpe0ljlEG6cI2Y9PxEjKiYEX2sgs1eUjWNVjFu4/iB0=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=