1use serde::{Deserialize, Serialize};
7use std::collections::{BTreeMap, HashMap};
8
9#[derive(Debug, Clone, Serialize, Deserialize, Default)]
11#[serde(rename_all = "camelCase")]
12pub struct ServerInfo {
13 #[serde(default)]
15 pub state: String,
16
17 #[serde(default)]
19 pub endpoint: String,
20
21 #[serde(default)]
23 pub scheme: String,
24
25 #[serde(default)]
27 pub uptime: u64,
28
29 #[serde(default)]
31 pub version: String,
32
33 #[serde(default, rename = "commitID")]
35 pub commit_id: String,
36
37 #[serde(default)]
39 pub network: HashMap<String, String>,
40
41 #[serde(default, rename = "drives")]
43 pub disks: Vec<DiskInfo>,
44
45 #[serde(default, rename = "poolNumber")]
47 pub pool_number: i32,
48
49 #[serde(default, rename = "mem_stats")]
51 pub mem_stats: MemStats,
52}
53
54#[derive(Debug, Clone, Serialize, Deserialize, Default)]
56#[serde(rename_all = "camelCase")]
57pub struct DiskInfo {
58 #[serde(default)]
60 pub endpoint: String,
61
62 #[serde(default, rename = "rootDisk")]
64 pub root_disk: bool,
65
66 #[serde(default, rename = "path")]
68 pub drive_path: String,
69
70 #[serde(default)]
72 pub healing: bool,
73
74 #[serde(default)]
76 pub scanning: bool,
77
78 #[serde(default)]
80 pub state: String,
81
82 #[serde(default)]
84 pub uuid: String,
85
86 #[serde(default, rename = "totalspace")]
88 pub total_space: u64,
89
90 #[serde(default, rename = "usedspace")]
92 pub used_space: u64,
93
94 #[serde(default, rename = "availspace")]
96 pub available_space: u64,
97
98 #[serde(default, alias = "pool_index")]
100 pub pool_index: i32,
101
102 #[serde(default, alias = "set_index")]
104 pub set_index: i32,
105
106 #[serde(default, alias = "disk_index")]
108 pub disk_index: i32,
109
110 #[serde(default, skip_serializing_if = "Option::is_none")]
112 pub heal_info: Option<HealingDiskInfo>,
113}
114
115#[derive(Debug, Clone, Serialize, Deserialize, Default)]
117#[serde(rename_all = "camelCase")]
118pub struct HealingDiskInfo {
119 #[serde(default)]
121 pub id: String,
122
123 #[serde(default)]
125 pub heal_id: String,
126
127 #[serde(default)]
129 pub pool_index: Option<usize>,
130
131 #[serde(default)]
133 pub set_index: Option<usize>,
134
135 #[serde(default)]
137 pub disk_index: Option<usize>,
138
139 #[serde(default)]
141 pub endpoint: String,
142
143 #[serde(default)]
145 pub path: String,
146
147 #[serde(default)]
149 pub objects_total_count: u64,
150
151 #[serde(default)]
153 pub objects_total_size: u64,
154
155 #[serde(default)]
157 pub items_healed: u64,
158
159 #[serde(default)]
161 pub items_failed: u64,
162
163 #[serde(default)]
165 pub bytes_done: u64,
166
167 #[serde(default)]
169 pub finished: bool,
170
171 #[serde(default)]
173 pub bucket: String,
174
175 #[serde(default)]
177 pub object: String,
178}
179
180#[derive(Debug, Clone, Serialize, Deserialize, Default)]
182pub struct MemStats {
183 #[serde(default)]
185 pub alloc: u64,
186
187 #[serde(default)]
189 pub total_alloc: u64,
190
191 #[serde(default)]
193 pub heap_alloc: u64,
194}
195
196#[derive(Debug, Clone, Serialize, Deserialize, Default)]
198#[serde(rename_all = "lowercase")]
199pub enum BackendType {
200 #[default]
202 #[serde(rename = "FS")]
203 Fs,
204 #[serde(rename = "Erasure")]
206 Erasure,
207}
208
209impl std::fmt::Display for BackendType {
210 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
211 match self {
212 BackendType::Fs => write!(f, "FS"),
213 BackendType::Erasure => write!(f, "Erasure"),
214 }
215 }
216}
217
218#[derive(Debug, Clone, Serialize, Deserialize, Default)]
220#[serde(rename_all = "camelCase")]
221pub struct BackendInfo {
222 #[serde(default, rename = "backendType")]
224 pub backend_type: BackendType,
225
226 #[serde(default, rename = "onlineDisks")]
228 pub online_disks: usize,
229
230 #[serde(default, rename = "offlineDisks")]
232 pub offline_disks: usize,
233
234 #[serde(default, rename = "standardSCParity")]
236 pub standard_sc_parity: Option<usize>,
237
238 #[serde(default, rename = "rrSCParity")]
240 pub rr_sc_parity: Option<usize>,
241
242 #[serde(default, rename = "totalSets")]
244 pub total_sets: Vec<usize>,
245
246 #[serde(default, rename = "totalDrivesPerSet")]
248 pub drives_per_set: Vec<usize>,
249}
250
251#[derive(Debug, Clone, Serialize, Deserialize, Default)]
253pub struct UsageInfo {
254 #[serde(default)]
256 pub size: u64,
257
258 #[serde(default, skip_serializing_if = "Option::is_none")]
260 pub error: Option<String>,
261}
262
263#[derive(Debug, Clone, Serialize, Deserialize, Default)]
265pub struct BucketsInfo {
266 #[serde(default)]
268 pub count: u64,
269
270 #[serde(default, skip_serializing_if = "Option::is_none")]
272 pub error: Option<String>,
273}
274
275#[derive(Debug, Clone, Serialize, Deserialize, Default)]
277pub struct ObjectsInfo {
278 #[serde(default)]
280 pub count: u64,
281
282 #[serde(default, skip_serializing_if = "Option::is_none")]
284 pub error: Option<String>,
285}
286
287#[derive(Debug, Clone, Serialize, Deserialize, Default)]
289#[serde(rename_all = "camelCase")]
290pub struct PoolErasureSetInfo {
291 #[serde(default)]
293 pub id: i32,
294
295 #[serde(default, rename = "rawUsage")]
297 pub raw_usage: u64,
298
299 #[serde(default, rename = "rawCapacity")]
301 pub raw_capacity: u64,
302
303 #[serde(default)]
305 pub usage: u64,
306
307 #[serde(default, rename = "objectsCount")]
309 pub objects_count: u64,
310
311 #[serde(default, rename = "versionsCount")]
313 pub versions_count: u64,
314
315 #[serde(default, rename = "deleteMarkersCount")]
317 pub delete_markers_count: u64,
318
319 #[serde(default, rename = "healDisks")]
321 pub heal_disks: i32,
322}
323
324#[derive(Debug, Clone, Serialize, Deserialize, Default)]
326#[serde(rename_all = "camelCase")]
327pub struct ClusterInfo {
328 #[serde(default)]
330 pub mode: Option<String>,
331
332 #[serde(default)]
334 pub domain: Option<Vec<String>>,
335
336 #[serde(default)]
338 pub region: Option<String>,
339
340 #[serde(default, rename = "deploymentID")]
342 pub deployment_id: Option<String>,
343
344 #[serde(default)]
346 pub buckets: Option<BucketsInfo>,
347
348 #[serde(default)]
350 pub objects: Option<ObjectsInfo>,
351
352 #[serde(default)]
354 pub usage: Option<UsageInfo>,
355
356 #[serde(default)]
358 pub backend: Option<BackendInfo>,
359
360 #[serde(default)]
362 pub servers: Option<Vec<ServerInfo>>,
363
364 #[serde(default)]
366 pub pools: Option<BTreeMap<i32, BTreeMap<i32, PoolErasureSetInfo>>>,
367}
368
369impl ClusterInfo {
370 pub fn online_disks(&self) -> usize {
372 self.servers
373 .as_ref()
374 .map(|servers| {
375 servers
376 .iter()
377 .flat_map(|s| &s.disks)
378 .filter(|d| d.state == "online" || d.state == "ok")
379 .count()
380 })
381 .unwrap_or(0)
382 }
383
384 pub fn offline_disks(&self) -> usize {
386 self.servers
387 .as_ref()
388 .map(|servers| {
389 servers
390 .iter()
391 .flat_map(|s| &s.disks)
392 .filter(|d| d.state == "offline")
393 .count()
394 })
395 .unwrap_or(0)
396 }
397
398 pub fn total_capacity(&self) -> u64 {
400 self.servers
401 .as_ref()
402 .map(|servers| {
403 servers
404 .iter()
405 .flat_map(|s| &s.disks)
406 .map(|d| d.total_space)
407 .sum()
408 })
409 .unwrap_or(0)
410 }
411
412 pub fn used_capacity(&self) -> u64 {
414 self.servers
415 .as_ref()
416 .map(|servers| {
417 servers
418 .iter()
419 .flat_map(|s| &s.disks)
420 .map(|d| d.used_space)
421 .sum()
422 })
423 .unwrap_or(0)
424 }
425}
426
427#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize, Default)]
429#[serde(rename_all = "lowercase")]
430pub enum HealScanMode {
431 #[default]
433 Normal,
434 Deep,
436}
437
438impl std::fmt::Display for HealScanMode {
439 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
440 match self {
441 HealScanMode::Normal => write!(f, "normal"),
442 HealScanMode::Deep => write!(f, "deep"),
443 }
444 }
445}
446
447impl std::str::FromStr for HealScanMode {
448 type Err = String;
449
450 fn from_str(s: &str) -> Result<Self, Self::Err> {
451 match s.to_lowercase().as_str() {
452 "normal" => Ok(HealScanMode::Normal),
453 "deep" => Ok(HealScanMode::Deep),
454 _ => Err(format!("Invalid heal scan mode: {s}")),
455 }
456 }
457}
458
459#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
460#[serde(rename_all = "lowercase")]
461pub enum HealRuntimeState {
462 Disabled,
463 Uninitialized,
464 Idle,
465 Active,
466 #[serde(other)]
467 Unknown,
468}
469
470#[derive(Debug, Clone, Serialize, Deserialize, Default)]
472#[serde(rename_all = "camelCase")]
473pub struct HealStartRequest {
474 #[serde(default, skip_serializing_if = "Option::is_none")]
476 pub bucket: Option<String>,
477
478 #[serde(default, skip_serializing_if = "Option::is_none")]
480 pub prefix: Option<String>,
481
482 #[serde(default)]
484 pub scan_mode: HealScanMode,
485
486 #[serde(default)]
488 pub remove: bool,
489
490 #[serde(default)]
492 pub recreate: bool,
493
494 #[serde(default)]
496 pub dry_run: bool,
497}
498
499#[derive(Debug, Clone, Serialize, Deserialize)]
501#[serde(rename_all = "camelCase")]
502pub struct HealTaskRequest {
503 pub bucket: String,
505
506 #[serde(default, skip_serializing_if = "Option::is_none")]
508 pub prefix: Option<String>,
509
510 pub client_token: String,
512}
513
514#[derive(Debug, Clone, Serialize, Deserialize, Default)]
516pub struct HealDriveInfo {
517 #[serde(default)]
519 pub uuid: String,
520
521 #[serde(default)]
523 pub endpoint: String,
524
525 #[serde(default)]
527 pub state: String,
528}
529
530#[derive(Debug, Clone, Serialize, Deserialize, Default)]
532#[serde(rename_all = "camelCase")]
533pub struct HealResultItem {
534 #[serde(default, rename = "resultId")]
536 pub result_index: usize,
537
538 #[serde(default, rename = "type")]
540 pub item_type: String,
541
542 #[serde(default)]
544 pub bucket: String,
545
546 #[serde(default)]
548 pub object: String,
549
550 #[serde(default, rename = "versionId")]
552 pub version_id: String,
553
554 #[serde(default)]
556 pub detail: String,
557
558 #[serde(default, rename = "parityBlocks")]
560 pub parity_blocks: usize,
561
562 #[serde(default, rename = "dataBlocks")]
564 pub data_blocks: usize,
565
566 #[serde(default, rename = "objectSize")]
568 pub object_size: u64,
569
570 #[serde(default)]
572 pub before: HealDriveInfos,
573
574 #[serde(default)]
576 pub after: HealDriveInfos,
577}
578
579#[derive(Debug, Clone, Serialize, Deserialize, Default)]
581pub struct HealDriveInfos {
582 #[serde(default)]
584 pub drives: Vec<HealDriveInfo>,
585}
586
587#[derive(Debug, Clone, Serialize, Deserialize, Default)]
589#[serde(rename_all = "camelCase")]
590pub struct HealStatus {
591 #[serde(default)]
593 pub heal_id: String,
594
595 #[serde(default)]
597 pub healing: bool,
598
599 #[serde(default, skip_serializing_if = "Option::is_none")]
600 pub state: Option<HealRuntimeState>,
601
602 #[serde(default, skip_serializing_if = "Option::is_none")]
604 pub summary: Option<String>,
605
606 #[serde(default, skip_serializing_if = "Option::is_none")]
608 pub detail: Option<String>,
609
610 #[serde(default)]
612 pub bucket: String,
613
614 #[serde(default)]
616 pub object: String,
617
618 #[serde(default, skip_serializing_if = "Option::is_none")]
620 pub scan_mode: Option<HealScanMode>,
621
622 #[serde(default)]
624 pub scan_cycle: u64,
625
626 #[serde(default)]
628 pub heal_queue_length: u64,
629
630 #[serde(default)]
632 pub heal_active_tasks: u64,
633
634 #[serde(default)]
636 pub items_scanned: u64,
637
638 #[serde(default)]
640 pub items_healed: u64,
641
642 #[serde(default)]
644 pub items_failed: u64,
645
646 #[serde(default)]
648 pub bytes_scanned: u64,
649
650 #[serde(default)]
652 pub bytes_healed: u64,
653
654 #[serde(default)]
656 pub started: Option<String>,
657
658 #[serde(default)]
660 pub last_update: Option<String>,
661}
662
663#[derive(Debug, Clone, Serialize, Deserialize, Default)]
665#[serde(rename_all = "camelCase")]
666pub struct PoolTarget {
667 pub pool: String,
669
670 #[serde(default)]
672 pub by_id: bool,
673}
674
675#[derive(Debug, Clone, Serialize, Deserialize, Default)]
677pub struct PoolStatus {
678 #[serde(default)]
680 pub id: usize,
681
682 #[serde(default, rename = "cmdline")]
684 pub cmd_line: String,
685
686 #[serde(default, rename = "lastUpdate")]
688 pub last_update: String,
689
690 #[serde(default)]
692 pub status: String,
693
694 #[serde(default, rename = "decommissionStatus")]
696 pub decommission_status: String,
697
698 #[serde(default, rename = "rebalanceStatus")]
700 pub rebalance_status: String,
701
702 #[serde(default, rename = "totalSize")]
704 pub total_size: u64,
705
706 #[serde(default, rename = "currentSize")]
708 pub current_size: u64,
709
710 #[serde(default, rename = "usedSize")]
712 pub used_size: u64,
713
714 #[serde(default)]
716 pub used: f64,
717
718 #[serde(default, rename = "decommissionInfo")]
720 pub decommission: Option<PoolDecommissionInfo>,
721}
722
723#[derive(Debug, Clone, Serialize, Deserialize, Default)]
725pub struct DecommissionStatus {
726 #[serde(default)]
728 pub pools: Vec<DecommissionPoolStatus>,
729}
730
731#[derive(Debug, Clone, Serialize, Deserialize, Default)]
733pub struct DecommissionPoolStatus {
734 #[serde(default)]
736 pub id: usize,
737
738 #[serde(default, rename = "cmdline")]
740 pub cmd_line: String,
741
742 #[serde(default)]
744 pub status: String,
745
746 #[serde(default, rename = "poolStatus")]
748 pub pool_status: String,
749
750 #[serde(default, rename = "decommissionInfo")]
752 pub decommission: Option<PoolDecommissionInfo>,
753}
754
755#[derive(Debug, Clone, Serialize, Deserialize, Default)]
757pub struct PoolDecommissionInfo {
758 #[serde(default, rename = "startTime")]
760 pub start_time: Option<String>,
761
762 #[serde(default, rename = "startSize")]
764 pub start_size: u64,
765
766 #[serde(default, rename = "totalSize")]
768 pub total_size: u64,
769
770 #[serde(default, rename = "currentSize")]
772 pub current_size: u64,
773
774 #[serde(default)]
776 pub complete: bool,
777
778 #[serde(default)]
780 pub failed: bool,
781
782 #[serde(default)]
784 pub canceled: bool,
785
786 #[serde(default)]
788 pub queued: bool,
789
790 #[serde(default, rename = "queuedBuckets")]
792 pub queued_buckets: Vec<String>,
793
794 #[serde(default, rename = "decommissionedBuckets")]
796 pub decommissioned_buckets: Vec<String>,
797
798 #[serde(default)]
800 pub bucket: String,
801
802 #[serde(default)]
804 pub prefix: String,
805
806 #[serde(default)]
808 pub object: String,
809
810 #[serde(default)]
812 pub stage: String,
813
814 #[serde(default, rename = "objectsDecommissioned")]
816 pub objects_decommissioned: u64,
817
818 #[serde(default, rename = "objectsDecommissionedFailed")]
820 pub objects_decommissioned_failed: u64,
821
822 #[serde(default, rename = "bytesDecommissioned")]
824 pub bytes_decommissioned: u64,
825
826 #[serde(default, rename = "bytesDecommissionedFailed")]
828 pub bytes_decommissioned_failed: u64,
829
830 #[serde(default, rename = "waitingReason")]
832 pub waiting_reason: Option<String>,
833}
834
835#[derive(Debug, Clone, Serialize, Deserialize, Default)]
837pub struct RebalanceStartResult {
838 #[serde(default)]
840 pub id: String,
841}
842
843#[derive(Debug, Clone, Serialize, Deserialize, Default)]
845pub struct RebalanceStatus {
846 #[serde(default)]
848 pub id: String,
849
850 #[serde(default)]
852 pub pools: Vec<RebalancePoolStatus>,
853
854 #[serde(default, rename = "stoppedAt")]
856 pub stopped_at: Option<String>,
857}
858
859#[derive(Debug, Clone, Serialize, Deserialize, Default)]
861pub struct RebalancePoolStatus {
862 #[serde(default)]
864 pub id: usize,
865
866 #[serde(default)]
868 pub status: String,
869
870 #[serde(default)]
872 pub used: f64,
873
874 #[serde(default, rename = "lastError")]
876 pub last_error: Option<String>,
877
878 #[serde(default, rename = "cleanupWarnings")]
880 pub cleanup_warnings: RebalanceCleanupWarnings,
881
882 #[serde(default)]
884 pub progress: Option<RebalancePoolProgress>,
885}
886
887#[derive(Debug, Clone, Serialize, Deserialize, Default)]
889pub struct RebalanceCleanupWarnings {
890 #[serde(default)]
892 pub count: u64,
893
894 #[serde(default, rename = "lastMsg")]
896 pub last_message: Option<String>,
897
898 #[serde(default, rename = "lastBucket")]
900 pub last_bucket: Option<String>,
901
902 #[serde(default, rename = "lastObject")]
904 pub last_object: Option<String>,
905
906 #[serde(default, rename = "lastAt")]
908 pub last_at: Option<String>,
909}
910
911#[derive(Debug, Clone, Serialize, Deserialize, Default)]
913pub struct RebalancePoolProgress {
914 #[serde(default, rename = "objects")]
916 pub num_objects: u64,
917
918 #[serde(default, rename = "versions")]
920 pub num_versions: u64,
921
922 #[serde(default)]
924 pub bytes: u64,
925
926 #[serde(default, rename = "remainingBuckets")]
928 pub remaining_buckets: usize,
929
930 #[serde(default)]
932 pub bucket: String,
933
934 #[serde(default)]
936 pub object: String,
937
938 #[serde(default)]
940 pub elapsed: u64,
941
942 #[serde(default)]
944 pub eta: u64,
945}
946
947#[cfg(test)]
948mod tests {
949 use super::*;
950
951 #[test]
952 fn test_backend_type_display() {
953 assert_eq!(BackendType::Fs.to_string(), "FS");
954 assert_eq!(BackendType::Erasure.to_string(), "Erasure");
955 }
956
957 #[test]
958 fn test_heal_scan_mode_display() {
959 assert_eq!(HealScanMode::Normal.to_string(), "normal");
960 assert_eq!(HealScanMode::Deep.to_string(), "deep");
961 }
962
963 #[test]
964 fn test_heal_scan_mode_from_str() {
965 assert_eq!(
966 "normal".parse::<HealScanMode>().unwrap(),
967 HealScanMode::Normal
968 );
969 assert_eq!("deep".parse::<HealScanMode>().unwrap(), HealScanMode::Deep);
970 assert!("invalid".parse::<HealScanMode>().is_err());
971 }
972
973 #[test]
974 fn test_cluster_info_default() {
975 let info = ClusterInfo::default();
976 assert!(info.mode.is_none());
977 assert!(info.servers.is_none());
978 assert_eq!(info.online_disks(), 0);
979 assert_eq!(info.offline_disks(), 0);
980 }
981
982 #[test]
983 fn test_cluster_info_disk_counts() {
984 let info = ClusterInfo {
985 servers: Some(vec![ServerInfo {
986 disks: vec![
987 DiskInfo {
988 state: "online".to_string(),
989 ..Default::default()
990 },
991 DiskInfo {
992 state: "online".to_string(),
993 ..Default::default()
994 },
995 DiskInfo {
996 state: "offline".to_string(),
997 ..Default::default()
998 },
999 ],
1000 ..Default::default()
1001 }]),
1002 ..Default::default()
1003 };
1004
1005 assert_eq!(info.online_disks(), 2);
1006 assert_eq!(info.offline_disks(), 1);
1007 }
1008
1009 #[test]
1010 fn test_cluster_info_capacity() {
1011 let info = ClusterInfo {
1012 servers: Some(vec![ServerInfo {
1013 disks: vec![
1014 DiskInfo {
1015 total_space: 1000,
1016 used_space: 300,
1017 ..Default::default()
1018 },
1019 DiskInfo {
1020 total_space: 2000,
1021 used_space: 500,
1022 ..Default::default()
1023 },
1024 ],
1025 ..Default::default()
1026 }]),
1027 ..Default::default()
1028 };
1029
1030 assert_eq!(info.total_capacity(), 3000);
1031 assert_eq!(info.used_capacity(), 800);
1032 }
1033
1034 #[test]
1035 fn test_disk_info_default() {
1036 let disk = DiskInfo::default();
1037 assert!(disk.endpoint.is_empty());
1038 assert!(!disk.healing);
1039 assert!(!disk.scanning);
1040 assert_eq!(disk.total_space, 0);
1041 }
1042
1043 #[test]
1044 fn test_disk_info_deserializes_snake_case_location_indexes() {
1045 let json = r#"{"pool_index":1,"set_index":2,"disk_index":3}"#;
1046
1047 let disk: DiskInfo = serde_json::from_str(json).unwrap();
1048
1049 assert_eq!(disk.pool_index, 1);
1050 assert_eq!(disk.set_index, 2);
1051 assert_eq!(disk.disk_index, 3);
1052 }
1053
1054 #[test]
1055 fn test_server_info_default() {
1056 let server = ServerInfo::default();
1057 assert!(server.state.is_empty());
1058 assert!(server.endpoint.is_empty());
1059 assert_eq!(server.uptime, 0);
1060 }
1061
1062 #[test]
1063 fn test_heal_start_request_default() {
1064 let req = HealStartRequest::default();
1065 assert!(req.bucket.is_none());
1066 assert!(req.prefix.is_none());
1067 assert_eq!(req.scan_mode, HealScanMode::Normal);
1068 assert!(!req.remove);
1069 assert!(!req.dry_run);
1070 }
1071
1072 #[test]
1073 fn test_heal_status_default() {
1074 let status = HealStatus::default();
1075 assert!(status.heal_id.is_empty());
1076 assert!(!status.healing);
1077 assert!(status.state.is_none());
1078 assert!(status.scan_mode.is_none());
1079 assert_eq!(status.scan_cycle, 0);
1080 assert_eq!(status.heal_queue_length, 0);
1081 assert_eq!(status.heal_active_tasks, 0);
1082 assert_eq!(status.items_scanned, 0);
1083 }
1084
1085 #[test]
1086 fn test_heal_runtime_state_unknown_value_is_preserved() {
1087 let state: HealRuntimeState = serde_json::from_str(r#""future""#).unwrap();
1088
1089 assert_eq!(state, HealRuntimeState::Unknown);
1090 }
1091
1092 #[test]
1093 fn test_pool_status_deserialization() {
1094 let json = r#"{"id":1,"cmdline":"/data/pool1/disk{1...4}","lastUpdate":"2026-05-06T00:00:00Z","status":"decommissioning","decommissionStatus":"running","rebalanceStatus":"none","totalSize":1000,"currentSize":600,"usedSize":400,"used":0.4,"decommissionInfo":{"startTime":"2026-05-06T00:00:01Z","startSize":100,"totalSize":1000,"currentSize":600,"complete":false,"failed":false,"canceled":false,"queued":true,"queuedBuckets":["bucket-a"],"decommissionedBuckets":["bucket-b"],"bucket":"bucket-a","prefix":"","object":"object.txt","stage":"migrate_object","objectsDecommissioned":2,"objectsDecommissionedFailed":1,"bytesDecommissioned":128,"bytesDecommissionedFailed":64,"waitingReason":"queued"}}"#;
1095
1096 let status: PoolStatus = serde_json::from_str(json).unwrap();
1097
1098 assert_eq!(status.id, 1);
1099 assert_eq!(status.cmd_line, "/data/pool1/disk{1...4}");
1100 assert_eq!(status.status, "decommissioning");
1101 assert_eq!(status.decommission_status, "running");
1102 assert_eq!(status.rebalance_status, "none");
1103 assert_eq!(status.used_size, 400);
1104 let info = status.decommission.expect("decommission info exists");
1105 assert!(info.queued);
1106 assert_eq!(info.queued_buckets, vec!["bucket-a"]);
1107 assert_eq!(info.bucket, "bucket-a");
1108 assert_eq!(info.object, "object.txt");
1109 assert_eq!(info.waiting_reason.as_deref(), Some("queued"));
1110 assert_eq!(info.objects_decommissioned, 2);
1111 assert_eq!(info.bytes_decommissioned_failed, 64);
1112 }
1113
1114 #[test]
1115 fn test_decommission_status_deserialization() {
1116 let json = r#"{"pools":[{"id":2,"cmdline":"/data/pool2/disk{1...4}","status":"failed","poolStatus":"blocked","decommissionInfo":{"failed":true,"totalSize":1000,"currentSize":900}}]}"#;
1117
1118 let status: DecommissionStatus = serde_json::from_str(json).unwrap();
1119
1120 assert_eq!(status.pools.len(), 1);
1121 assert_eq!(status.pools[0].id, 2);
1122 assert_eq!(status.pools[0].status, "failed");
1123 assert_eq!(status.pools[0].pool_status, "blocked");
1124 assert!(
1125 status.pools[0]
1126 .decommission
1127 .as_ref()
1128 .is_some_and(|info| info.failed)
1129 );
1130 }
1131
1132 #[test]
1133 fn test_rebalance_status_deserialization() {
1134 let json = r#"{"id":"rebalance-1","pools":[{"id":0,"status":"Started","used":0.5,"lastError":null,"cleanupWarnings":{"count":1,"lastMsg":"cleanup warning","lastBucket":"bucket","lastObject":"object","lastAt":"2026-06-12T00:00:00Z"},"progress":{"objects":3,"versions":4,"bytes":1024,"remainingBuckets":2,"bucket":"bucket","object":"object","elapsed":10,"eta":20}}],"stoppedAt":null}"#;
1135
1136 let status: RebalanceStatus = serde_json::from_str(json).unwrap();
1137
1138 assert_eq!(status.id, "rebalance-1");
1139 assert_eq!(status.pools.len(), 1);
1140 assert_eq!(status.pools[0].used, 0.5);
1141 assert_eq!(status.pools[0].cleanup_warnings.count, 1);
1142 assert_eq!(
1143 status.pools[0].cleanup_warnings.last_message.as_deref(),
1144 Some("cleanup warning")
1145 );
1146 let progress = status.pools[0]
1147 .progress
1148 .as_ref()
1149 .expect("progress should exist");
1150 assert_eq!(progress.num_objects, 3);
1151 assert_eq!(progress.remaining_buckets, 2);
1152 }
1153
1154 #[test]
1155 fn test_rebalance_status_defaults_cleanup_warnings() {
1156 let json = r#"{"id":"rebalance-1","pools":[{"id":0,"status":"Completed","used":0.5,"lastError":null,"progress":null}],"stoppedAt":null}"#;
1157
1158 let status: RebalanceStatus = serde_json::from_str(json).unwrap();
1159
1160 assert_eq!(status.pools[0].cleanup_warnings.count, 0);
1161 assert_eq!(status.pools[0].cleanup_warnings.last_message, None);
1162 }
1163
1164 #[test]
1165 fn test_serialization() {
1166 let info = ClusterInfo {
1167 mode: Some("distributed".to_string()),
1168 deployment_id: Some("test-123".to_string()),
1169 ..Default::default()
1170 };
1171
1172 let json = serde_json::to_string(&info).unwrap();
1173 assert!(json.contains("distributed"));
1174 assert!(json.contains("test-123"));
1175
1176 let deserialized: ClusterInfo = serde_json::from_str(&json).unwrap();
1177 assert_eq!(deserialized.mode, Some("distributed".to_string()));
1178 }
1179}