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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
pub mod entity;
use proc_macro2::{Span, TokenStream as TokenStream2};
use quote::{quote, ToTokens, TokenStreamExt};
use std::hash::{Hash, Hasher};
use syn::parse::{Parse, ParseStream};
use syn::ItemMod;
#[derive(Debug, Clone)]
pub struct Schema {
pub module: ItemMod,
}
impl Parse for Schema {
fn parse(input: ParseStream) -> Result<Self, syn::Error> {
let module: ItemMod = input.parse()?;
crate::ident_is_acceptable_to_postgres(&module.ident)?;
Ok(Self { module })
}
}
impl ToTokens for Schema {
fn to_tokens(&self, tokens: &mut TokenStream2) {
let attrs = &self.module.attrs;
let vis = &self.module.vis;
let mod_token = &self.module.mod_token;
let ident = &self.module.ident;
let (_content_brace, content_items) =
&self.module.content.as_ref().expect("Can only support `mod {}` right now.");
let mut hasher = std::collections::hash_map::DefaultHasher::new();
content_items.hash(&mut hasher);
let postfix = hasher.finish();
let mut updated_content = content_items.clone();
let sql_graph_entity_fn_name = syn::Ident::new(
&format!("__pgx_internals_schema_{}_{}", ident, postfix),
Span::call_site(),
);
updated_content.push(syn::parse_quote! {
#[no_mangle]
#[doc(hidden)]
pub extern "Rust" fn #sql_graph_entity_fn_name() -> ::pgx::utils::sql_entity_graph::SqlGraphEntity {
extern crate alloc;
use alloc::vec::Vec;
use alloc::vec;
let submission = pgx::utils::sql_entity_graph::SchemaEntity {
module_path: module_path!(),
name: stringify!(#ident),
file: file!(),
line: line!(),
};
::pgx::utils::sql_entity_graph::SqlGraphEntity::Schema(submission)
}
});
let _semi = &self.module.semi;
let inv = quote! {
#(#attrs)*
#vis #mod_token #ident {
#(#updated_content)*
}
};
tokens.append_all(inv);
}
}