refactor(resource): change weights type from u32 to u64 (#1483)

Signed-off-by: Gaius <gaius.qi@gmail.com>

Author: gaius-qi

Cargo.lock

@@ -13,7 +13,7 @@ members = [
 ]
 
 [workspace.package]
-version = "1.1.1"
+version = "1.1.2"
 authors = ["The Dragonfly Developers"]
 homepage = "https://d7y.io/"
 repository = "https://github.com/dragonflyoss/client.git"
@@ -23,14 +23,14 @@ readme = "README.md"
 edition = "2021"
 
 [workspace.dependencies]
-dragonfly-client = { path = "dragonfly-client", version = "1.1.1" }
-dragonfly-client-core = { path = "dragonfly-client-core", version = "1.1.1" }
-dragonfly-client-config = { path = "dragonfly-client-config", version = "1.1.1" }
-dragonfly-client-storage = { path = "dragonfly-client-storage", version = "1.1.1" }
-dragonfly-client-backend = { path = "dragonfly-client-backend", version = "1.1.1" }
-dragonfly-client-metric = { path = "dragonfly-client-metric", version = "1.1.1" }
-dragonfly-client-util = { path = "dragonfly-client-util", version = "1.1.1" }
-dragonfly-client-init = { path = "dragonfly-client-init", version = "1.1.1" }
+dragonfly-client = { path = "dragonfly-client", version = "1.1.2" }
+dragonfly-client-core = { path = "dragonfly-client-core", version = "1.1.2" }
+dragonfly-client-config = { path = "dragonfly-client-config", version = "1.1.2" }
+dragonfly-client-storage = { path = "dragonfly-client-storage", version = "1.1.2" }
+dragonfly-client-backend = { path = "dragonfly-client-backend", version = "1.1.2" }
+dragonfly-client-metric = { path = "dragonfly-client-metric", version = "1.1.2" }
+dragonfly-client-util = { path = "dragonfly-client-util", version = "1.1.2" }
+dragonfly-client-init = { path = "dragonfly-client-init", version = "1.1.2" }
 dragonfly-api = "=2.1.81"
 thiserror = "2.0"
 futures = "0.3.31"

Cargo.toml

@@ -111,7 +111,7 @@ pub struct ParentSelector {
     id_generator: Arc<IDGenerator>,
 
     /// Maps parent host IDs to their current bandwidth weights.
-    weights: Arc<DashMap<String, u32>>,
+    weights: Arc<DashMap<String, u64>>,
 
     /// Active connections indexed by parent host ID and each connection tracks usage and manages its sync task.
     connections: Arc<DashMap<String, Connection>>,
@@ -149,7 +149,7 @@ impl ParentSelector {
     /// (better idle bandwidth) have a higher probability of being selected. If weight
     /// calculation fails, falls back to uniform random selection.
     pub fn select(&self, parents: Vec<CollectedParent>) -> CollectedParent {
-        let weights: Vec<u32> = parents
+        let weights: Vec<u64> = parents
             .iter()
             .map(|parent| {
                 let Some(parent_host) = parent.host.as_ref() else {
@@ -304,7 +304,7 @@ impl ParentSelector {
         host_id: String,
         peer_id: String,
         parent_host_id: String,
-        weights: Arc<DashMap<String, u32>>,
+        weights: Arc<DashMap<String, u64>>,
         dfdaemon_upload_client: DfdaemonUploadClient,
         mut shutdown: Shutdown,
         mut dfdaemon_shutdown: Shutdown,
@@ -332,7 +332,7 @@ impl ParentSelector {
                             let idle_tx_bandwidth = Self::get_idle_tx_bandwidth(&message);
 
                             info!("update host {} idle TX bandwidth to {}", parent_host_id, idle_tx_bandwidth);
-                            weights.insert(parent_host_id.clone(), idle_tx_bandwidth as u32);
+                            weights.insert(parent_host_id.clone(), idle_tx_bandwidth);
                         }
                         None => break,
                     }
@@ -406,7 +406,7 @@ pub struct PersistentCacheParentSelector {
     id_generator: Arc<IDGenerator>,
 
     /// Maps parent host IDs to their current bandwidth weights.
-    weights: Arc<DashMap<String, u32>>,
+    weights: Arc<DashMap<String, u64>>,
 
     /// Active connections indexed by parent host ID and each connection tracks usage and manages its sync task.
     connections: Arc<DashMap<String, Connection>>,
@@ -444,7 +444,7 @@ impl PersistentCacheParentSelector {
     /// (better idle bandwidth) have a higher probability of being selected. If weight
     /// calculation fails, falls back to uniform random selection.
     pub fn select(&self, parents: Vec<CollectedParent>) -> CollectedParent {
-        let weights: Vec<u32> = parents
+        let weights: Vec<u64> = parents
             .iter()
             .map(|parent| {
                 let Some(parent_host) = parent.host.as_ref() else {
@@ -605,7 +605,7 @@ impl PersistentCacheParentSelector {
         host_id: String,
         peer_id: String,
         parent_host_id: String,
-        weights: Arc<DashMap<String, u32>>,
+        weights: Arc<DashMap<String, u64>>,
         dfdaemon_upload_client: DfdaemonUploadClient,
         mut shutdown: Shutdown,
         mut dfdaemon_shutdown: Shutdown,
@@ -636,7 +636,7 @@ impl PersistentCacheParentSelector {
                             let idle_tx_bandwidth = Self::get_idle_tx_bandwidth(&message);
 
                             info!("update host {} idle TX bandwidth to {}", parent_host_id, idle_tx_bandwidth);
-                            weights.insert(parent_host_id.clone(), idle_tx_bandwidth as u32);
+                            weights.insert(parent_host_id.clone(), idle_tx_bandwidth);
                         }
                         None => break,
                     }