Add String() func for NodeEventType and NodeStateType

Signed-off-by: Etienne Champetier <[email protected]>
hashicorp · Mar 15, 2021 · 51df881 · 51df881
1 parent 822dee6
commit 51df881
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/event_delegate.go b/event_delegate.go
@@ -39,6 +39,10 @@ const (
 	NodeUpdate
 )
 
+func (net NodeEventType) String() string {
+	return []string{"NodeJoin", "NodeLeave", "NodeUpdate"}[net]
+}
+
 // NodeEvent is a single event related to node activity in the memberlist.
 // The Node member of this struct must not be directly modified. It is passed
 // as a pointer to avoid unnecessary copies. If you wish to modify the node,

diff --git a/state.go b/state.go
@@ -22,6 +22,10 @@ const (
 	StateLeft
 )
 
+func (nst NodeStateType) String() string {
+	return []string{"StateAlive", "StateSuspect", "StateDead", "StateLeft"}[nst]
+}
+
 // Node represents a node in the cluster.
 type Node struct {
 	Name  string