1e919dd6b1
This allows lots of behaviour to be baked into the new Client, Torrent and Piece wrappers, rather than duplicating (badly) them in all the backend implementations.
52 lines
1.1 KiB
Go
52 lines
1.1 KiB
Go
package storage
|
|
|
|
import (
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/anacrolix/torrent/metainfo"
|
|
)
|
|
|
|
type fileStoragePiece struct {
|
|
*fileTorrentStorage
|
|
p metainfo.Piece
|
|
io.WriterAt
|
|
io.ReaderAt
|
|
}
|
|
|
|
func (me *fileStoragePiece) pieceKey() metainfo.PieceKey {
|
|
return metainfo.PieceKey{me.infoHash, me.p.Index()}
|
|
}
|
|
|
|
func (fs *fileStoragePiece) GetIsComplete() bool {
|
|
ret, err := fs.completion.Get(fs.pieceKey())
|
|
if err != nil || !ret {
|
|
return false
|
|
}
|
|
// If it's allegedly complete, check that its constituent files have the
|
|
// necessary length.
|
|
for _, fi := range extentCompleteRequiredLengths(fs.p.Info, fs.p.Offset(), fs.p.Length()) {
|
|
s, err := os.Stat(fs.fileInfoName(fi))
|
|
if err != nil || s.Size() < fi.Length {
|
|
ret = false
|
|
break
|
|
}
|
|
}
|
|
if ret {
|
|
return true
|
|
}
|
|
// The completion was wrong, fix it.
|
|
fs.completion.Set(fs.pieceKey(), false)
|
|
return false
|
|
}
|
|
|
|
func (fs *fileStoragePiece) MarkComplete() error {
|
|
fs.completion.Set(fs.pieceKey(), true)
|
|
return nil
|
|
}
|
|
|
|
func (fs *fileStoragePiece) MarkNotComplete() error {
|
|
fs.completion.Set(fs.pieceKey(), false)
|
|
return nil
|
|
}
|