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
|
#include <pg_query.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdbool.h>
#include <sys/mman.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h>
#include <libgen.h>
#include <assert.h>
int main() {
bool ret_code = EXIT_SUCCESS;
char *sample_buffer;
struct stat sample_stat;
int fd;
FILE* f_out;
PgQueryPlpgsqlParseResult result;
fd = open("test/plpgsql_samples.sql", O_RDONLY);
if (fd < 0) {
printf("Could not read samples file\n");
return EXIT_FAILURE;
}
fstat(fd, &sample_stat);
sample_buffer = mmap(0, sample_stat.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
if (sample_buffer != (void *) - 1)
{
result = pg_query_parse_plpgsql(sample_buffer);
munmap(sample_buffer, sample_stat.st_size);
close(fd);
} else {
printf("Could not mmap samples file\n");
close(fd);
return EXIT_FAILURE;
}
if (result.error) {
printf("ERROR: %s\n", result.error->message);
printf("CONTEXT: %s\n", result.error->context);
printf("LOCATION: %s, %s:%d\n\n", result.error->funcname, result.error->filename, result.error->lineno);
pg_query_free_plpgsql_parse_result(result);
return EXIT_FAILURE;
}
f_out = fopen("test/plpgsql_samples.actual.json", "w");
fprintf(f_out, "%s\n", result.plpgsql_funcs);
fclose(f_out);
pg_query_free_plpgsql_parse_result(result);
pg_query_exit();
return ret_code;
}
|
