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)