fix task group

Sources/Punchnet/SDLContext.swift

@@ -120,7 +120,7 @@ public class SDLContext: @unchecked Sendable {
                     do {
                         try await self.startSuperClient()
                     } catch let err {
-                        NSLog("SuperClient get error: \(err)")
+                        NSLog("[SDLContext] SuperClient get error: \(err), will restart")
                         await self.arpServer.clear()
                         try? await Task.sleep(for: .seconds(2))
                     }

Sources/Punchnet/SDLSuperClient.swift

@@ -9,8 +9,6 @@ import Foundation
 import NIOCore
 import NIOPosix
 
-
-
 // --MARK: 和SuperNode的客户端
 actor SDLSuperClient {
     private let group = MultiThreadedEventLoopGroup(numberOfThreads: 1)
@@ -69,6 +67,10 @@ actor SDLSuperClient {
                 }
                 
                 group.addTask {
+                    defer {
+                        self.inboundContinuation.finish()
+                    }
+                    
                     for try await var packet in inbound {
                         if let message = SDLSuperClientDecoder.decode(buffer: &packet) {
                             SDLLogger.log("[SDLSuperTransport] read message: \(message)", level: .warning)
@@ -82,9 +84,14 @@ actor SDLSuperClient {
                             }
                         }
                     }
+                    NSLog("[SDLSuperClient] inbound closed")
                 }
                 
                 group.addTask {
+                    defer {
+                        self.writeContinuation.finish()
+                    }
+                    
                     for try await message in self.writeStream {
                         var buffer = self.asyncChannel.channel.allocator.buffer(capacity: message.data.count + 5)
                         buffer.writeInteger(message.packetId, as: UInt32.self)
@@ -92,17 +99,28 @@ actor SDLSuperClient {
                         buffer.writeBytes(message.data)
                         try await outbound.write(buffer)
                     }
+                    NSLog("[SDLSuperClient] outbound closed")
                 }
                 
                 // --MARK: 心跳机制
                 group.addTask {
-                    while !Task.isCancelled {
+                    while true {
+                        do {
                             await self.ping()
-                        try? await Task.sleep(nanoseconds: 5 * 1_000_000_000)
+                            try await Task.sleep(nanoseconds: 5 * 1_000_000_000)
+                        } catch let err {
+                            NSLog("[SDLSuperClient] heartbeat cancelled with error: \(err)")
+                            break
+                        }
                     }
                 }
                 
-                try await group.waitForAll()
+                // 迭代等待所有任务的退出, 第一个异常会被抛出
+                for try await _ in group {
+                    
+                }
+                
+                NSLog("[SDLSuperClient] group closed")
             }
         }
     }
@@ -182,7 +200,6 @@ actor SDLSuperClient {
     
     deinit {
         try! group.syncShutdownGracefully()
-        self.inboundContinuation.finish()
     }
     
 }

Sources/Punchnet/SDLUDPHole.swift

@@ -100,7 +100,11 @@ actor SDLUDPHole {
                     }
                 }
                 
-                try await group.waitForAll()
+                for try await _ in group {
+                    
+                }
+                
+                SDLLogger.log("[SDLUDPHole] group closed", level: .debug)
             }
         }
     }