def to_runner_api_monitoring_info(self, name, transform_id):
from apache_beam.metrics import monitoring_infos
return monitoring_infos.int64_user_gauge(
name.namespace,
name.name,
self.get_cumulative().to_runner_api_monitoring_info(),
ptransform=transform_id)
def to_runner_api_monitoring_infos(self, transform_id):
"""Returns a list of MonitoringInfos for the metrics in this container."""
all_user_metrics = []
for k, v in self.counters.items():
all_user_metrics.append(
monitoring_infos.int64_user_counter(
k.namespace,
k.name,
v.to_runner_api_monitoring_info(),
ptransform=transform_id))
for k, v in self.distributions.items():
all_user_metrics.append(
monitoring_infos.int64_user_distribution(
k.namespace,
k.name,
v.get_cumulative().to_runner_api_monitoring_info(),
ptransform=transform_id))
for k, v in self.gauges.items():
all_user_metrics.append(
monitoring_infos.int64_user_gauge(
k.namespace,
k.name,
v.get_cumulative().to_runner_api_monitoring_info(),
ptransform=transform_id))
return {monitoring_infos.to_key(mi): mi for mi in all_user_metrics}
def test_int64_user_gauge(self):
metric = GaugeCell().get_cumulative()
result = monitoring_infos.int64_user_gauge('gaugenamespace',
'gaugename', metric)
_, gauge_value = monitoring_infos.extract_gauge_value(result)
self.assertEqual(0, gauge_value)
