Use switch

yuuki · Mar 4, 2018 · eec0997 · eec0997
1 parent 2374cdc
commit eec0997
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/conntrack/conntrack.go b/conntrack/conntrack.go
@@ -89,9 +89,10 @@ func (f *HostFlow) HasDirection(dire FlowDirection) bool {
 
 // String returns the string representation of HostFlow.
 func (f *HostFlow) String() string {
-	if f.direction == FlowActive {
+	switch f.direction {
+	case FlowActive:
 		return fmt.Sprintf("localhost:many\t --> \t%s:%s \t%s", f.addr, f.port, f.stat)
-	} else if f.direction == FlowPassive {
+	case FlowPassive:
 		return fmt.Sprintf("localhost:%s\t <-- \t%s:many \t%s", f.port, f.addr, f.stat)
 	}
 	return ""