mgr/telemetry: add `stats_by_class` to the `basic_pool_usage` collection

Any device classes that are not default ('hdd', 'ssd', 'nvme') are screened out.

Signed-off-by: Laura Flores <lflores@redhat.com>
(cherry picked from commit 285d14457c157a3e4dfd12363e0ba02b8add57fa)

diff --git a/src/pybind/mgr/telemetry/module.py b/src/pybind/mgr/telemetry/module.py
index 74456920a741236c5dccdb14b16f91b8d19b84ab..9dd53f8dc9bba5d1a82436a40da13ec38fc87b06 100644 (file)
--- a/src/pybind/mgr/telemetry/module.py
+++ b/src/pybind/mgr/telemetry/module.py
@@ -1066,6 +1066,12 @@ class Module(MgrModule):
                 'total_bytes': df['stats']['total_bytes'],
                 'total_avail_bytes': df['stats']['total_avail_bytes']
             }
+            # basic_pool_usage collection (2/2)
+            if self.is_enabled_collection(Collection.basic_pool_usage):
+                report['usage']['stats_by_class'] = {} # type: ignore
+                for device_class in df['stats_by_class']:
+                    if device_class in ['hdd', 'ssd', 'nvme']:
+                        report['usage']['stats_by_class'][device_class] = df['stats_by_class'][device_class] # type: ignore
 
             services: DefaultDict[str, int] = defaultdict(int)
             for key, value in service_map['services'].items():