schema-sql-generator 0.2.0

A set of tools to manage relational database schemas
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

use crate::common::generator_context::GeneratorContext;
use crate::common::sql_writer::SqlWriter;
use schema_model::model::types::OtherSqlOrder;

pub trait OtherSqlGenerator {
    fn output_other_sql_top(&self);
    fn output_other_sql_bottom(&self);
    fn output_other_sql(&self, writer: &mut SqlWriter, statement_separator: &str, sql: &str);
}

pub struct DefaultOtherSqlGenerator {
    context: GeneratorContext,
}

impl DefaultOtherSqlGenerator {
    pub fn new(context: GeneratorContext) -> Self {
        Self { context }
    }

    pub fn context(&self) -> &GeneratorContext {
        &self.context
    }
}

impl OtherSqlGenerator for DefaultOtherSqlGenerator {
    fn output_other_sql_top(&self) {
        let database_type = self.context.settings().database_type();
        let statement_separator = self.context.settings().statement_separator();
        let database_model = self.context.settings().database_model();

        self.context.with_writer(|writer| {
            database_model.schemas().iter().for_each(|schema| {
                schema
                    .other_sql()
                    .iter()
                    .filter(|sql| sql.database_type() == database_type)
                    .filter(|sql| sql.order() == OtherSqlOrder::Top)
                    .filter(|sql| !sql.sql().is_empty())
                    .for_each(|sql| {
                        self.output_other_sql(writer, statement_separator, sql.sql());
                    })
            });
        });
    }

    fn output_other_sql_bottom(&self) {
        let database_type = self.context.settings().database_type();
        let statement_separator = self.context.settings().statement_separator();
        let database_model = self.context.settings().database_model();

        self.context.with_writer(|writer| {
            database_model.schemas().iter().for_each(|schema| {
                schema
                    .other_sql()
                    .iter()
                    .filter(|sql| sql.database_type() == database_type)
                    .filter(|sql| sql.order() == OtherSqlOrder::Bottom)
                    .filter(|sql| !sql.sql().is_empty())
                    .for_each(|sql| {
                        self.output_other_sql(writer, statement_separator, sql.sql());
                    })
            });
        });
    }

    fn output_other_sql(&self, writer: &mut SqlWriter, statement_separator: &str, sql: &str) {
        writer.print(sql);
        writer.println(statement_separator);
        writer.newline();
    }
}