From 9d35da1374d7440424f2404dc7dfc249f0e68ec1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ra=C3=BAl=20Kripalani?= <raul@protocol.ai>
Date: Thu, 14 May 2020 17:16:17 +0100
Subject: [PATCH] implement Stringer for
 network.{Direction,Connectedness,Reachability}. (#150)

---
 network/network.go | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/network/network.go b/network/network.go
index 43fd011..9d20ab2 100644
--- a/network/network.go
+++ b/network/network.go
@@ -34,6 +34,14 @@ const (
 	DirOutbound
 )
 
+func (d Direction) String() string {
+	str := [...]string{"Unknown", "Inbound", "Outbound"}
+	if d < 0 || int(d) >= len(str) {
+		return "(unrecognized)"
+	}
+	return str[d]
+}
+
 // Connectedness signals the capacity for a connection with a given node.
 // It is used to signal to services and other peers whether a node is reachable.
 type Connectedness int
@@ -53,6 +61,14 @@ const (
 	CannotConnect
 )
 
+func (c Connectedness) String() string {
+	str := [...]string{"NotConnected", "Connected", "CanConnect", "CannotConnect"}
+	if c < 0 || int(c) >= len(str) {
+		return "(unrecognized)"
+	}
+	return str[c]
+}
+
 // Reachability indicates how reachable a node is.
 type Reachability int
 
@@ -72,6 +88,14 @@ const (
 	ReachabilityPrivate
 )
 
+func (r Reachability) String() string {
+	str := [...]string{"Unknown", "Public", "Private"}
+	if r < 0 || int(r) >= len(str) {
+		return "(unrecognized)"
+	}
+	return str[r]
+}
+
 // Stat stores metadata pertaining to a given Stream/Conn.
 type Stat struct {
 	Direction Direction