From 5ed71dad8884a29eca2c1dcbb63483f5592ff8b4 Mon Sep 17 00:00:00 2001
From: sta <sta.blockhead@gmail.com>
Date: Wed, 18 Mar 2020 19:46:15 +0900
Subject: [PATCH] [Modify] Polish it

---
 websocket-sharp/Net/WebHeaderCollection.cs | 23 +++++++++++++++-------
 1 file changed, 16 insertions(+), 7 deletions(-)

diff --git a/websocket-sharp/Net/WebHeaderCollection.cs b/websocket-sharp/Net/WebHeaderCollection.cs
index 67f98493..4189c9b9 100644
--- a/websocket-sharp/Net/WebHeaderCollection.cs
+++ b/websocket-sharp/Net/WebHeaderCollection.cs
@@ -876,15 +876,24 @@ namespace WebSocketSharp.Net
     }
 
     private void doWithCheckingState (
-      Action <string, string> action, string name, string value, bool setState)
+      Action <string, string> action, string name, string value, bool setState
+    )
     {
-      var type = checkHeaderType (name);
-      if (type == HttpHeaderType.Request)
-        doWithCheckingState (action, name, value, false, setState);
-      else if (type == HttpHeaderType.Response)
+      var headerType = checkHeaderType (name);
+
+      if (headerType == HttpHeaderType.Response) {
         doWithCheckingState (action, name, value, true, setState);
-      else
-        action (name, value);
+
+        return;
+      }
+
+      if (headerType == HttpHeaderType.Request) {
+        doWithCheckingState (action, name, value, false, setState);
+
+        return;
+      }
+
+      action (name, value);
     }
 
     private void doWithCheckingState (