aboutsummaryrefslogtreecommitdiff
path: root/weed/filer/reader_pattern.go
blob: b0906e99f1a28806cd476b52984f419e0cddfe4e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
package filer

import (
	"sync/atomic"
)

type ReaderPattern struct {
	isSequentialCounter int64
	lastReadStopOffset  int64
}

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{
		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)
		}
	} else {
		if counter > -ModeChangeLimit {
			atomic.AddInt64(&rp.isSequentialCounter, -1)
		}
	}
}

func (rp *ReaderPattern) IsRandomMode() bool {
	return atomic.LoadInt64(&rp.isSequentialCounter) < 0
}