Change pieceIndex to peer_protocol.Integer

[btrtrc.git] / metainfo / piece.go

diff --git a/metainfo/piece.go b/metainfo/piece.go
index 55cdc43c143e32d7e1b183e0250e00da076ac437..8f50fa456c3911424779cba31cf09a0a595429b9 100644 (file)
--- a/metainfo/piece.go
+++ b/metainfo/piece.go
@@ -1,14 +1,18 @@
 package metainfo
 
-import "github.com/anacrolix/missinggo"
+import (
+       "github.com/anacrolix/missinggo"
+)
 
 type Piece struct {
        Info *Info
-       i    int
+       i    pieceIndex
 }
 
+type pieceIndex = int
+
 func (p Piece) Length() int64 {
-       if p.i == p.Info.NumPieces()-1 {
+       if int(p.i) == p.Info.NumPieces()-1 {
                return p.Info.TotalLength() - int64(p.i)*p.Info.PieceLength
        }
        return p.Info.PieceLength
@@ -23,6 +27,6 @@ func (p Piece) Hash() (ret Hash) {
        return
 }
 
-func (p Piece) Index() int {
+func (p Piece) Index() pieceIndex {
        return p.i
 }