def _create_metric_key(monitoring_info): step_name = monitoring_infos.get_step_name(monitoring_info) if not step_name: raise ValueError('Failed to deduce step_name from MonitoringInfo: {}' .format(monitoring_info)) namespace, name = monitoring_infos.parse_namespace_and_name(monitoring_info) return MetricKey(step_name, MetricName(namespace, name))
def test_parse_namespace_and_name_for_user_distribution_metric(self): urn = (monitoring_infos.USER_DISTRIBUTION_COUNTER_URN_PREFIX + "counternamespace:countername") input = monitoring_infos.create_monitoring_info(urn, "typeurn", None) namespace, name = monitoring_infos.parse_namespace_and_name(input) self.assertEqual(namespace, "counternamespace") self.assertEqual(name, "countername")
def test_parse_namespace_and_name_for_user_distribution_metric(self): urn = monitoring_infos.USER_DISTRIBUTION_COUNTER_URN labels = {} labels[monitoring_infos.NAMESPACE_LABEL] = "counternamespace" labels[monitoring_infos.NAME_LABEL] = "countername" input = monitoring_infos.create_monitoring_info( urn, "typeurn", None, labels) namespace, name = monitoring_infos.parse_namespace_and_name(input) self.assertEqual(namespace, "counternamespace") self.assertEqual(name, "countername")
def test_parse_namespace_and_name_for_nonuser_metric(self): input = monitoring_infos.create_monitoring_info( "beam:dummy:metric", "typeurn", None) namespace, name = monitoring_infos.parse_namespace_and_name(input) self.assertEqual(namespace, "beam") self.assertEqual(name, "dummy:metric")
def _to_metric_key(self, monitoring_info): # Right now this assumes that all metrics have a PTRANSFORM ptransform_id = monitoring_info.labels['PTRANSFORM'] namespace, name = monitoring_infos.parse_namespace_and_name(monitoring_info) return MetricKey(ptransform_id, MetricName(namespace, name))
def _to_metric_key(self, monitoring_info): # Right now this assumes that all metrics have a PTRANSFORM ptransform_id = monitoring_info.labels['PTRANSFORM'] namespace, name = monitoring_infos.parse_namespace_and_name(monitoring_info) return MetricKey( ptransform_id, metrics.metricbase.MetricName(namespace, name))