models 0.1.3

A migration management library for applications using PostgresSQL, MySQL or SQLite.
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

mod sorter;
use super::*;
pub use sorter::Sorter;
pub use std::collections::HashSet;
pub(crate) struct Queue {
    tables: HashMap<String, Table>,
    sorter: Sorter,
}

impl Queue {
    pub fn new() -> Self {
        Self {
            tables: HashMap::new(),
            sorter: Sorter::new(),
        }
    }
    pub fn len(&self) -> usize {
        self.tables.len()
    }

    pub fn insert(&mut self, table: Table) {
        let table_name = &table.name();
        self.tables.insert(table_name.clone(), table.clone());
        self.sorter.insert(table_name.clone());
        for dep in table.deps() {
            self.sorter.add_dependency(dep, table_name.clone())
        }
    }

    pub fn pop(&mut self) -> Option<Table> {
        self.sorter
            .pop()
            .and_then(|value| self.tables.remove(&value))
    }
    pub fn remove_unregistered(&mut self) {
        self.sorter.remove_unregistered_depedencies()
    }

    pub fn remaining_tables(&self) -> Vec<String> {
        self.tables.clone().into_iter().map(|(k, _)| k).collect()
    }
}