From 80b92335bb0c74a18fa8ef959275d40d2a01beae Mon Sep 17 00:00:00 2001
From: "r.khavronenko" <r.khavronenko@vertamedia.com>
Date: Mon, 6 Feb 2017 16:22:09 +0200
Subject: [PATCH] allow setting basic auth headers for prometheus datasource

---
 pkg/tsdb/prometheus/prometheus.go | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/pkg/tsdb/prometheus/prometheus.go b/pkg/tsdb/prometheus/prometheus.go
index f3704c2a430..c73e166138d 100644
--- a/pkg/tsdb/prometheus/prometheus.go
+++ b/pkg/tsdb/prometheus/prometheus.go
@@ -22,6 +22,18 @@ type PrometheusExecutor struct {
 	Transport *http.Transport
 }
 
+type basicAuthTransport struct {
+	*http.Transport
+
+	username string
+	password string
+}
+
+func (bat basicAuthTransport) RoundTrip(req *http.Request) (*http.Response, error) {
+	req.SetBasicAuth(bat.username, bat.password)
+	return http.DefaultTransport.RoundTrip(req)
+}
+
 func NewPrometheusExecutor(dsInfo *models.DataSource) (tsdb.Executor, error) {
 	transport, err := dsInfo.GetHttpTransport()
 	if err != nil {
@@ -51,6 +63,14 @@ func (e *PrometheusExecutor) getClient() (prometheus.QueryAPI, error) {
 		Transport: e.Transport,
 	}
 
+	if e.BasicAuth {
+		cfg.Transport = basicAuthTransport{
+			Transport: e.Transport,
+			username: e.BasicAuthUser,
+			password: e.BasicAuthPassword,
+		}
+	}
+
 	client, err := prometheus.New(cfg)
 	if err != nil {
 		return nil, err