Kaynağa Gözat

Merge pull request #281 from sbarow/fix_test_coverage

Move while loop read to own function
Damian Kołakowski 8 yıl önce
ebeveyn
işleme
fdcafa2eb6
1 değiştirilmiş dosya ile 5 ekleme ve 1 silme
  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: