pg_query_parse_plpgsql does not support function with sql_body

[MGorkov]

When parsing function with sql_body (docs) the pg_query_parse_plpgsql aborts with error:
"src/pg_query_parse_plpgsql.c:143: compile_create_function_stmt: Assertion 'proc_source != NULL' failed."

simple_plpgsql.c :

#include <pg_query.h>
#include <stdio.h>
#include <stdlib.h>

int main() {
  PgQueryPlpgsqlParseResult result;

  result = pg_query_parse_plpgsql(" \
  CREATE OR REPLACE FUNCTION add(a integer, b integer) \
RETURNS integer \
LANGUAGE SQL \
IMMUTABLE \
STRICT \
RETURN a + b;");

  if (result.error) {
    printf("error: %s at %d\n", result.error->message, result.error->cursorpos);
  } else {
    printf("%s\n", result.plpgsql_funcs);
  }

  pg_query_free_plpgsql_parse_result(result);

  // Optional, this ensures all memory is freed upon program exit (useful when running Valgrind)
  pg_query_exit();

  return 0;
}

[lfittl]

@MGorkov Thanks for reaching out!

This style of writing a function only works for SQL functions, not PL/pgSQL. The pg_query_parse_plpgsql function is intended to be used for parsing PL/pgSQL, and as such does not support the sql_body syntax.

Instead for SQL functions you should use the regular parse function, as that will be able to read sql_body:

#include <pg_query.h>
#include <stdio.h>
#include <stdlib.h>

int main() {
  PgQueryParseResult result;

  result = pg_query_parse(" \
  CREATE OR REPLACE FUNCTION add(a integer, b integer) \
RETURNS integer \
LANGUAGE SQL \
IMMUTABLE \
STRICT \
RETURN a + b;");

  if (result.error) {
    printf("error: %s at %d\n", result.error->message, result.error->cursorpos);
  } else {
    printf("%s\n", result.parse_tree);
  }

  pg_query_free_parse_result(result);

  // Optional, this ensures all memory is freed upon program exit (useful when running Valgrind)
  pg_query_exit();

  return 0;
}

{
    "version": 160001,
    "stmts": [
        {
            "stmt": {
                "CreateFunctionStmt": {
                    // ...
                    ],
                    "sql_body": {
                        "ReturnStmt": {
                            "returnval": {
                                "A_Expr": {
                                    "kind": "AEXPR_OP",
                                    "name": [
                                        {
                                            "String": {
                                                "sval": "+"
                                            }
                                        }
                                    ],
                                    "lexpr": {
                                        "ColumnRef": {
                                            "fields": [
                                                {
                                                    "String": {
                                                        "sval": "a"
                                                    }
                                                }
                                            ],
                                            "location": 109
                                        }
                                    },
                                    "rexpr": {
                                        "ColumnRef": {
                                            "fields": [
                                                {
                                                    "String": {
                                                        "sval": "b"
                                                    }
                                                }
                                            ],
                                            "location": 113
                                        }
                                    },
                                    "location": 111
                                }
                            }
                        }
                    }
                }
            },
            "stmt_len": 114
        }
    ]
}