74 lines
1.3 KiB
Go
74 lines
1.3 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
)
|
|
|
|
//更新数据模块过期
|
|
func (this *DB) UpDateModelExpired(key string, childs map[string]struct{}, expired time.Duration) {
|
|
this.mu.RLock()
|
|
exp, ok := this.data[key]
|
|
this.mu.RUnlock()
|
|
if ok {
|
|
if childs != nil {
|
|
if exp.keys == nil {
|
|
exp.keys = make(map[string]struct{})
|
|
}
|
|
for k, _ := range childs {
|
|
exp.keys[k] = struct{}{}
|
|
}
|
|
}
|
|
exp.expired = time.Now().Add(expired)
|
|
} else {
|
|
exp = &ModelDataExpired{
|
|
key: key,
|
|
keys: childs,
|
|
expired: time.Now().Add(expired),
|
|
}
|
|
this.mu.Lock()
|
|
this.data[key] = exp
|
|
this.mu.Unlock()
|
|
}
|
|
}
|
|
|
|
//定时清理过期数据
|
|
func (this *DB) run() {
|
|
timer := time.NewTicker(time.Minute * 1)
|
|
defer timer.Stop()
|
|
for {
|
|
select {
|
|
case <-timer.C:
|
|
this.scanning()
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
//扫描过期
|
|
func (this *DB) scanning() {
|
|
now := time.Now()
|
|
this.mu.Lock()
|
|
temp := make([]*ModelDataExpired, 0, len(this.data))
|
|
for k, v := range this.data {
|
|
if v.expired.Before(now) { //过期
|
|
temp = append(temp, v)
|
|
delete(this.data, k)
|
|
}
|
|
}
|
|
this.mu.Unlock()
|
|
ctx := context.Background()
|
|
pipe := this.local.Redis.Pipeline()
|
|
for _, v := range temp {
|
|
pipe.Del(ctx, v.key)
|
|
if v.keys != nil {
|
|
for k1, _ := range v.keys {
|
|
pipe.Del(ctx, k1)
|
|
}
|
|
}
|
|
}
|
|
if _, err := pipe.Exec(ctx); err != nil {
|
|
this.options.Log.Errorln(err)
|
|
}
|
|
}
|