fest: Add compiler support for NOTIFY and LISTEN (PostgreSQL) (#2363)

Author: Jille

internal/compiler/output_columns.go

@@ -115,7 +115,7 @@ func (c *Compiler) outputColumns(qc *QueryCatalog, node ast.Node) ([]*Column, er
 		}
 	case *ast.CallStmt:
 		targets = &ast.List{}
-	case *ast.TruncateStmt, *ast.RefreshMatViewStmt:
+	case *ast.TruncateStmt, *ast.RefreshMatViewStmt, *ast.NotifyStmt, *ast.ListenStmt:
 		targets = &ast.List{}
 	case *ast.UpdateStmt:
 		targets = n.ReturningList
@@ -491,6 +491,8 @@ func (c *Compiler) sourceTables(qc *QueryCatalog, node ast.Node) ([]*Table, erro
 		}
 	case *ast.CallStmt:
 		list = &ast.List{}
+	case *ast.NotifyStmt, *ast.ListenStmt:
+		list = &ast.List{}
 	default:
 		return nil, fmt.Errorf("sourceTables: unsupported node type: %T", n)
 	}

internal/compiler/parse.go

@@ -48,6 +48,8 @@ func (c *Compiler) parseQuery(stmt ast.Node, src string, o opts.Parser) (*Query,
 		if err != nil {
 			return nil, err
 		}
+	case *ast.ListenStmt:
+	case *ast.NotifyStmt:
 	case *ast.TruncateStmt:
 	case *ast.UpdateStmt:
 	case *ast.RefreshMatViewStmt:

internal/endtoend/testdata/notifylisten/postgresql/pgx/v5/go/db.go

@@ -0,0 +1,8 @@
+-- name: NotifyTest :exec
+NOTIFY test;
+
+-- name: NotifyWithMessage :exec
+NOTIFY test, 'msg';
+
+-- name: ListenTest :exec
+LISTEN test;

internal/endtoend/testdata/notifylisten/postgresql/pgx/v5/go/models.go

@@ -0,0 +1,13 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "engine": "postgresql",
+      "sql_package": "pgx/v5",
+      "name": "querytest",
+      "schema": "query.sql",
+      "queries": "query.sql",
+    }
+  ]
+}