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

use crate::object_id::ObjectId;
use crate::quoting::AttemptedKeywordUsage::TypeOrFunctionName;
use crate::quoting::{quote_value_string, IdentifierQuoter, Quotable};
use itertools::Itertools;
use serde::{Deserialize, Serialize};

#[derive(Debug, Eq, PartialEq, Default, Clone, Serialize, Deserialize)]
pub struct PostgresEnum {
    pub name: String,
    pub values: Vec<String>,
    pub comment: Option<String>,
    pub object_id: ObjectId,
}

impl PostgresEnum {
    pub fn get_create_statement(&self, identifier_quoter: &IdentifierQuoter) -> String {
        let mut sql = format!(
            "create type {} as enum (",
            self.name.quote(identifier_quoter, TypeOrFunctionName)
        );
        sql.push_str(&self.values.iter().map(|v| quote_value_string(v)).join(", "));
        sql.push_str(");");

        if let Some(comment) = &self.comment {
            sql.push_str("\ncomment on type ");
            sql.push_str(&self.name.quote(identifier_quoter, TypeOrFunctionName));
            sql.push_str(" is ");
            sql.push_str(&quote_value_string(comment));
            sql.push(';');
        }

        sql
    }
}