include select with newline in clickhouse queries

Signed-off-by: jnichols <mail@jnichols.info>
trickstercache · Jan 10, 2024 · 55936b6 · 55936b6
1 parent f31766c
commit 55936b6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pkg/backends/clickhouse/handler_query.go b/pkg/backends/clickhouse/handler_query.go
@@ -52,7 +52,9 @@ func (c *Client) QueryHandler(w http.ResponseWriter, r *http.Request) {
  r = request.SetBody(r, body)
  }
  sqlQuery = strings.ToLower(sqlQuery)
- if (!strings.HasPrefix(sqlQuery, "select ")) && (!(strings.Index(sqlQuery, " select ") > 0)) {
+ if !(strings.HasPrefix(sqlQuery, "select ") ||
+ strings.HasPrefix(sqlQuery, "select\n") ||
+ strings.Contains(sqlQuery, " select ")) {
  c.ProxyHandler(w, r)
  return
  }