Support GRAPH statement (#181)

Author: apstndb

statement.go

@@ -82,7 +82,7 @@ type QueryStats struct {
 
 var (
 	// SQL
-	selectRe = regexp.MustCompile(`(?is)^(?:WITH|CALL|@{.+|SELECT)\s.+$`)
+	selectRe = regexp.MustCompile(`(?is)^(?:WITH|CALL|@{.+|SELECT|GRAPH)\s.+$`)
 
 	// DDL
 	createDatabaseRe = regexp.MustCompile(`(?is)^CREATE\s+DATABASE\s.+$`)

statement_test.go

@@ -507,6 +507,21 @@ func TestBuildStatement(t *testing.T) {
 			input: "EXPLAIN WITH t1 AS (SELECT 1) SELECT * FROM t1",
 			want:  &ExplainStatement{Explain: "WITH t1 AS (SELECT 1) SELECT * FROM t1"},
 		},
+		{
+			desc:  "GRAPH statement",
+			input: "GRAPH FinGraph MATCH (n) RETURN LABELS(n) AS label, n.id",
+			want:  &SelectStatement{Query: "GRAPH FinGraph MATCH (n) RETURN LABELS(n) AS label, n.id"},
+		},
+		{
+			desc:  "EXPLAIN GRAPH statement",
+			input: "EXPLAIN GRAPH FinGraph MATCH (n) RETURN LABELS(n) AS label, n.id",
+			want:  &ExplainStatement{Explain: "GRAPH FinGraph MATCH (n) RETURN LABELS(n) AS label, n.id"},
+		},
+		{
+			desc:  "EXPLAIN ANALYZE GRAPH statement",
+			input: "EXPLAIN ANALYZE GRAPH FinGraph MATCH (n) RETURN LABELS(n) AS label, n.id",
+			want:  &ExplainAnalyzeStatement{Query: "GRAPH FinGraph MATCH (n) RETURN LABELS(n) AS label, n.id"},
+		},
 		{
 			desc:  "DESCRIBE SELECT statement",
 			input: "DESCRIBE SELECT * FROM t1",