49 lines
833 B
Go
49 lines
833 B
Go
package memcache
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// janitor for collecting expired items and cleaning them.
|
|
type janitor struct {
|
|
ctx context.Context
|
|
interval time.Duration
|
|
done chan struct{}
|
|
once sync.Once
|
|
}
|
|
|
|
func newJanitor(ctx context.Context, interval time.Duration) *janitor {
|
|
j := &janitor{
|
|
ctx: ctx,
|
|
interval: interval,
|
|
done: make(chan struct{}),
|
|
}
|
|
return j
|
|
}
|
|
|
|
// stop to stop the janitor.
|
|
func (j *janitor) stop() {
|
|
j.once.Do(func() { close(j.done) })
|
|
}
|
|
|
|
// run with the given cleanup callback function.
|
|
func (j *janitor) run(cleanup func()) {
|
|
go func() {
|
|
ticker := time.NewTicker(j.interval)
|
|
defer ticker.Stop()
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
cleanup()
|
|
case <-j.done:
|
|
cleanup() // last call
|
|
return
|
|
case <-j.ctx.Done():
|
|
j.stop()
|
|
}
|
|
}
|
|
}()
|
|
}
|