uqa-pg-query 0.5.0

PostgreSQL 18 libpg_query parser pin used by UQA Engine
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

#include "pg_query_diagnostics.h"
#include "utils/guc.h"

static __thread PgQueryDiagnosticScope *active_diagnostics = NULL;

void
pg_query_capture_diagnostic(const ErrorData *error)
{
	PgQueryDiagnostic diagnostic;
	int sqlstate = error->sqlerrcode;
	int i;

	if (active_diagnostics == NULL || active_diagnostics->callback == NULL)
		return;
	diagnostic.severity = error->elevel;
	/* Error codes use elog.h's packed six-bit SQLSTATE representation. */
	for (i = 0; i < 5; i++)
	{
		diagnostic.sqlstate[i] = PGUNSIXBIT(sqlstate);
		sqlstate >>= 6;
	}
	diagnostic.sqlstate[5] = '\0';
	diagnostic.message = error->message;
	diagnostic.detail = error->detail;
	diagnostic.hint = error->hint;
	diagnostic.context = error->context;
	diagnostic.cursorpos = error->cursorpos;
	active_diagnostics->callback(active_diagnostics->context, &diagnostic);
}

static void
capture_log(ErrorData *error)
{
	pg_query_capture_diagnostic(error);
	/* The caller receives the diagnostic without process-wide stderr writes. */
	error->output_to_server = false;
}

void
pg_query_begin_diagnostics(PgQueryDiagnosticScope *scope,
	PgQueryDiagnosticCallback callback, void *context)
{
	scope->callback = callback;
	scope->context = context;
	scope->previous = active_diagnostics;
	scope->previous_log_hook = emit_log_hook;
	scope->previous_log_min_messages = log_min_messages;
	active_diagnostics = scope;
	emit_log_hook = capture_log;
	/* Parser NOTICEs, including truncated identifiers, are client diagnostics. */
	log_min_messages = NOTICE;
}

void
pg_query_end_diagnostics(PgQueryDiagnosticScope *scope)
{
	emit_log_hook = scope->previous_log_hook;
	log_min_messages = scope->previous_log_min_messages;
	active_diagnostics = scope->previous;
}