Track completion known to implementation state

[btrtrc.git] / storage / wrappers.go

diff --git a/storage/wrappers.go b/storage/wrappers.go
index b4137e5be8e88fc3507b2729b27e5f8b9be89075..b8ba050b5503cd1729bff7f35e9d9285ffbe6283 100644 (file)
--- a/storage/wrappers.go
+++ b/storage/wrappers.go
@@ -37,8 +37,9 @@ type Piece struct {
 }
 
 func (p Piece) WriteAt(b []byte, off int64) (n int, err error) {
-       if p.GetIsComplete() {
-               err = errors.New("piece completed")
+       c := p.Completion()
+       if c.Ok && c.Complete {
+               err = errors.New("piece already completed")
                return
        }
        if off+int64(len(b)) > p.mip.Length() {