def to_runner_api_monitoring_info(self, name, transform_id):
from apache_beam.metrics import monitoring_infos
return monitoring_infos.int64_user_counter(
name.namespace,
name.name,
self.get_cumulative(),
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 to_runner_api_monitoring_info(self, name, transform_id):
from apache_beam.metrics import monitoring_infos
return monitoring_infos.int64_user_counter(
name.namespace,
name.name,
metrics_pb2.Metric(counter_data=metrics_pb2.CounterData(
int64_value=self.get_cumulative())),
ptransform=transform_id)
def test_int64_user_counter(self):
expected_labels = {}
expected_labels[monitoring_infos.NAMESPACE_LABEL] = "counternamespace"
expected_labels[monitoring_infos.NAME_LABEL] = "countername"
metric = CounterCell().get_cumulative()
result = monitoring_infos.int64_user_counter('counternamespace',
'countername', metric)
counter_value = monitoring_infos.extract_counter_value(result)
self.assertEqual(0, counter_value)
self.assertEqual(result.labels, expected_labels)
def to_runner_api_monitoring_info(self, name, transform_id):
from apache_beam.metrics import monitoring_infos
if not name.urn:
# User counter case.
return monitoring_infos.int64_user_counter(
name.namespace,
name.name,
self.get_cumulative(),
ptransform=transform_id)
else:
# Arbitrary URN case.
return monitoring_infos.int64_counter(
name.urn, self.get_cumulative(), labels=name.labels)
