ston1th/cachefs
0
0
Fork 0
Code Issues Releases Activity

added preload transfer rate

Browse source
This commit is contained in:
ston1th 2022-10-08 20:35:05 +02:00
commit 470eeb921c
5 changed files with 111 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

7
pkg/fs/file.go
View file

@ -42,6 +42,7 @@ func (p *preload) Read(data []byte) (n int, err error) {
if p.f.hasChunk(n) {
p.skipped++
_, err = p.f.Seek(int64(n), io.SeekCurrent)
n = 0
return
}
n, err = p.f.readToCache(data)
@ -49,8 +50,9 @@ func (p *preload) Read(data []byte) (n int, err error) {
return
}
func (f *File) Preload(ctx context.Context, fin func(), errf func(), stop func()) {
func (f *File) Preload(ctx context.Context, r *Rate, fin, errf, stop func()) {
log := f.log
defer r.Stop()
defer f.Close()
if f.offline {
log.V(2).Error(errors.New("no preload in offline mode"), "error preloading file")
@ -65,7 +67,8 @@ func (f *File) Preload(ctx context.Context, fin func(), errf func(), stop func()
log.V(2).Info("preload started")
p := &preload{f: f, ctx: ctx}
_, err := io.Copy(Discard, p)
r.SetReader(p)
_, err := io.Copy(Discard, r)
if err == context.Canceled {
log.V(2).Info("preload canceled", "skipped", p.skipped, "written", p.written)
stop()