seaweedfs/weed/filer/reader_pattern.go

42 lines
919 B
Go
Raw Normal View History

package filer
import (
"sync/atomic"
)
type ReaderPattern struct {
2022-07-13 17:20:03 +08:00
isSequentialCounter int64
lastReadStopOffset int64
}
2022-08-08 01:14:01 +08:00
const ModeChangeLimit = 3
// For streaming read: only cache the first chunk
// For random read: only fetch the requested range, instead of the whole chunk
func NewReaderPattern() *ReaderPattern {
return &ReaderPattern{
2022-07-13 17:20:03 +08:00
isSequentialCounter: 0,
lastReadStopOffset: 0,
}
}
func (rp *ReaderPattern) MonitorReadAt(offset int64, size int) {
lastOffset := atomic.SwapInt64(&rp.lastReadStopOffset, offset+int64(size))
counter := atomic.LoadInt64(&rp.isSequentialCounter)
if lastOffset == offset {
if counter < ModeChangeLimit {
atomic.AddInt64(&rp.isSequentialCounter, 1)
2022-08-08 01:14:01 +08:00
}
2022-07-13 17:20:03 +08:00
} else {
if counter > -ModeChangeLimit {
atomic.AddInt64(&rp.isSequentialCounter, -1)
2022-08-08 01:14:01 +08:00
}
}
}
func (rp *ReaderPattern) IsRandomMode() bool {
return atomic.LoadInt64(&rp.isSequentialCounter) < 0
}