forked from YouROK/TorrServer
/
diskpiece.go
85 lines (69 loc) · 1.6 KB
/
diskpiece.go
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package torrstor
import (
"io"
"os"
"path/filepath"
"strconv"
"sync"
"time"
"github.com/Terisback/TorrServer/server/log"
"github.com/Terisback/TorrServer/server/settings"
)
type DiskPiece struct {
piece *Piece
name string
mu sync.RWMutex
}
func NewDiskPiece(p *Piece) *DiskPiece {
name := filepath.Join(settings.BTsets.TorrentsSavePath, p.cache.hash.HexString(), strconv.Itoa(p.Id))
ff, err := os.Stat(name)
if err == nil {
p.Size = ff.Size()
p.Complete = ff.Size() == p.cache.pieceLength
p.Accessed = ff.ModTime().Unix()
}
return &DiskPiece{piece: p, name: name}
}
func (p *DiskPiece) WriteAt(b []byte, off int64) (n int, err error) {
p.mu.Lock()
defer p.mu.Unlock()
ff, err := os.OpenFile(p.name, os.O_RDWR|os.O_CREATE, 0o666)
if err != nil {
log.TLogln("Error open file:", err)
return 0, err
}
defer ff.Close()
n, err = ff.WriteAt(b, off)
p.piece.Size += int64(n)
if p.piece.Size > p.piece.cache.pieceLength {
p.piece.Size = p.piece.cache.pieceLength
}
p.piece.Accessed = time.Now().Unix()
return
}
func (p *DiskPiece) ReadAt(b []byte, off int64) (n int, err error) {
p.mu.Lock()
defer p.mu.Unlock()
ff, err := os.OpenFile(p.name, os.O_RDONLY, 0o666)
if os.IsNotExist(err) {
return 0, io.EOF
}
if err != nil {
log.TLogln("Error open file:", err)
return 0, err
}
defer ff.Close()
n, err = ff.ReadAt(b, off)
p.piece.Accessed = time.Now().Unix()
if int64(len(b))+off >= p.piece.Size {
go p.piece.cache.cleanPieces()
}
return n, nil
}
func (p *DiskPiece) Release() {
p.mu.Lock()
defer p.mu.Unlock()
p.piece.Size = 0
p.piece.Complete = false
os.Remove(p.name)
}