Merge pull request #2 from codebrick-corp/feat-add-identifier

added an identifier for dms task to metric label

Author: estherk0

README.md

@@ -43,11 +43,11 @@ Belows are the list of metrics that `aws-dms-task-exporter` exports.
 Sample metrics
 ```
 # TYPE dms_task_stats gauge
-dms_task_stats{action="delete",region="ap-southeast-1",schema="example_schema",table="inventories"} 40601
-dms_task_stats{action="insert",region="ap-southeast-1",schema="example_schema",table="inventories"} 4.145428e+06
-dms_task_stats{action="update",region="ap-southeast-1",schema="example_schema",table="inventories"} 1.24051e+06
+dms_task_stats{action="delete",region="ap-southeast-1",identifier="inventory-change-task",schema="example_schema",table="inventories"} 40601
+dms_task_stats{action="insert",region="ap-southeast-1",identifier="inventory-change-task",schema="example_schema",table="inventories"} 4.145428e+06
+dms_task_stats{action="update",region="ap-southeast-1",identifier="inventory-change-task",schema="example_schema",table="inventories"} 1.24051e+06
 ```
 
 Name | Description | Labels
 -----|-----|-----
-dms_task_stats | DMS Task Table Statistics showing counts of Insert, Delete, Update of source tables | action, region, schema, table
+dms_task_stats | DMS Task Table Statistics showing counts of Insert, Delete, Update of source tables | action, region, schema, table, identifier

aws-dms-task-exporter.go

@@ -18,7 +18,7 @@ var (
 	MetricName        = "task_stats"
 	MetricNamespace   = "dms"
 	MetricHelpMessage = "Gauge for dms tasks statistics"
-	MetricLabels      = []string{"region", "schema", "table", "action"}
+	MetricLabels      = []string{"region", "identifier", "schema", "table", "action"}
 )
 
 type collector struct {
@@ -106,9 +106,9 @@ func (c *collector) fetch(gauge *prometheus.GaugeVec) {
 			continue
 		}
 		for _, stat := range stats {
-			gauge.WithLabelValues(viper.GetString("AWS_REGION"), *stat.schemaName, *stat.tableName, "insert").Set(float64(*stat.inserts))
-			gauge.WithLabelValues(viper.GetString("AWS_REGION"), *stat.schemaName, *stat.tableName, "delete").Set(float64(*stat.deletes))
-			gauge.WithLabelValues(viper.GetString("AWS_REGION"), *stat.schemaName, *stat.tableName, "update").Set(float64(*stat.updates))
+			gauge.WithLabelValues(viper.GetString("AWS_REGION"), *task.identifier, *stat.schemaName, *stat.tableName, "insert").Set(float64(*stat.inserts))
+			gauge.WithLabelValues(viper.GetString("AWS_REGION"), *task.identifier, *stat.schemaName, *stat.tableName, "delete").Set(float64(*stat.deletes))
+			gauge.WithLabelValues(viper.GetString("AWS_REGION"), *task.identifier, *stat.schemaName, *stat.tableName, "update").Set(float64(*stat.updates))
 		}
 	}
 }