ソースを参照

Merge pull request #281 from sbarow/fix_test_coverage

Move while loop read to own function
Damian Kołakowski 8 年 前
コミット
fdcafa2eb6
1 ファイル変更5 行追加1 行削除
  1. 5 1
      Sources/WebSockets.swift

+ 5 - 1
Sources/WebSockets.swift

@@ -98,11 +98,15 @@ public func websocket(
                 }
             }
             
-            do {
+            func read() throws {
                 while true {
                     let frame = try session.readFrame()
                     try handleOperationCode(frame)
                 }
+            }
+            
+            do {
+                try read()
             } catch let error {
                 switch error {
                 case WebSocketSession.Control.close: