pg_parse 0.14.0

PostgreSQL parser that uses the actual PostgreSQL server source to parse SQL queries and return the internal PostgreSQL parse tree.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

#include <stdbool.h>
#include <stddef.h>
#include <stdlib.h>

#include "pg_query.h"
#include "pg_query_internal.h"
#include <nodes/nodes.h>
#include <nodes/nodeFuncs.h>

static
bool
is_utility_stmt_actual(RawStmt *raw_stmt)
{
	switch (nodeTag(raw_stmt->stmt))
	{
		case T_SelectStmt:
		case T_InsertStmt:
		case T_UpdateStmt:
		case T_DeleteStmt:
		case T_MergeStmt:
			return false;

		default:
			return true;
	}
}

PgQueryIsUtilityResult
pg_query_is_utility_stmt(const char *query)
{
	PgQueryIsUtilityResult result = {0};
	MemoryContext ctx = pg_query_enter_memory_context();

	PgQueryInternalParsetreeAndError parsetree_and_error = pg_query_raw_parse(query, 0);

	if (parsetree_and_error.error)
	{
		result.error = parsetree_and_error.error;
	}
	else
	{
		ListCell   *lc;

		result.length = list_length(parsetree_and_error.tree);
		result.items = malloc(sizeof(bool) * result.length);

		foreach(lc, parsetree_and_error.tree)
		{
			RawStmt    *raw_stmt = lfirst_node(RawStmt, lc);

			result.items[foreach_current_index(lc)] = is_utility_stmt_actual(raw_stmt);
		}
	}

	if (parsetree_and_error.stderr_buffer)
		free(parsetree_and_error.stderr_buffer);

	pg_query_exit_memory_context(ctx);

	return result;
}

void
pg_query_free_is_utility_result(PgQueryIsUtilityResult result)
{
	if (result.error)
		pg_query_free_error(result.error);

	free(result.items);
}