Skip to content
This repository was archived by the owner on Oct 20, 2024. It is now read-only.

Commit 51c4b26

Browse files
authored
feat: Adds string func to nodeID (#40)
1 parent 4421c74 commit 51c4b26

File tree

2 files changed

+26
-0
lines changed

2 files changed

+26
-0
lines changed

ie/nodeId.go

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,19 @@ func NewNodeID(nodeID string) (NodeID, error) {
5555
}, nil
5656
}
5757

58+
func (n NodeID) String() string {
59+
switch n.Type {
60+
case IPv4:
61+
return net.IP(n.Value).To4().String()
62+
case IPv6:
63+
return net.IP(n.Value).To16().String()
64+
case FQDN:
65+
return string(n.Value)
66+
default:
67+
return ""
68+
}
69+
}
70+
5871
func (n NodeID) Serialize() []byte {
5972
buf := new(bytes.Buffer)
6073

ie/nodeId_test.go

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,19 @@ func TestNewNodeIDIPv4(t *testing.T) {
3838

3939
}
4040

41+
func TestString(t *testing.T) {
42+
nodeIDstring := "1.2.3.4"
43+
nodeID, err := ie.NewNodeID(nodeIDstring)
44+
if err != nil {
45+
t.Fatalf("Expected no error, got %v", err)
46+
}
47+
48+
if nodeID.String() != nodeIDstring {
49+
t.Errorf("Expected NodeID string, got %s", nodeID.String())
50+
}
51+
52+
}
53+
4154
func TestNewNodeIDIPv6(t *testing.T) {
4255
nodeID, err := ie.NewNodeID("2001:db8::68")
4356

0 commit comments

Comments
 (0)