Track completion known to implementation state

[btrtrc.git] / storage / completion.go

diff --git a/storage/completion.go b/storage/completion.go
index c464a0a7b5e06dfad2aaa1a0c852012a251bf67e..3d2e4d77dd191e85082ac04df9da6fc5740069c1 100644 (file)
--- a/storage/completion.go
+++ b/storage/completion.go
@@ -6,10 +6,14 @@ import (
        "github.com/anacrolix/torrent/metainfo"
 )
 
+type PieceCompletionGetSetter interface {
+       Get(metainfo.PieceKey) (Completion, error)
+       Set(metainfo.PieceKey, bool) error
+}
+
 // Implementations track the completion of pieces. It must be concurrent-safe.
 type PieceCompletion interface {
-       Get(metainfo.PieceKey) (bool, error)
-       Set(metainfo.PieceKey, bool) error
+       PieceCompletionGetSetter
        Close() error
 }