support both get and post requests

ikethecoder · Nov 24, 2023 · aaae466 · aaae466
1 parent ccba226
commit aaae466
Showing 1 changed file with 16 additions and 10 deletions.
diff --git a/query-proxy/internal/app/query-proxy/reverse.go b/query-proxy/internal/app/query-proxy/reverse.go
@@ -41,16 +41,22 @@ func modifyRequest(r *http.Request, prometheusServerURL *url.URL, prometheusQuer
 		return err
 	}
 
-	form := r.Form
-	form.Set(prometheusQueryParameter, expr.String())
-	body := form.Encode()
-	r.Body = io.NopCloser(strings.NewReader(body))
-	r.ContentLength = int64(len(body))
-	r.URL.RawQuery = ""
-	r.URL.Fragment = ""
-	r.Header.Set("Content-Type", "application/x-www-form-urlencoded")
-
-	log.Println("TRANSFORMED TO ", body)
+	if r.Method == "GET" {
+		q := r.URL.Query()
+		q.Set(prometheusQueryParameter, expr.String())
+		r.URL.RawQuery = q.Encode()
+		log.Println("TRANSFORMED QUERY TO", r.URL)
+	} else {
+		form := r.Form
+		form.Set(prometheusQueryParameter, expr.String())
+		body := form.Encode()
+		r.Body = io.NopCloser(strings.NewReader(body))
+		r.ContentLength = int64(len(body))
+		r.URL.RawQuery = ""
+		r.URL.Fragment = ""
+		r.Header.Set("Content-Type", "application/x-www-form-urlencoded")
+		log.Println("TRANSFORMED BODY TO", body)
+	}
 
 	return nil
 }