mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-09-15 05:50:10 +00:00
8a4ece1101
* [NOD-1223] Move all network stuff into a new network package. * [NOD-1223] Delete the unused package testutil. * [NOD-1223] Move infrastructure stuff into a new instrastructure package. * [NOD-1223] Move domain stuff into a new domain package.
71 lines
1.4 KiB
Go
71 lines
1.4 KiB
Go
package id
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/rand"
|
|
"encoding/hex"
|
|
"io"
|
|
)
|
|
|
|
// IDLength of array used to store the ID.
|
|
const IDLength = 16
|
|
|
|
// ID identifies a network connection
|
|
type ID struct {
|
|
bytes []byte
|
|
}
|
|
|
|
// GenerateID generates a new ID
|
|
func GenerateID() (*ID, error) {
|
|
id := new(ID)
|
|
err := id.Deserialize(rand.Reader)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return id, nil
|
|
}
|
|
|
|
// IsEqual returns whether id equals to other.
|
|
func (id *ID) IsEqual(other *ID) bool {
|
|
return bytes.Equal(id.bytes, other.bytes)
|
|
}
|
|
|
|
func (id *ID) String() string {
|
|
return hex.EncodeToString(id.bytes)
|
|
}
|
|
|
|
// Deserialize decodes a block from r into the receiver.
|
|
func (id *ID) Deserialize(r io.Reader) error {
|
|
id.bytes = make([]byte, IDLength)
|
|
_, err := io.ReadFull(r, id.bytes)
|
|
return err
|
|
}
|
|
|
|
// Serialize serializes the receiver into the given writer.
|
|
func (id *ID) Serialize(w io.Writer) error {
|
|
_, err := w.Write(id.bytes)
|
|
return err
|
|
}
|
|
|
|
// SerializeToBytes serializes the receiver and returns a byte slice of the serialized id.
|
|
func (id *ID) SerializeToBytes() ([]byte, error) {
|
|
w := &bytes.Buffer{}
|
|
err := id.Serialize(w)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return w.Bytes(), nil
|
|
}
|
|
|
|
// FromBytes returns an ID deserialized from the given byte slice.
|
|
func FromBytes(serializedID []byte) *ID {
|
|
r := bytes.NewReader(serializedID)
|
|
newID := new(ID)
|
|
err := newID.Deserialize(r)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return newID
|
|
}
|