feat: msg local cache

pkg/common/localcache/option.go

@@ -0,0 +1,113 @@
+package localcache
+
+import (
+	"context"
+	"github.com/openimsdk/open-im-server/v3/pkg/common/localcache/local"
+	"time"
+)
+
+func defaultOption() *option {
+	return &option{
+		enable:          true,
+		localSlotNum:    500,
+		localSlotSize:   20000,
+		localSuccessTTL: time.Minute,
+		localFailedTTL:  time.Second * 5,
+		delFn:           make([]func(ctx context.Context, key ...string), 0, 2),
+		target:          emptyTarget{},
+	}
+}
+
+type option struct {
+	enable          bool
+	localSlotNum    int
+	localSlotSize   int
+	localSuccessTTL time.Duration
+	localFailedTTL  time.Duration
+	delFn           []func(ctx context.Context, key ...string)
+	delCh           func(fn func(key ...string))
+	target          local.Target
+}
+
+type Option func(o *option)
+
+func WithDisable() Option {
+	return func(o *option) {
+		o.enable = false
+	}
+}
+
+func WithLocalSlotNum(localSlotNum int) Option {
+	if localSlotNum < 1 {
+		panic("localSlotNum should be greater than 0")
+	}
+	return func(o *option) {
+		o.localSlotNum = localSlotNum
+	}
+}
+
+func WithLocalSlotSize(localSlotSize int) Option {
+	if localSlotSize < 1 {
+		panic("localSlotSize should be greater than 0")
+	}
+	return func(o *option) {
+		o.localSlotSize = localSlotSize
+	}
+}
+
+func WithLocalSuccessTTL(localSuccessTTL time.Duration) Option {
+	if localSuccessTTL < 0 {
+		panic("localSuccessTTL should be greater than 0")
+	}
+	return func(o *option) {
+		o.localSuccessTTL = localSuccessTTL
+	}
+}
+
+func WithLocalFailedTTL(localFailedTTL time.Duration) Option {
+	if localFailedTTL < 0 {
+		panic("localFailedTTL should be greater than 0")
+	}
+	return func(o *option) {
+		o.localFailedTTL = localFailedTTL
+	}
+}
+
+func WithTarget(target local.Target) Option {
+	if target == nil {
+		panic("target should not be nil")
+	}
+	return func(o *option) {
+		o.target = target
+	}
+}
+
+func WithDeleteKeyBefore(fn func(ctx context.Context, key ...string)) Option {
+	if fn == nil {
+		panic("fn should not be nil")
+	}
+	return func(o *option) {
+		o.delFn = append(o.delFn, fn)
+	}
+}
+
+func WithDeleteLocal(fn func(fn func(key ...string))) Option {
+	if fn == nil {
+		panic("fn should not be nil")
+	}
+	return func(o *option) {
+		o.delCh = fn
+	}
+}
+
+type emptyTarget struct{}
+
+func (e emptyTarget) IncrGetHit() {}
+
+func (e emptyTarget) IncrGetSuccess() {}
+
+func (e emptyTarget) IncrGetFailed() {}
+
+func (e emptyTarget) IncrDelHit() {}
+
+func (e emptyTarget) IncrDelNotFound() {}