fix(local): get ServerStatData after push_score

Author: zonyitoo

crates/shadowsocks-service/src/local/loadbalancing/ping_balancer.rs

@@ -823,14 +823,12 @@ impl PingChecker {
             ServerType::Udp => self.server.udp_score(),
         };
 
-        let score = match self.check_delay().await {
+        let (score, stat_data) = match self.check_delay().await {
             Ok(d) => server_score.push_score(Score::Latency(d)).await,
             // Penalty
             Err(..) => server_score.push_score(Score::Errored).await,
         };
 
-        let stat_data = server_score.stat_data().await;
-
         if stat_data.fail_rate > 0.5 {
             warn!(
                 "balancer: checked & updated remote {} server {} (score: {}), {:?}",

crates/shadowsocks-service/src/local/loadbalancing/server_data.rs

@@ -40,17 +40,17 @@ impl ServerScore {
     }
 
     /// Append a `Score` into statistic and recalculate score of the server
-    pub async fn push_score(&self, score: Score) -> u32 {
-        let updated_score = {
+    pub async fn push_score(&self, score: Score) -> (u32, ServerStatData) {
+        let (updated_score, data) = {
             let mut stat = self.stat_data.lock().await;
-            stat.push_score(score)
+            (stat.push_score(score), stat.data().clone())
         };
         self.score.store(updated_score, Ordering::Release);
-        updated_score
+        (updated_score, data)
     }
 
     /// Report request failure of this server, which will eventually records an `Errored` score
-    pub async fn report_failure(&self) -> u32 {
+    pub async fn report_failure(&self) -> (u32, ServerStatData) {
         self.push_score(Score::Errored).await
     }