From 85497e3f38a792cdf767bb6b07b06103777a2dde Mon Sep 17 00:00:00 2001
From: Julius Volz <julius@soundcloud.com>
Date: Tue, 5 Aug 2014 19:56:35 +0200
Subject: [PATCH] Add function to drop common labels in a vector.

This fixes https://github.com/prometheus/prometheus/issues/384.

Change-Id: I2973c4baeb8a4618ec3875fb11c6fcf5d111784b
---
 rules/ast/functions.go | 44 ++++++++++++++++++++++++++++++++++++++++++
 rules/rules_test.go    |  9 +++++++++
 2 files changed, 53 insertions(+)

diff --git a/rules/ast/functions.go b/rules/ast/functions.go
index 3759953d7..ccfe3e271 100644
--- a/rules/ast/functions.go
+++ b/rules/ast/functions.go
@@ -246,6 +246,44 @@ func bottomkImpl(timestamp clientmodel.Timestamp, view *viewAdapter, args []Node
 	return Vector(bottomk)
 }
 
+// === drop_common_labels(node VectorNode) Vector ===
+func dropCommonLabelsImpl(timestamp clientmodel.Timestamp, view *viewAdapter, args []Node) interface{} {
+	vector := args[0].(VectorNode).Eval(timestamp, view)
+	if len(vector) < 1 {
+		return Vector{}
+	}
+	common := clientmodel.LabelSet{}
+	for k, v := range vector[0].Metric {
+		// TODO(julius): Revisit this when https://github.com/prometheus/prometheus/issues/380
+		// is implemented.
+		if k == clientmodel.MetricNameLabel {
+			continue
+		}
+		common[k] = v
+	}
+
+	for _, el := range vector[1:] {
+		for k, v := range common {
+			if el.Metric[k] != v {
+				// Deletion of map entries while iterating over them is safe.
+				// From http://golang.org/ref/spec#For_statements:
+				// "If map entries that have not yet been reached are deleted during
+				// iteration, the corresponding iteration values will not be produced."
+				delete(common, k)
+			}
+		}
+	}
+
+	for _, el := range vector {
+		for k, _ := range el.Metric {
+			if _, ok := common[k]; ok {
+				delete(el.Metric, k)
+			}
+		}
+	}
+	return vector
+}
+
 // === sampleVectorImpl() Vector ===
 func sampleVectorImpl(timestamp clientmodel.Timestamp, view *viewAdapter, args []Node) interface{} {
 	return Vector{
@@ -450,6 +488,12 @@ var functions = map[string]*Function{
 		returnType: VECTOR,
 		callFn:     deltaImpl,
 	},
+	"drop_common_labels": {
+		name:       "drop_common_labels",
+		argTypes:   []ExprType{VECTOR},
+		returnType: VECTOR,
+		callFn:     dropCommonLabelsImpl,
+	},
 	"max_over_time": {
 		name:       "max_over_time",
 		argTypes:   []ExprType{MATRIX},
diff --git a/rules/rules_test.go b/rules/rules_test.go
index a6c793c46..d05919d2d 100644
--- a/rules/rules_test.go
+++ b/rules/rules_test.go
@@ -540,6 +540,15 @@ func TestExpressions(t *testing.T) {
 			fullRanges:     0,
 			intervalRanges: 0,
 		},
+		{
+			expr: `drop_common_labels(http_requests{group="production",job="api-server"})`,
+			output: []string{
+				`http_requests{instance="0"} => 100 @[%v]`,
+				`http_requests{instance="1"} => 200 @[%v]`,
+			},
+			fullRanges:     0,
+			intervalRanges: 2,
+		},
 	}
 
 	tieredStorage, closer := newTestStorage(t)