diff options
Diffstat (limited to '')
-rw-r--r-- | src/pybind/mgr/dashboard/services/cluster.py | 87 |
1 files changed, 87 insertions, 0 deletions
diff --git a/src/pybind/mgr/dashboard/services/cluster.py b/src/pybind/mgr/dashboard/services/cluster.py new file mode 100644 index 000000000..9caaf1963 --- /dev/null +++ b/src/pybind/mgr/dashboard/services/cluster.py @@ -0,0 +1,87 @@ +# -*- coding: utf-8 -*- +from enum import Enum +from typing import NamedTuple + +from .. import mgr + + +class ClusterCapacity(NamedTuple): + total_avail_bytes: int + total_bytes: int + total_used_raw_bytes: int + total_objects: int + total_pool_bytes_used: int + average_object_size: int + + +class ClusterModel: + + class Status(Enum): + INSTALLED = 0 + POST_INSTALLED = 1 + + status: Status + + def __init__(self, status=Status.POST_INSTALLED.name): + """ + :param status: The status of the cluster. Assume that the cluster + is already functional by default. + :type status: str + """ + self.status = self.Status[status] + + def dict(self): + return {'status': self.status.name} + + def to_db(self): + mgr.set_store('cluster/status', self.status.name) + + @classmethod + def from_db(cls): + """ + Get the stored cluster status from the configuration key/value store. + If the status is not set, assume it is already fully functional. + """ + return cls(status=mgr.get_store('cluster/status', cls.Status.POST_INSTALLED.name)) + + @classmethod + def get_capacity(cls) -> ClusterCapacity: + df = mgr.get('df') + total_pool_bytes_used = 0 + average_object_size = 0 + total_data_pool_objects = 0 + total_data_pool_bytes_used = 0 + rgw_pools_data = cls.get_rgw_pools() + + for pool in df['pools']: + pool_name = str(pool['name']) + if pool_name in rgw_pools_data: + if pool_name.endswith('.data'): + objects = pool['stats']['objects'] + pool_bytes_used = pool['stats']['bytes_used'] + total_pool_bytes_used += pool_bytes_used + total_data_pool_objects += objects + replica = rgw_pools_data[pool_name] + total_data_pool_bytes_used += pool_bytes_used / replica + + average_object_size = total_data_pool_bytes_used / total_data_pool_objects if total_data_pool_objects != 0 else 0 # noqa E501 #pylint: disable=line-too-long + + return ClusterCapacity( + total_avail_bytes=df['stats']['total_avail_bytes'], + total_bytes=df['stats']['total_bytes'], + total_used_raw_bytes=df['stats']['total_used_raw_bytes'], + total_objects=total_data_pool_objects, + total_pool_bytes_used=total_pool_bytes_used, + average_object_size=average_object_size + )._asdict() + + @classmethod + def get_rgw_pools(cls): + rgw_pool_size = {} + + osd_map = mgr.get('osd_map') + for pool in osd_map['pools']: + if 'rgw' in pool.get('application_metadata', {}): + name = pool['pool_name'] + rgw_pool_size[name] = pool['size'] + return rgw_pool_size |