Update charm_refresh

Author: carlcsaposs-canonical

src/abstract_charm.py

@@ -4,6 +4,7 @@
 """MySQL Router charm"""
 
 import abc
+import dataclasses
 import logging
 import typing
 
@@ -23,7 +24,8 @@
 logger = logging.getLogger(__name__)
 
 
-class RouterRefresh(charm_refresh.CharmSpecific):
+@dataclasses.dataclass(eq=False)
+class RouterRefresh(charm_refresh.CharmSpecificCommon, abc.ABC):
     """MySQL Router refresh callbacks & configuration"""
 
     @staticmethod
@@ -58,7 +60,7 @@ class MySQLRouterCharm(ops.CharmBase, abc.ABC):
     _READ_WRITE_X_PORT = 6448
     _READ_ONLY_X_PORT = 6449
 
-    refresh: charm_refresh.Refresh
+    refresh: charm_refresh.Common
 
     def __init__(self, *args) -> None:
         super().__init__(*args)

src/charm.py

@@ -13,6 +13,7 @@
 if is_wrong_architecture() and __name__ == "__main__":
     ops.main.main(WrongArchitectureWarningCharm)
 
+import dataclasses
 import enum
 import functools
 import json
@@ -52,6 +53,11 @@ class _ServiceType(enum.Enum):
     LOAD_BALANCER = "LoadBalancer"
 
 
+@dataclasses.dataclass(eq=False)
+class KubernetesRouterRefresh(abstract_charm.RouterRefresh, charm_refresh.CharmSpecificKubernetes):
+    """MySQL Router Kubernetes refresh callbacks & configuration"""
+
+
 @trace_charm(
     tracing_endpoint="tracing_endpoint",
     extra_types=(
@@ -88,9 +94,8 @@ def __init__(self, *args) -> None:
             self.on[rock.CONTAINER_NAME].pebble_ready, self._on_workload_container_pebble_ready
         )
         try:
-            self.refresh = charm_refresh.Refresh(
-                abstract_charm.RouterRefresh(
-                    cloud=charm_refresh.Cloud.KUBERNETES,
+            self.refresh = charm_refresh.Kubernetes(
+                KubernetesRouterRefresh(
                     workload_name="Router",
                     refresh_user_docs_url="https://charmhub.io/mysql-router-k8s/docs/h-upgrade",
                     oci_resource_name="mysql-router-image",