postgresql: CREATE TRIGGER EXECUTE FUNCTION

Also implement CREATE PROCEDURE

Author: kyleconroy

internal/endtoend/testdata/ddl_create_procedure/postgresql/go/db.go

@@ -0,0 +1,6 @@
+-- name: Placeholder :exec
+SELECT 1;
+
+-- FIXME: Implement CALL
+-- name: CallInsertData :select
+CALL insert_data(1, 2);

internal/endtoend/testdata/ddl_create_procedure/postgresql/go/models.go

@@ -0,0 +1,11 @@
+CREATE TABLE tbl (
+        value integer
+);
+
+-- https://www.postgresql.org/docs/current/sql-createprocedure.html
+CREATE PROCEDURE insert_data(a integer, b integer)
+LANGUAGE SQL
+AS $$
+INSERT INTO tbl VALUES (a);
+INSERT INTO tbl VALUES (b);
+$$;

internal/endtoend/testdata/ddl_create_procedure/postgresql/go/query.sql.go

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

internal/endtoend/testdata/ddl_create_procedure/postgresql/query.sql

@@ -0,0 +1,2 @@
+-- name: Placeholder :exec
+SELECT 1;

internal/endtoend/testdata/ddl_create_procedure/postgresql/schema.sql

@@ -0,0 +1,24 @@
+CREATE SCHEMA tdd;
+
+CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
+CREATE EXTENSION IF NOT EXISTS pgcrypto
+    SCHEMA public
+    VERSION "1.3";
+
+CREATE FUNCTION tdd.trigger_set_timestamp() RETURNS trigger
+LANGUAGE plpgsql
+AS $$BEGIN
+  NEW.ts_updated = NOW();
+  RETURN NEW;
+END;
+$$;
+
+CREATE TABLE tdd.tests (
+    test_id uuid DEFAULT uuid_generate_v4() PRIMARY KEY,
+    title text DEFAULT ''::text NOT NULL,
+    descr text DEFAULT ''::text NOT NULL,
+    ts_created timestamp with time zone DEFAULT now() NOT NULL,
+    ts_updated timestamp with time zone DEFAULT now() NOT NULL
+);
+
+CREATE TRIGGER set_timestamp BEFORE UPDATE ON tdd.tests FOR EACH ROW EXECUTE FUNCTION tdd.trigger_set_timestamp();

internal/endtoend/testdata/ddl_create_procedure/postgresql/sqlc.json

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

internal/endtoend/testdata/ddl_create_trigger/postgresql/go/db.go

@@ -420,13 +420,17 @@ func translate(node *nodes.Node) (ast.Node, error) {
 		if err != nil {
 			return nil, err
 		}
-		rt, err := parseRelationFromNodes(n.ReturnType.Names)
-		if err != nil {
-			return nil, err
+		var rt *ast.TypeName
+		if n.ReturnType != nil {
+			rel, err := parseRelationFromNodes(n.ReturnType.Names)
+			if err != nil {
+				return nil, err
+			}
+			rt = rel.TypeName()
 		}
 		stmt := &ast.CreateFunctionStmt{
 			Func:       fn.FuncName(),
-			ReturnType: rt.TypeName(),
+			ReturnType: rt,
 			Replace:    n.Replace,
 			Params:     &ast.List{},
 		}