Fix processing empty or comment-only queries

BemiHQ · Nov 22, 2024 · a96c6db · a96c6db
1 parent 4b55496
commit a96c6db
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 3 deletions.
diff --git a/scripts/install.sh b/scripts/install.sh
@@ -1,6 +1,6 @@
 #!/bin/bash
 
-VERSION="0.13.1"
+VERSION="0.13.2"
 
 # Detect OS and architecture
 OS=$(uname -s | tr '[:upper:]' '[:lower:]')

diff --git a/src/main.go b/src/main.go
@@ -6,7 +6,7 @@ import (
 	"time"
 )
 
-const VERSION = "0.13.1"
+const VERSION = "0.13.2"
 
 func main() {
 	flag.Parse()

diff --git a/src/query_handler.go b/src/query_handler.go
@@ -357,6 +357,8 @@ func (queryHandler *QueryHandler) remapQuery(query string) (string, error) {
 	var statementNode *pgQuery.Node
 	if len(queryTree.Stmts) > 0 {
 		statementNode = queryTree.Stmts[0].Stmt
+	} else {
+		return FALLBACK_SQL_QUERY, nil
 	}
 
 	if statementNode != nil && statementNode.GetSelectStmt() != nil {
@@ -373,7 +375,7 @@ func (queryHandler *QueryHandler) remapQuery(query string) (string, error) {
 		return FALLBACK_SQL_QUERY, nil
 	}
 
-	LogDebug(queryHandler.config, queryTree)
+	LogDebug(queryHandler.config, "Query tree:", queryTree)
 	return "", errors.New("Unsupported query type")
 }
 

diff --git a/src/query_handler_test.go b/src/query_handler_test.go
@@ -59,6 +59,11 @@ func TestHandleQuery(t *testing.T) {
 			"description": {"Success"},
 			"values":      {},
 		},
+		// Empty query
+		"-- ping": {
+			"description": {"1"},
+			"values":      {"1"},
+		},
 		// DISCARD
 		"DISCARD ALL": {
 			"description": {"1"},