feat: adding prefix_prefer_source option

README.md

@@ -155,6 +155,15 @@ tenant:
   # https://grafana.com/docs/mimir/latest/configure/about-tenant-ids/
   # env: CT_TENANT_PREFIX
   prefix: foobar-
+
+  # If true will use the tenant ID of the inbound request as the prefix of the new tenant id.
+  # Will be automatically suffixed with a `-` character.
+  # Example:
+  #   Prometheus forwards metrics with `X-Scope-OrgID: Prom-A` set in the inbound request.
+  #   This would result in the tenant prefix being set to `Prom-A-`.
+  # https://grafana.com/docs/mimir/latest/configure/about-tenant-ids/
+  # env: CT_TENANT_PREFIX_PREFER_SOURCE
+  prefix_prefer_source: false
 ```
 
 ### Prometheus configuration example

VERSION

@@ -1 +1 @@
-1.12.4
+1.12.5

config.go

@@ -37,13 +37,14 @@ type config struct {
 	}
 
 	Tenant struct {
-		Label       string   `env:"CT_TENANT_LABEL"`
-		LabelList   []string `yaml:"label_list" env:"CT_TENANT_LABEL_LIST" envSeparator:","`
-		Prefix      string   `yaml:"prefix" env:"CT_TENANT_PREFIX"`
-		LabelRemove bool     `yaml:"label_remove" env:"CT_TENANT_LABEL_REMOVE"`
-		Header      string   `env:"CT_TENANT_HEADER"`
-		Default     string   `env:"CT_TENANT_DEFAULT"`
-		AcceptAll   bool     `yaml:"accept_all" env:"CT_TENANT_ACCEPT_ALL"`
+		Label              string   `env:"CT_TENANT_LABEL"`
+		LabelList          []string `yaml:"label_list" env:"CT_TENANT_LABEL_LIST" envSeparator:","`
+		Prefix             string   `yaml:"prefix" env:"CT_TENANT_PREFIX"`
+		PrefixPreferSource bool     `yaml:"prefix_prefer_source" env:"CT_TENANT_PREFIX_PREFER_SOURCE`
+		LabelRemove        bool     `yaml:"label_remove" env:"CT_TENANT_LABEL_REMOVE"`
+		Header             string   `env:"CT_TENANT_HEADER"`
+		Default            string   `env:"CT_TENANT_DEFAULT"`
+		AcceptAll          bool     `yaml:"accept_all" env:"CT_TENANT_ACCEPT_ALL"`
 	}
 
 	pipeIn  *fhu.InmemoryListener

config.yml

@@ -25,6 +25,7 @@ tenant:
     - tenant
     - other_tenant
   prefix: ""
+  prefix_prefer_source: false
   label_remove: true
   header: X-Scope-OrgID
   default: ""

deploy/k8s/chart/Chart.yaml

@@ -1,8 +1,8 @@
 apiVersion: v2
 description: A Helm Chart for cortex-tenant
 name: cortex-tenant
-version: 0.4.1 # This is the chart version
-appVersion: 1.12.4 # version number of the application being deployed.
+version: 0.4.2 # This is the chart version
+appVersion: 1.12.5 # version number of the application being deployed.
 type: application
 sources:
   - https://github.com/blind-oracle/cortex-tenant

deploy/k8s/chart/templates/configmap.yaml

@@ -30,6 +30,7 @@ data:
         {{- . | toYaml | nindent 8 }}
       {{- end }}
       prefix: {{ .Values.config.tenant.prefix }}
+      prefix_prefer_source: {{ .Values.config.tenant.prefix_prefer_source }}
       label_remove: {{ .Values.config.tenant.label_remove }}
       header: {{ .Values.config.tenant.header }}
       default: {{ .Values.config.tenant.default }}

deploy/k8s/chart/values.schema.json

@@ -275,6 +275,12 @@
               "title": "Prefix",
               "description": "Optional hard-coded prefix with delimeter for all tenant values"
             },
+            "prefix_prefer_source": {
+              "type": "boolean",
+              "title": "Prefix Prefer Source",
+              "description": "If true will use the tenant ID of the inbound request as the prefix of the new tenant id.",
+              "default": false
+            },
             "label_remove": {
               "type": "boolean",
               "title": "Label Remove",

deploy/k8s/chart/values.yaml

@@ -101,12 +101,20 @@ config:
     # (env: `CT_TENANT_LABEL`)
     label: tenant
     # -- List of labels examined for tenant information. If set takes precedent over `label`
+    # (env: `CT_TENANT_LABEL_LIST`)
     label_list: []
     # -- Optional hard-coded prefix with delimeter for all tenant values.
     # Delimeters allowed for use:
     # https://grafana.com/docs/mimir/latest/configure/about-tenant-ids/
     # (env: `CT_TENANT_PREFIX`)
     prefix: ""
+    # -- If true will use the tenant ID of the inbound request as the prefix of the new tenant id.
+    # Will be automatically suffixed with a `-` character.
+    # Example:
+    #   Prometheus forwards metrics with `X-Scope-OrgID: Prom-A` set in the inbound request.
+    #   This would result in the tenant prefix being set to `Prom-A-`.
+    # (env: `CT_TENANT_PREFIX_PREFER_SOURCE`)
+    prefix_prefer_source: false
     # -- Whether to remove the tenant label from the request
     # (env: `CT_TENANT_LABEL_REMOVE`)
     label_remove: false

deploy/k8s/manifests/config-file-configmap.yml

@@ -42,6 +42,13 @@ data:
       # Delimeters allowed for use:
       # https://grafana.com/docs/mimir/latest/configure/about-tenant-ids/
       prefix: ""
+      # If true will use the tenant ID of the inbound request as the prefix of the new tenant id.
+      # Will be automatically suffixed with a `-` character.
+      # Example:
+      #   Prometheus forwards metrics with `X-Scope-OrgID: Prom-A` set in the inbound request.
+      #   This would result in the tenant prefix being set to `Prom-A-`.
+      # https://grafana.com/docs/mimir/latest/configure/about-tenant-ids/
+      prefix_prefer_source: false
       # Whether to remove the tenant label from the request
       label_remove: false
       # To which header to add the tenant ID

processor.go

@@ -170,6 +170,13 @@ func (p *processor) handle(ctx *fh.RequestCtx) {
 		return
 	}
 
+	if p.cfg.Tenant.PrefixPreferSource {
+		sourceTenantPrefix := string(ctx.Request.Header.Peek(p.cfg.Tenant.Header))
+		if sourceTenantPrefix != "" {
+			p.cfg.Tenant.Prefix = sourceTenantPrefix + "-"
+		}
+	}
+
 	clientIP := ctx.RemoteAddr()
 	reqID, _ := uuid.NewRandom()