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
//! The postgresql SQL generator backend


use generators::{DatabaseGenerator, TableGenerator};

/// An SQL generator for PGSQL flavoured SQL
#[derive(Default)]
pub struct PGSQL {}

impl PGSQL {
    pub fn new() -> Self {
        return PGSQL {};
    }
}


/// This block implements 
impl DatabaseGenerator for PGSQL {
    
    fn create_table(name: &str) -> String {
        return format!("create table \"{}\"", name);
    }
    
    fn create_table_if_not_exists(name: &str) -> String {
        return format!("create table if not exists \"{}\"", name);
    }
    
    fn drop_table(name: &str) -> String {
        return format!("drop table \"{}\"", name);
    }
    
    fn drop_table_if_exists(name: &str) -> String {
        return format!("drop table if exists \"{}\"", name);
    }
    
    fn rename_table(old: &str, new: &str) -> String {
        return format!("alter table \"{}\" rename to \"{}\"", old, new);
    }
    
    fn alter_table(name: &str) -> String {
        return format!("alter table \"{}\"", name);
    }
}


impl TableGenerator for PGSQL {

    /// Drop an existing column from the table
    fn drop_column(name: &str) -> String {
        return format!("drop column \"{}\"", name);
    }

    /// Rename an existing column
    fn rename_column(old: &str, new: &str) -> String {
        return format!("rename column \"{}\" to \"{}\"", old, new);
    }

    /// Add an auto-incrementing primary key
    fn increments() -> String {
        return format!("\"id\" serial primary key");
    }
    
    /// Add an integer column
    fn integer(name: &str) -> String {
        return format!("\"{}\" int", name);
    }
    
    /// Add a text column
    fn text(name: &str) -> String {
        return format!("\"{}\" text", name);
    }
    
    /// Add a string column
    fn string(name: &str) -> String {
        let limit = 255; // FIXME: make this easily configurable
        return format!("\"{}\" varchar({})", name, limit);
    }
    
    /// Add a timestamp column
    fn timestamp(name: &str) -> String {
        return format!("\"{}\" timestamptz", name);
    }
}