safe-migrate 0.8.0

Check PostgreSQL migrations against a synchronized database baseline
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150

use super::Resolver;
use crate::_internal::analysis::facts::{
    AlterPublicationActionFact, AlterPublicationFact, AlterSubscriptionFact, CreatePublicationFact,
    CreateSubscriptionFact, DropPublicationFact, DropSubscriptionFact, PublicationObjectFact,
    PublicationScope,
};
use crate::_internal::analysis::mutations::{
    AlterPublicationMutation, AlterSubscriptionMutation, CreatePublicationMutation,
    CreateSubscriptionMutation, DropPublicationMutation, DropSubscriptionMutation, Mutation,
};
use crate::_internal::analysis::state::AnalysisState;
use crate::_internal::ast::identifiers::{Ident, QualifiedName};

impl Resolver {
    fn resolve_publication_object(
        object: &PublicationObjectFact,
        state: &AnalysisState,
    ) -> PublicationObjectFact {
        match object {
            PublicationObjectFact::Table {
                name,
                only,
                include_partitions,
                columns,
                row_filter,
            } => {
                let id = Self::resolve_relation_lookup_name(name, state);
                PublicationObjectFact::Table {
                    name: QualifiedName::new(
                        Some(Ident::new(id.schema, true)),
                        Ident::new(id.name, true),
                    ),
                    only: *only,
                    include_partitions: *include_partitions,
                    columns: columns.clone(),
                    row_filter: row_filter.clone(),
                }
            }
            PublicationObjectFact::CurrentSchemaShorthand => PublicationObjectFact::SchemaTables {
                schema: state
                    .local
                    .search_path
                    .first()
                    .cloned()
                    .unwrap_or_else(|| "public".to_string()),
                row_filter: None,
            },
            other => other.clone(),
        }
    }

    fn resolve_publication_scope(
        scope: &PublicationScope,
        state: &AnalysisState,
    ) -> PublicationScope {
        match scope {
            PublicationScope::AllTables { except } => PublicationScope::AllTables {
                except: except.clone(),
            },
            PublicationScope::Explicit(objects) => PublicationScope::Explicit(
                objects
                    .iter()
                    .map(|object| Self::resolve_publication_object(object, state))
                    .collect(),
            ),
        }
    }

    fn resolve_alter_publication_action(
        action: &AlterPublicationActionFact,
        state: &AnalysisState,
    ) -> AlterPublicationActionFact {
        match action {
            AlterPublicationActionFact::AddObjects(objects) => {
                AlterPublicationActionFact::AddObjects(
                    objects
                        .iter()
                        .map(|object| Self::resolve_publication_object(object, state))
                        .collect(),
                )
            }
            AlterPublicationActionFact::DropObjects(objects) => {
                AlterPublicationActionFact::DropObjects(
                    objects
                        .iter()
                        .map(|object| Self::resolve_publication_object(object, state))
                        .collect(),
                )
            }
            AlterPublicationActionFact::SetObjects(scope) => {
                AlterPublicationActionFact::SetObjects(Self::resolve_publication_scope(
                    scope, state,
                ))
            }
            other => other.clone(),
        }
    }

    pub(super) fn resolve_create_publication(
        fact: &CreatePublicationFact,
        state: &AnalysisState,
    ) -> Mutation {
        Mutation::CreatePublication(CreatePublicationMutation {
            name: fact.name.clone(),
            scope: Self::resolve_publication_scope(&fact.scope, state),
            params: fact.params.clone(),
        })
    }

    pub(super) fn resolve_alter_publication(
        fact: &AlterPublicationFact,
        state: &AnalysisState,
    ) -> Mutation {
        Mutation::AlterPublication(AlterPublicationMutation {
            name: fact.name.clone(),
            action: Self::resolve_alter_publication_action(&fact.action, state),
        })
    }

    pub(super) fn resolve_drop_publication(fact: &DropPublicationFact) -> Mutation {
        Mutation::DropPublication(DropPublicationMutation {
            names: fact.names.clone(),
            if_exists: fact.if_exists,
            cascade: fact.cascade,
        })
    }

    pub(super) fn resolve_create_subscription(fact: &CreateSubscriptionFact) -> Mutation {
        Mutation::CreateSubscription(CreateSubscriptionMutation {
            name: fact.name.clone(),
            connection: fact.connection.clone(),
            publications: fact.publications.clone(),
            params: fact.params.clone(),
        })
    }

    pub(super) fn resolve_alter_subscription(fact: &AlterSubscriptionFact) -> Mutation {
        Mutation::AlterSubscription(AlterSubscriptionMutation {
            name: fact.name.clone(),
            action: fact.action.clone(),
        })
    }

    pub(super) fn resolve_drop_subscription(fact: &DropSubscriptionFact) -> Mutation {
        Mutation::DropSubscription(DropSubscriptionMutation {
            name: fact.name.clone(),
            if_exists: fact.if_exists,
        })
    }
}