From: Matt Joiner Date: Thu, 21 Aug 2014 11:07:44 +0000 (+1000) Subject: Move some tracker.CompactPeer into util X-Git-Tag: v1.0.0~1632 X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=cb68c984fa24b5b5438a7863ae4d287140ce303f;p=btrtrc.git Move some tracker.CompactPeer into util --- diff --git a/tracker/tracker.go b/tracker/tracker.go index 1914eae1..ffae4341 100644 --- a/tracker/tracker.go +++ b/tracker/tracker.go @@ -1,9 +1,6 @@ package tracker import ( - "bytes" - "encoding" - "encoding/binary" "errors" "net" "net/url" @@ -36,18 +33,6 @@ type Peer struct { Port int } -type CompactPeer struct { - IP [4]byte - Port uint16 -} - -var _ encoding.BinaryUnmarshaler = &CompactPeer{} - -func (cp *CompactPeer) UnmarshalBinary(b []byte) (err error) { - err = binary.Read(bytes.NewReader(b), binary.BigEndian, cp) - return -} - const ( None AnnounceEvent = iota Completed // The local peer just completed the torrent. diff --git a/util/types.go b/util/types.go index ec874622..03dcb1fa 100644 --- a/util/types.go +++ b/util/types.go @@ -1,11 +1,13 @@ package util import ( - "bitbucket.org/anacrolix/go.torrent/tracker" + "bytes" + "encoding" + "encoding/binary" "github.com/anacrolix/libtorgo/bencode" ) -type CompactPeers []tracker.CompactPeer +type CompactPeers []CompactPeer func (me *CompactPeers) UnmarshalBencode(bb []byte) (err error) { var b []byte @@ -19,7 +21,7 @@ func (me *CompactPeers) UnmarshalBencode(bb []byte) (err error) { func (me *CompactPeers) UnmarshalBinary(b []byte) (err error) { for i := 0; i < len(b); i += 6 { - var p tracker.CompactPeer + var p CompactPeer err = p.UnmarshalBinary([]byte(b[i : i+6])) if err != nil { return @@ -28,3 +30,15 @@ func (me *CompactPeers) UnmarshalBinary(b []byte) (err error) { } return } + +type CompactPeer struct { + IP [4]byte + Port uint16 +} + +var _ encoding.BinaryUnmarshaler = &CompactPeer{} + +func (cp *CompactPeer) UnmarshalBinary(b []byte) (err error) { + err = binary.Read(bytes.NewReader(b), binary.BigEndian, cp) + return +}