From 98e5c2cb46ef6bbc034510bfe4ac4bcde4617a9b Mon Sep 17 00:00:00 2001
From: yanue <yanue@outlook.com>
Date: Sat, 28 Dec 2024 11:16:54 +0800
Subject: [PATCH] =?UTF-8?q?=20=E5=AE=8C=E5=96=84=20ping=20=E5=8A=9F?=
 =?UTF-8?q?=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 V2rayU/Service/Ping/PingRunning.swift | 115 ++++++++++++++++++++++++++
 1 file changed, 115 insertions(+)
 create mode 100644 V2rayU/Service/Ping/PingRunning.swift

diff --git a/V2rayU/Service/Ping/PingRunning.swift b/V2rayU/Service/Ping/PingRunning.swift
new file mode 100644
index 0000000..458dbef
--- /dev/null
+++ b/V2rayU/Service/Ping/PingRunning.swift
@@ -0,0 +1,115 @@
+//
+//  PingRunning.swift
+//  V2rayU
+//
+//  Created by yanue on 2024/12/27.
+//
+
+import Foundation
+
+actor PingRunning {
+    static let shared = PingRunning()
+    
+    private let maxRetries = 3
+    private let maxFailures = 3
+    
+    private var failureCount = 0
+    private var isExecuting = false
+    private var item: ProfileModel = ProfileModel()
+
+    /// 开始 Ping 流程
+    func startPing(item: ProfileModel) async {
+        guard !isExecuting else {
+            print("Ping task is already running.")
+            return
+        }
+        // 替换
+        self.item = item
+        // 控制
+        isExecuting = true
+        defer { isExecuting = false }
+
+        var retries = 0
+        var success = false
+        let ping = Ping()
+        let port = getHttpProxyPort()
+        while retries < maxRetries && !success {
+            do {
+                let pingTime = try await ping.doPing(bindPort: port)
+                print("Ping success, time: \(pingTime)ms")
+                resetFailureCount()
+                success = true
+            } catch {
+                retries += 1
+                print("Ping failed (\(retries)/\(maxRetries)): \(error)")
+            }
+        }
+
+        if !success {
+            await handleFailure()
+        }
+    }
+
+    private func updateSpeed(pingTime: Int) {
+        // 更新 speed
+        ProfileViewModel.update_speed(uuid: self.item.uuid, speed: pingTime)
+    }
+    
+    /// 重置失败计数
+    private func resetFailureCount() {
+        failureCount = 0
+    }
+
+    /// 处理失败逻辑
+    private func handleFailure() async {
+        // 更新 ping 结果
+        self.updateSpeed(pingTime: -1)
+        failureCount += 1
+        if failureCount >= maxFailures {
+            failureCount = 0
+            print("Ping failed \(maxFailures) times, switching to backup...")
+            await switchServer()
+        }
+    }
+
+    /// 切换到备用服务器
+    private func switchServer() async {
+        // 实现切换逻辑，比如更新 AppState.shared.pingURL 或其他参数
+        await chooseNewServer(uuid: self.item.uuid)
+    }
+}
+
+func chooseNewServer(uuid: String) async {
+    guard UserDefaults.getBool(forKey: .autoSelectFastestServer) else {
+        NSLog(" - choose new server: disabled")
+        return
+    }
+    
+    let serverList = ProfileViewModel.all()
+    guard serverList.count > 1 else {
+        return
+    }
+
+    var pingedSvrs = [String: Int]()
+    var allSvrs = [String]()
+    
+    for svr in serverList where svr.uuid !=  uuid {
+        allSvrs.append(svr.uuid)
+        if svr.speed != -1 {
+            pingedSvrs[svr.uuid] = svr.speed
+        }
+    }
+
+    let newSvrName: String
+    if let fastestSvr = pingedSvrs.sorted(by: { $0.value < $1.value }).first {
+        newSvrName = fastestSvr.key
+    } else if let randomSvr = allSvrs.randomElement() {
+        newSvrName = randomSvr
+    } else {
+        return
+    }
+
+    NSLog(" - choose new server: \(newSvrName)")
+    UserDefaults.set(forKey: .runningProfile, value: newSvrName)
+    V2rayLaunch.restartV2ray()
+}