From 39decb03c0d3ff44f08cc4a587813e771f2d853b Mon Sep 17 00:00:00 2001 From: Dimitri Sokolyuk Date: Thu, 16 Jun 2016 02:54:33 +0200 Subject: split --- peer/id.go | 16 ++++++++++++++++ peer/id_test.go | 14 ++++++++++++++ peer/messages.go | 15 --------------- peer/messages_test.go | 14 -------------- 4 files changed, 30 insertions(+), 29 deletions(-) create mode 100644 peer/id.go create mode 100644 peer/id_test.go delete mode 100644 peer/messages_test.go diff --git a/peer/id.go b/peer/id.go new file mode 100644 index 0000000..197da5a --- /dev/null +++ b/peer/id.go @@ -0,0 +1,16 @@ +package peer + +import ( + "crypto/rand" + "fmt" + "math/big" +) + +func NewID() (string, error) { + limit := new(big.Int).Lsh(big.NewInt(1), 80) + id, err := rand.Int(rand.Reader, limit) + if err != nil { + return "", err + } + return fmt.Sprintf("%0.20x", id), nil +} diff --git a/peer/id_test.go b/peer/id_test.go new file mode 100644 index 0000000..da80656 --- /dev/null +++ b/peer/id_test.go @@ -0,0 +1,14 @@ +package peer + +import "testing" + +func TestID(t *testing.T) { + id, err := NewID() + if err != nil { + t.Error(err) + } + if len(id) != 20 { + t.Error("wrong length") + } + t.Log(id) +} diff --git a/peer/messages.go b/peer/messages.go index 1ddaf83..a14de46 100644 --- a/peer/messages.go +++ b/peer/messages.go @@ -1,11 +1,5 @@ package peer -import ( - "crypto/rand" - "fmt" - "math/big" -) - type Message int const ( @@ -23,12 +17,3 @@ const ( ) const Proto = `BitTorrent protocol` - -func NewID() (string, error) { - limit := new(big.Int).Lsh(big.NewInt(1), 80) - id, err := rand.Int(rand.Reader, limit) - if err != nil { - return "", err - } - return fmt.Sprintf("%0.20x", id), nil -} diff --git a/peer/messages_test.go b/peer/messages_test.go deleted file mode 100644 index da80656..0000000 --- a/peer/messages_test.go +++ /dev/null @@ -1,14 +0,0 @@ -package peer - -import "testing" - -func TestID(t *testing.T) { - id, err := NewID() - if err != nil { - t.Error(err) - } - if len(id) != 20 { - t.Error("wrong length") - } - t.Log(id) -} -- cgit v1.2.3