mirror of
https://github.com/openimsdk/open-im-server.git
synced 2025-04-06 04:15:46 +08:00
116 lines
2.3 KiB
Go
116 lines
2.3 KiB
Go
package localcache
|
|
|
|
import (
|
|
"context"
|
|
"github.com/openimsdk/open-im-server/v3/pkg/common/localcache/local"
|
|
"time"
|
|
)
|
|
|
|
func defaultOption() *option {
|
|
return &option{
|
|
localSlotNum: 500,
|
|
localSlotSize: 20000,
|
|
linkSlotNum: 500,
|
|
localSuccessTTL: time.Minute,
|
|
localFailedTTL: time.Second * 5,
|
|
delFn: make([]func(ctx context.Context, key ...string), 0, 2),
|
|
target: emptyTarget{},
|
|
}
|
|
}
|
|
|
|
type option struct {
|
|
localSlotNum int
|
|
localSlotSize int
|
|
linkSlotNum 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 WithLocalDisable() Option {
|
|
return WithLinkSlotNum(0)
|
|
}
|
|
|
|
func WithLinkDisable() Option {
|
|
return WithLinkSlotNum(0)
|
|
}
|
|
|
|
func WithLinkSlotNum(linkSlotNum int) Option {
|
|
return func(o *option) {
|
|
o.linkSlotNum = linkSlotNum
|
|
}
|
|
}
|
|
|
|
func WithLocalSlotNum(localSlotNum int) Option {
|
|
return func(o *option) {
|
|
o.localSlotNum = localSlotNum
|
|
}
|
|
}
|
|
|
|
func WithLocalSlotSize(localSlotSize int) Option {
|
|
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() {}
|