create mock for Gauge class

Author: FrenchGithubUser

synapse/app/phone_stats_home.py

@@ -24,11 +24,9 @@
 import sys
 from typing import TYPE_CHECKING, Mapping, Sized
 
-from prometheus_client import Gauge
-
 from twisted.internet import defer
 
-from synapse.metrics import SERVER_NAME_LABEL
+from synapse.metrics import SERVER_NAME_LABEL, SynapseGauge
 from synapse.types import JsonDict
 from synapse.util.constants import (
     MILLISECONDS_PER_SECOND,
@@ -57,22 +55,22 @@
 _stats_process: list[tuple[int, "resource.struct_rusage"]] = []
 
 # Gauges to expose monthly active user control metrics
-current_mau_gauge = Gauge(
+current_mau_gauge = SynapseGauge(
     "synapse_admin_mau_current",
     "Current MAU",
     labelnames=[SERVER_NAME_LABEL],
 )
-current_mau_by_service_gauge = Gauge(
+current_mau_by_service_gauge = SynapseGauge(
     "synapse_admin_mau_current_mau_by_service",
     "Current MAU by service",
     labelnames=["app_service", SERVER_NAME_LABEL],
 )
-max_mau_gauge = Gauge(
+max_mau_gauge = SynapseGauge(
     "synapse_admin_mau_max",
     "MAU Limit",
     labelnames=[SERVER_NAME_LABEL],
 )
-registered_reserved_users_mau_gauge = Gauge(
+registered_reserved_users_mau_gauge = SynapseGauge(
     "synapse_admin_mau_registered_reserved_users",
     "Registered users with reserved threepids",
     labelnames=[SERVER_NAME_LABEL],

synapse/federation/federation_server.py

@@ -32,7 +32,7 @@
     Union,
 )
 
-from prometheus_client import Gauge, Histogram
+from prometheus_client import Histogram
 
 from twisted.python import failure
 
@@ -79,7 +79,7 @@
     tag_args,
     trace,
 )
-from synapse.metrics import SERVER_NAME_LABEL, SynapseCounter
+from synapse.metrics import SERVER_NAME_LABEL, SynapseCounter, SynapseGauge
 from synapse.metrics.background_process_metrics import wrap_as_background_process
 from synapse.replication.http.federation import (
     ReplicationFederationSendEduRestServlet,
@@ -122,7 +122,7 @@
     labelnames=[SERVER_NAME_LABEL],
 )
 
-last_pdu_ts_metric = Gauge(
+last_pdu_ts_metric = SynapseGauge(
     "synapse_federation_last_received_pdu_time",
     "The timestamp of the last PDU which was successfully received from the given domain",
     labelnames=("origin_server_name", SERVER_NAME_LABEL),

synapse/federation/sender/transaction_manager.py

@@ -20,8 +20,6 @@
 import logging
 from typing import TYPE_CHECKING
 
-from prometheus_client import Gauge
-
 from synapse.api.constants import EduTypes
 from synapse.api.errors import HttpResponseException
 from synapse.events import EventBase
@@ -34,7 +32,7 @@
     tags,
     whitelisted_homeserver,
 )
-from synapse.metrics import SERVER_NAME_LABEL
+from synapse.metrics import SERVER_NAME_LABEL, SynapseGauge
 from synapse.types import JsonDict
 from synapse.util.json import json_decoder
 from synapse.util.metrics import measure_func
@@ -45,7 +43,7 @@
 logger = logging.getLogger(__name__)
 issue_8631_logger = logging.getLogger("synapse.8631_debug")
 
-last_pdu_ts_metric = Gauge(
+last_pdu_ts_metric = SynapseGauge(
     "synapse_federation_last_sent_pdu_time",
     "The timestamp of the last PDU which was successfully sent to the given domain",
     labelnames=("destination_server_name", SERVER_NAME_LABEL),