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

use crate::quoting::{quote_value_string, AttemptedKeywordUsage, Quotable};
use crate::{IdentifierQuoter, ObjectId, PostgresSchema};
use serde::{Deserialize, Serialize};

#[derive(Debug, Eq, PartialEq, Clone, Default, Serialize, Deserialize)]
pub struct PostgresDomain {
    pub name: String,
    pub object_id: ObjectId,
    pub base_type_name: String,
    pub default_value: Option<String>,
    pub constraint: Option<PostgresDomainConstraint>,
    pub not_null: bool,
    pub description: Option<String>,
    pub depends_on: Vec<ObjectId>,
    pub data_type_length: Option<i32>,
}

#[derive(Debug, Eq, PartialEq, Clone, Serialize, Deserialize)]
pub struct PostgresDomainConstraint {
    pub name: String,
    pub definition: String,
}

impl PostgresDomain {
    pub fn get_create_sql(
        &self,
        schema: &PostgresSchema,
        identifier_quoter: &IdentifierQuoter,
    ) -> String {
        let mut sql = format!(
            "create domain {}.{} as {}",
            schema
                .name
                .quote(identifier_quoter, AttemptedKeywordUsage::TypeOrFunctionName),
            self.name
                .quote(identifier_quoter, AttemptedKeywordUsage::TypeOrFunctionName),
            self.base_type_name
        );

        if let Some(length) = self.data_type_length {
            sql.push_str(&format!("({})", length));
        }
        if let Some(default_value) = &self.default_value {
            sql.push_str(&format!(" default {}", default_value));
        }
        if self.not_null {
            sql.push_str(" not null");
        }
        if let Some(constraint) = &self.constraint {
            sql.push_str(&format!(
                " constraint {} check {}",
                constraint
                    .name
                    .quote(identifier_quoter, AttemptedKeywordUsage::TypeOrFunctionName),
                constraint.definition
            ));
        }
        sql.push(';');

        if let Some(description) = &self.description {
            sql.push_str(&format!(
                "\ncomment on domain {}.{} is {};",
                schema
                    .name
                    .quote(identifier_quoter, AttemptedKeywordUsage::TypeOrFunctionName),
                self.name
                    .quote(identifier_quoter, AttemptedKeywordUsage::TypeOrFunctionName),
                quote_value_string(description)
            ));
        }

        sql
    }
}