mirror of
https://github.com/chrislusf/seaweedfs
synced 2024-06-28 13:23:03 +02:00
41 lines
795 B
Go
41 lines
795 B
Go
//+build rocksdb
|
|
|
|
package rocksdb
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/tecbot/gorocksdb"
|
|
|
|
"github.com/chrislusf/seaweedfs/weed/filer"
|
|
)
|
|
|
|
type TTLFilter struct {
|
|
skipLevel0 bool
|
|
}
|
|
|
|
func NewTTLFilter() gorocksdb.CompactionFilter {
|
|
return &TTLFilter{
|
|
skipLevel0: true,
|
|
}
|
|
}
|
|
|
|
func (t *TTLFilter) Filter(level int, key, val []byte) (remove bool, newVal []byte) {
|
|
// decode could be slow, causing write stall
|
|
// level >0 sst can run compaction in parallel
|
|
if !t.skipLevel0 || level > 0 {
|
|
entry := filer.Entry{}
|
|
if err := entry.DecodeAttributesAndChunks(val); err == nil {
|
|
if entry.TtlSec > 0 &&
|
|
entry.Crtime.Add(time.Duration(entry.TtlSec)*time.Second).Before(time.Now()) {
|
|
return true, nil
|
|
}
|
|
}
|
|
}
|
|
return false, val
|
|
}
|
|
|
|
func (t *TTLFilter) Name() string {
|
|
return "TTLFilter"
|
|
}
|