From 0b412b875af8a26fe37fd07bf41b87a39cca6bf8 Mon Sep 17 00:00:00 2001
From: Dmitriy Mann <2284562+mdsjip@users.noreply.github.com>
Date: Tue, 9 May 2023 15:47:02 +0200
Subject: [PATCH] fix(analysis): Graphite query - remove whitespaces (#2752)

* fix(analysis): Graphite query - remove whitespaces

Signed-off-by: mdsjip <2284562+mdsjip@users.noreply.github.com>

* refactor(analysis): Graphite query - proper naming and regexp usage

Signed-off-by: mdsjip <2284562+mdsjip@users.noreply.github.com>

---------

Signed-off-by: mdsjip <2284562+mdsjip@users.noreply.github.com>
---
 metricproviders/graphite/api.go      |  9 +++++----
 metricproviders/graphite/api_test.go | 23 +++++++++++++++++++++++
 2 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/metricproviders/graphite/api.go b/metricproviders/graphite/api.go
index fc304395b1..06833ac028 100644
--- a/metricproviders/graphite/api.go
+++ b/metricproviders/graphite/api.go
@@ -27,9 +27,11 @@ type APIClient struct {
 	logCTX log.Entry
 }
 
+var spaceRegex = regexp.MustCompile(`\s+`)
+
 // Query performs a Graphite API query with the query it's passed
 func (api APIClient) Query(quer string) ([]dataPoint, error) {
-	query := api.trimQuery(quer)
+	query := api.sanitizeQuery(quer)
 	u, err := url.Parse(fmt.Sprintf("./render?%s", query))
 	if err != nil {
 		return []dataPoint{}, err
@@ -75,9 +77,8 @@ func (api APIClient) Query(quer string) ([]dataPoint, error) {
 	return result[0].DataPoints, nil
 }
 
-func (api APIClient) trimQuery(q string) string {
-	space := regexp.MustCompile(`\s+`)
-	return space.ReplaceAllString(q, " ")
+func (api APIClient) sanitizeQuery(q string) string {
+	return spaceRegex.ReplaceAllLiteralString(q, "")
 }
 
 type dataPoint struct {
diff --git a/metricproviders/graphite/api_test.go b/metricproviders/graphite/api_test.go
index bca62bbae2..74f0ef1176 100644
--- a/metricproviders/graphite/api_test.go
+++ b/metricproviders/graphite/api_test.go
@@ -91,6 +91,29 @@ func TestQuery(t *testing.T) {
 		nil,
 		`[]`,
 		200,
+	}, {
+		"query with surrounding whitespace",
+		fmt.Sprintf("\n  %s \t  \n", query),
+		targetQuery,
+		fromQuery,
+		goodResult,
+		nil,
+		fmt.Sprintf(`[
+			{
+				"datapoints": [
+					[
+						%f,
+						%d
+					]
+				],
+				"target": "sumSeries(app.http.*.*.count)",
+				"tags": {
+					"aggregatedBy": "sum",
+					"name": "sumSeries(app.http.*.*.count)"
+				}
+			}
+		]`, value, timestamp),
+		200,
 	}, {
 		"graphite response body with invalid JSON",
 		query,