elefant-tools 0.0.2

A library for doing things like pg_dump and pg_restore, with extra features, and probably more bugs.
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
71
72
73
74
75
76
77
78
79
80
81
82

use crate::object_id::ObjectId;
use crate::pg_interval::Interval;
use crate::quoting::AttemptedKeywordUsage::TypeOrFunctionName;
use crate::quoting::{quote_value_string, IdentifierQuoter, Quotable};
use crate::whitespace_ignorant_string::WhitespaceIgnorantString;
use serde::{Deserialize, Serialize};

#[derive(Debug, Eq, PartialEq, Clone, Serialize, Deserialize)]
pub struct TimescaleDbUserDefinedJob {
    pub function_name: String,
    pub function_schema: String,
    pub schedule_interval: Interval,
    pub config: Option<WhitespaceIgnorantString>,
    pub scheduled: bool,
    pub check_config_name: Option<String>,
    pub check_config_schema: Option<String>,
    pub fixed_schedule: bool,
    pub object_id: ObjectId,
}

impl Default for TimescaleDbUserDefinedJob {
    fn default() -> Self {
        TimescaleDbUserDefinedJob {
            function_name: String::new(),
            function_schema: String::new(),
            schedule_interval: Interval::new(0, 1, 0),
            config: None,
            scheduled: false,
            check_config_name: None,
            check_config_schema: None,
            fixed_schedule: false,
            object_id: ObjectId::default(),
        }
    }
}

impl TimescaleDbUserDefinedJob {
    pub fn get_create_sql(&self, identifier_quoter: &IdentifierQuoter) -> String {
        let mut sql = "select add_job('".to_string();
        sql.push_str(
            &self
                .function_schema
                .quote(identifier_quoter, TypeOrFunctionName),
        );
        sql.push('.');
        sql.push_str(
            &self
                .function_name
                .quote(identifier_quoter, TypeOrFunctionName),
        );
        sql.push_str("', interval '");
        sql.push_str(&self.schedule_interval.to_postgres());
        sql.push('\'');

        if let Some(config) = &self.config {
            sql.push_str(", config => ");
            sql.push_str(&quote_value_string(config));
        }

        if !self.scheduled {
            sql.push_str(", scheduled => false");
        }

        if let (Some(check_config_name), Some(check_config_schema)) =
            (&self.check_config_name, &self.check_config_schema)
        {
            sql.push_str(", check_config => '");
            sql.push_str(&check_config_schema.quote(identifier_quoter, TypeOrFunctionName));
            sql.push('.');
            sql.push_str(&check_config_name.quote(identifier_quoter, TypeOrFunctionName));
            sql.push('\'');
        }

        if !self.fixed_schedule {
            sql.push_str(", fixed_schedule => false");
        }

        sql.push_str(");");

        sql
    }
}