This commit is contained in:
YouROK
2020-12-08 16:55:59 +03:00
parent 39a99f4a9a
commit e4b9fea084
5 changed files with 18 additions and 34 deletions

View File

@@ -2,6 +2,7 @@ package torrstor
import (
"io"
"sync"
"github.com/anacrolix/torrent"
)
@@ -14,6 +15,7 @@ type Reader struct {
cache *Cache
isClosed bool
mu sync.Mutex
///Preload
isPreload bool
@@ -33,6 +35,8 @@ func NewReader(file *torrent.File, cache *Cache) *Reader {
}
func (r *Reader) Seek(offset int64, whence int) (n int64, err error) {
r.mu.Lock()
defer r.mu.Unlock()
if r.isClosed {
return 0, io.EOF
}
@@ -50,6 +54,8 @@ func (r *Reader) Seek(offset int64, whence int) (n int64, err error) {
}
func (r *Reader) Read(p []byte) (n int, err error) {
r.mu.Lock()
defer r.mu.Unlock()
if r.isClosed {
return 0, io.EOF
}
@@ -60,6 +66,8 @@ func (r *Reader) Read(p []byte) (n int, err error) {
}
func (r *Reader) SetReadAHead(length int64) {
r.mu.Lock()
defer r.mu.Unlock()
r.Reader.SetReadahead(length)
r.readahead = length
}
@@ -73,6 +81,8 @@ func (r *Reader) ReadAHead() int64 {
}
func (r *Reader) Close() error {
r.mu.Lock()
defer r.mu.Unlock()
r.isClosed = true
delete(r.cache.readers, r)
return r.Reader.Close()

View File

@@ -1,10 +1,8 @@
package torrstor
import (
"fmt"
"io"
"github.com/dustin/go-humanize"
"server/log"
)
@@ -30,16 +28,13 @@ func (r *Reader) preload() {
r.isPreload = true
//TODO remove logs
fmt.Println("Start buffering...", humanize.Bytes(uint64(r.offset)), humanize.Bytes(uint64(r.endOffsetPreload)))
go func() {
buffReader := r.file.NewReader()
defer func() {
r.isPreload = false
buffReader.Close()
fmt.Println("End buffering...")
}()
buffReader.SetReadahead(1)
buffReader.SetReadahead(0)
buffReader.Seek(r.currOffsetPreload, io.SeekStart)
buff := make([]byte, 1024)
for r.currOffsetPreload < r.endOffsetPreload && !r.isClosed {

File diff suppressed because one or more lines are too long