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
use crate::{PgxSql, SqlGraphEntity, SqlGraphIdentifier, ToSql, ToSqlConfigEntity};
#[derive(Debug, Clone, Hash, PartialEq, Eq, PartialOrd, Ord)]
pub struct PgTriggerEntity {
pub function_name: &'static str,
pub to_sql_config: ToSqlConfigEntity,
pub file: &'static str,
pub line: u32,
pub module_path: &'static str,
pub full_path: &'static str,
}
impl PgTriggerEntity {
fn wrapper_function_name(&self) -> String {
self.function_name.to_string() + "_wrapper"
}
}
impl From<PgTriggerEntity> for SqlGraphEntity {
fn from(val: PgTriggerEntity) -> Self {
SqlGraphEntity::Trigger(val)
}
}
impl ToSql for PgTriggerEntity {
#[tracing::instrument(
level = "error",
skip(self, context),
fields(identifier = %self.rust_identifier()),
)]
fn to_sql(&self, context: &PgxSql) -> eyre::Result<String> {
let self_index = context.triggers[self];
let schema = context.schema_prefix_for(&self_index);
let sql = format!(
"\n\
-- {file}:{line}\n\
-- {full_path}\n\
CREATE FUNCTION {schema}\"{function_name}\"()\n\
\tRETURNS TRIGGER\n\
\tLANGUAGE c\n\
\tAS 'MODULE_PATHNAME', '{wrapper_function_name}';",
schema = schema,
file = self.file,
line = self.line,
full_path = self.full_path,
function_name = self.function_name,
wrapper_function_name = self.wrapper_function_name(),
);
Ok(sql)
}
}
impl SqlGraphIdentifier for PgTriggerEntity {
fn dot_identifier(&self) -> String {
format!("trigger fn {}", self.full_path)
}
fn rust_identifier(&self) -> String {
self.full_path.to_string()
}
fn file(&self) -> Option<&'static str> {
Some(self.file)
}
fn line(&self) -> Option<u32> {
Some(self.line)
}
}