Switch to switch statement for readability

Relates to pion/webrtc#1270
pion · Jul 15, 2020 · e9599dc · e9599dc
1 parent b910e4a
commit e9599dc
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/association.go b/association.go
@@ -982,10 +982,12 @@ func (a *Association) handleHeartbeat(c *chunkHeartbeat) []*packet {
 func (a *Association) handleCookieEcho(c *chunkCookieEcho) []*packet {
 	state := a.getState()
 	a.log.Debugf("[%s] COOKIE-ECHO received in state '%s'", a.name, getAssociationStateString(state))
-	if state != established {
-		if state != closed && state != cookieWait && state != cookieEchoed {
-			return nil
-		}
+	switch state {
+	default:
+		return nil
+	case established:
+		break
+	case closed, cookieWait, cookieEchoed:
 		if !bytes.Equal(a.myCookie.cookie, c.cookie) {
 			return nil
 		}