use enumer generate tool for state strings

replication/mainfsm.go

@@ -18,7 +18,7 @@ import (
 	"github.com/zrepl/zrepl/replication/pdu"
 )
 
-//go:generate stringer -type=State
+//go:generate enumer -type=State
 type State uint
 
 const (

replication/state_enumer.go

@@ -0,0 +1,76 @@
+// Code generated by "enumer -type=State"; DO NOT EDIT.
+
+package replication
+
+import (
+	"fmt"
+)
+
+const (
+	_StateName_0 = "PlanningPlanningError"
+	_StateName_1 = "Working"
+	_StateName_2 = "WorkingWait"
+	_StateName_3 = "Completed"
+	_StateName_4 = "ContextDone"
+)
+
+var (
+	_StateIndex_0 = [...]uint8{0, 8, 21}
+	_StateIndex_1 = [...]uint8{0, 7}
+	_StateIndex_2 = [...]uint8{0, 11}
+	_StateIndex_3 = [...]uint8{0, 9}
+	_StateIndex_4 = [...]uint8{0, 11}
+)
+
+func (i State) String() string {
+	switch {
+	case 1 <= i && i <= 2:
+		i -= 1
+		return _StateName_0[_StateIndex_0[i]:_StateIndex_0[i+1]]
+	case i == 4:
+		return _StateName_1
+	case i == 8:
+		return _StateName_2
+	case i == 16:
+		return _StateName_3
+	case i == 32:
+		return _StateName_4
+	default:
+		return fmt.Sprintf("State(%d)", i)
+	}
+}
+
+var _StateValues = []State{1, 2, 4, 8, 16, 32}
+
+var _StateNameToValueMap = map[string]State{
+	_StateName_0[0:8]:  1,
+	_StateName_0[8:21]: 2,
+	_StateName_1[0:7]:  4,
+	_StateName_2[0:11]: 8,
+	_StateName_3[0:9]:  16,
+	_StateName_4[0:11]: 32,
+}
+
+// StateString retrieves an enum value from the enum constants string name.
+// Throws an error if the param is not part of the enum.
+func StateString(s string) (State, error) {
+	if val, ok := _StateNameToValueMap[s]; ok {
+		return val, nil
+	}
+	return 0, fmt.Errorf("%s does not belong to State values", s)
+}
+
+// StateValues returns all values of the enum
+func StateValues() []State {
+	return _StateValues
+}
+
+// IsAState returns "true" if the value is listed in the enum definition. "false" otherwise
+func (i State) IsAState() bool {
+	for _, v := range _StateValues {
+		if i == v {
+			return true
+		}
+	}
+	return false
+}

replication/state_string.go

@@ -1,35 +0,0 @@
-// Code generated by "stringer -type=State"; DO NOT EDIT.
-
-package replication
-
-import "strconv"
-
-const (
-	_State_name_0 = "PlanningPlanningError"
-	_State_name_1 = "Working"
-	_State_name_2 = "WorkingWait"
-	_State_name_3 = "Completed"
-	_State_name_4 = "ContextDone"
-)
-
-var (
-	_State_index_0 = [...]uint8{0, 8, 21}
-)
-
-func (i State) String() string {
-	switch {
-	case 1 <= i && i <= 2:
-		i -= 1
-		return _State_name_0[_State_index_0[i]:_State_index_0[i+1]]
-	case i == 4:
-		return _State_name_1
-	case i == 8:
-		return _State_name_2
-	case i == 16:
-		return _State_name_3
-	case i == 32:
-		return _State_name_4
-	default:
-		return "State(" + strconv.FormatInt(int64(i), 10) + ")"
-	}
-}