Add dump stat for GCE

ocw/lib/dump_state.py

@@ -4,6 +4,7 @@
 from webui.PCWConfig import PCWConfig
 from ocw.lib.azure import Azure
 from ocw.lib.ec2 import EC2
+from ocw.lib.gce import GCE
 from ocw.enums import ProviderChoice
 from ocw.lib.influx import Influx
 
@@ -71,6 +72,37 @@ def dump_state():
                     namespace,
                     EC2(namespace).count_all_vpc
                 )
+            if ProviderChoice.GCE in providers:
+                Influx().dump_resource(
+                    ProviderChoice.GCE.value,
+                    Influx.VMS_QUANTITY,
+                    namespace,
+                    GCE(namespace).count_all_instances
+                )
+                Influx().dump_resource(
+                    ProviderChoice.GCE.value,
+                    Influx.IMAGES_QUANTITY,
+                    namespace,
+                    GCE(namespace).count_all_images
+                )
+                Influx().dump_resource(
+                    ProviderChoice.GCE.value,
+                    Influx.DISK_QUANTITY,
+                    namespace,
+                    GCE(namespace).count_all_disks
+                )
+                Influx().dump_resource(
+                    ProviderChoice.GCE.value,
+                    Influx.BLOB_QUANTITY,
+                    namespace,
+                    GCE(namespace).count_all_blobs
+                )
+                Influx().dump_resource(
+                    ProviderChoice.GCE.value,
+                    Influx.NETWORK_QUANTITY,
+                    namespace,
+                    GCE(namespace).count_all_networks
+                )
         except Exception:
             logger.exception(
                 "[%s] Dump state failed!: \n %s", namespace, traceback.format_exc()

ocw/lib/gce.py

@@ -264,3 +264,22 @@ def cleanup_networks(self) -> None:
                 self._delete_resource(
                     self.compute_client().networks, network["name"], project=self.project, network=network["name"]
                 )
+
+    def count_all_instances(self) -> int:
+        return len(self.list_all_instances())
+
+    def count_all_images(self) -> int:
+        return len(self._paginated(self.compute_client().images, project=self.project))
+
+    def count_all_disks(self) -> int:
+        all_disks = 0
+        for region in self.list_regions():
+            for zone in self.list_zones(region):
+                all_disks += len(self._paginated(self.compute_client().disks, project=self.project, zone=zone))
+        return all_disks
+
+    def count_all_blobs(self) -> int:
+        return len(self._paginated(self.storage_client().objects, bucket=self.__bucket))
+
+    def count_all_networks(self) -> int:
+        return len(self._paginated(self.compute_client().networks, project=self.project))

ocw/lib/influx.py

@@ -17,9 +17,11 @@ class Influx:
     VMS_QUANTITY: str = "vms_quantity"
     IMAGES_QUANTITY: str = "images_quantity"
     DISK_QUANTITY: str = "disk_quantity"
+    BLOB_QUANTITY: str = "blob_quantity"
     VOLUMES_QUANTITY: str = "volumes_quanity"
     IMAGE_VERSION_QUANTITY: str = "img_version_quantity"
     VPC_QUANTITY: str = "vpc_quantity"
+    NETWORK_QUANTITY: str = "network_quantity"
     NAMESPACE_TAG: str = "namespace"
 
     def __init__(self) -> None: