Mimir push reverse proxy in go - Revision history

Busk at 03:41, 24 February 2025

2025-02-24T03:41:01Z

Busk at 03:38, 24 February 2025

2025-02-24T03:38:17Z

Show changes

Busk at 23:23, 23 February 2025

2025-02-23T23:23:03Z

Busk at 23:17, 23 February 2025

2025-02-23T23:17:22Z

Busk at 23:16, 23 February 2025

2025-02-23T23:16:00Z

Busk at 23:15, 23 February 2025

2025-02-23T23:15:23Z

Busk at 21:20, 23 February 2025

2025-02-23T21:20:17Z

Busk at 20:58, 23 February 2025

2025-02-23T20:58:12Z

Show changes

Busk at 16:27, 23 February 2025

2025-02-23T16:27:27Z

Busk at 16:03, 23 February 2025

2025-02-23T16:03:57Z

@@ Line 10: / Line 10: @@
 );
-INSERT INTO users (username, password) VALUES ('admin', 'password123');
+INSERT INTO users (username, password, org_id) VALUES ('admin', 'password123', test);
 ```
@@ Line 302: / Line 302: @@
      http://localhost:8080/api/v1/push
 ```

@@ Line 249: / Line 249: @@
 current_time_seconds=$(date +%s)
 ts=$((current_time_seconds - 0))
 curl -X POST -g \
      -u admin:password123 \
-     --data "my_metric2{j=\"xfour\",job=\"value1\",instance=\"value2\"} 14.34 ${ts}" \
+     --data "my_metric2{app=\"demo\",job=\"testjob\",instance=\"testistance\"} $rn ${ts}" \
      http://localhost:8080/api/v1/push?job=my_job&instance=my_instance
 ```

@@ Line 243: / Line 243: @@
      -u admin:password123 \
      --data "my_metric1{label1=\"value1\",label2=\"value2\"} 12.34 $(date +%s)" \
      http://localhost:8080/api/v1/push?job=my_job&instance=my_instance
 ```

@@ Line 75: / Line 75: @@
          // Add the username label HERE: labels = append(labels,
          pb.Label{Name: "username", Value: username})
          ts := pb.TimeSeries{

@@ Line 28: / Line 28: @@
 string var httpClient *http.Client
-func parseRawMetric(raw string) (*pb.WriteRequest, error) {
+func parseRawMetric(raw string, username string) (*pb.WriteRequest,
          raw = strings.TrimSpace(raw)
@@ Line 112: / Line 170: @@
                          http.Error(w, "Unauthorized",
                          http.StatusUnauthorized) return
+                 }
@@ Line 119: / Line 184: @@
                  } r.Body.Close()
-                 writeReq, err := parseRawMetric(string(data)) // writeReq
+                 // writeReq, err := parseRawMetric(string(data)) //
-                 is now used if err != nil {
+                 writeReq is now used
                          http.Error(w, fmt.Sprintf("Failed to parse
                          metric: %v", err), http.StatusBadRequest)

@@ Line 211: / Line 211: @@
 ```
 set -a
-MIMIR_URL=https://mimir.uvoo.io/api/v1/push
+MIMIR_URL=https://mimir-push-url.example.com/api/v1/push
 MIMIR_USERNAME=mybasicauthuser
 MIMIR_PASSWORD=mybasicauthpass

@@ Line 1: / Line 1: @@
 ```
 CREATE TABLE users (
@@ Line 13: / Line 17: @@
 import (
-	"database/sql"
+    "database/sql"
-	"encoding/base64"
+    "encoding/base64"
-	"fmt"
+    "log"
-	"io"
+    "net/http"
-	"log"
+    "net/http/httputil"
-	"net/http"
+    "net/url"
-	"net/http/httputil"
+    "os"
-	"net/url"
+    "strings"
-	_ "github.com/mattn/go-sqlite3"
+    _ "github.com/mattn/go-sqlite3"
+)
@@ Line 30: / Line 33: @@
 func main() {
-	var err error
+    var err error
-	db, err = sql.Open("sqlite3", "./auth.db")
+    db, err = sql.Open("sqlite3", "./auth.db")
-	if err != nil {
+    if err != nil {
-		log.Fatalf("Failed to open database: %v", err)
+        log.Fatalf("Failed to open database: %v", err)
-−
+    }
-	defer db.Close()
+    defer db.Close()
-	targetURL := "http://mimir-server:9009" // Change this to your Mimir endpoint
+    proxy := httputil.NewSingleHostReverseProxy(target)
-	target, err := url.Parse(targetURL)
+    // Modify the proxy's director to add backend Basic Auth credentials
-	if err != nil {
+    originalDirector := proxy.Director
-		log.Fatalf("Invalid target URL: %v", err)
+    proxy.Director = func(req *http.Request) {
+        originalDirector(req)
-	proxy := httputil.NewSingleHostReverseProxy(target)
+    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
-	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+        if !authenticate(r) {
-		if !authenticate(r) {
+            http.Error(w, "Unauthorized", http.StatusUnauthorized)
-			http.Error(w, "Unauthorized", http.StatusUnauthorized)
+            return
-			return
+        proxy.ServeHTTP(w, r)
-		proxy.ServeHTTP(w, r)
+    })
-	log.Println("Reverse proxy running on :8080")
+    log.Println("Reverse proxy running on :8080")
-	log.Fatal(http.ListenAndServe(":8080", nil))
+    log.Fatal(http.ListenAndServe(":8080", nil))
+}
 // Authenticate using Basic Auth with SQLite
 func authenticate(r *http.Request) bool {
-	authHeader := r.Header.Get("Authorization")
+    authHeader := r.Header.Get("Authorization")
-	if authHeader == "" || !strings.HasPrefix(authHeader, "Basic ") {
+    if authHeader == "" || !strings.HasPrefix(authHeader, "Basic ") {
-		return false
+        return false
-−
+    }
-	decoded, err := base64.StdEncoding.DecodeString(strings.TrimPrefix(authHeader, "Basic "))
+    decoded, err := base64.StdEncoding.DecodeString(strings.TrimPrefix(authHeader, "Basic "))
-	if err != nil {
+    if err != nil {
-		return false
+        return false
-−
+    }
-	parts := strings.SplitN(string(decoded), ":", 2)
+    parts := strings.SplitN(string(decoded), ":", 2)
-	if len(parts) != 2 {
+    if len(parts) != 2 {
-		return false
+        return false
-−
+    }
-	username, password := parts[0], parts[1]
+    username, password := parts[0], parts[1]
-	return validateUser(username, password)
+    return validateUser(username, password)
+}
 // Validate user from SQLite database
 func validateUser(username, password string) bool {
-	var storedPassword string
+    var storedPassword string
-	err := db.QueryRow("SELECT password FROM users WHERE username = ?", username).Scan(&storedPassword)
+    err := db.QueryRow("SELECT password FROM users WHERE username = ?", username).Scan(&storedPassword)
-	if err != nil {
+    if err != nil {
-		return false
+        return false
-−
+    }
-	return password == stor9edPassword
+    return password == storedPassword
+}
 ```
 ```

@@ Line 1: / Line 1: @@
-```CREATE TABLE users (
+```
      username TEXT PRIMARY KEY,
      password TEXT NOT NULL