Merge pull request #99 from pushkarnk/sync-keep-alive

Synchronize access to the keepAliveState property
Kitura · Oct 15, 2018 · 214abc9 · 214abc9
2 parents 0d79b8d + d37ef06
commit 214abc9
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/Sources/KituraNet/HTTP/HTTPRequestHandler.swift b/Sources/KituraNet/HTTP/HTTPRequestHandler.swift
@@ -35,8 +35,24 @@ internal class HTTPRequestHandler: ChannelInboundHandler {
     /// We'd want to send an error response only once
     var errorResponseSent = false
 
-    var keepAliveState: KeepAliveState = .unlimited
-
+    var keepAliveState: KeepAliveState {
+        set {
+            self.syncQueue.sync {
+                _keepAliveState = newValue
+            }
+        }
+
+        get {
+            return self.syncQueue.sync {
+                return _keepAliveState
+            }
+        }
+    }
+
+    private let syncQueue = DispatchQueue(label: "HTTPServer.keepAliveSync")
+
+    private var _keepAliveState: KeepAliveState = .unlimited
+
     static let keepAliveTimeout: TimeInterval = 60
 
     private(set) var clientRequestedKeepAlive = false