birdseed 0.2.0

birdseed enables you to seed a libellis database with fake data, clear all tables, or rebuild all tables per the current embedded migrations
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

use diesel::pg::PgConnection;
use diesel::prelude::*;

use std::error::Error;

use rayon::prelude::*;

use indicatif::ProgressBar;

use crate::pg_pool::Pool;

use crate::models::choice::{Choice, NewChoice, UpdateChoiceData};

/// Populates the database with fake choices
pub fn populate(
    pool: &Pool,
    question_ids: &Vec<i32>,
    row_count: u32,
    bar: &ProgressBar,
) -> Result<Vec<i32>, Box<dyn Error>> {
    bar.set_message(&format!("Seeding {} choices", (row_count * 4)));

    let choice_ids: Vec<i32> = question_ids
        .par_iter()
        .map(|q_id| {
            // For each question, inject 4 random text choices
            (0..4)
                .into_par_iter()
                .map(|_| {
                    let pool = pool.clone();
                    let conn = pool.get().unwrap();

                    let c_title = format!("{}", fake!(Lorem.sentence(1, 4)));
                    let c_type = "text".to_string();
                    let choice = create(&conn, *q_id, &c_type, &c_title);
                    bar.inc(1);
                    choice.id
                })
                .collect()
        })
        .collect::<Vec<Vec<i32>>>()
        .concat();

    Ok(choice_ids)
}

/// Gets a single choices from the database by the given choice id.
pub fn get(conn: &PgConnection, choice_id: i32) -> Result<Choice, diesel::result::Error> {
    use crate::schema::choices::dsl::*;

    choices.find(choice_id).first(conn)
}

/// Gets all surveys filtered by a search parameter
pub fn get_all(conn: &PgConnection) -> Result<Vec<Choice>, diesel::result::Error> {
    use crate::schema::choices::dsl::*;

    choices.get_results(conn)
}

/// Creates a single choice (for a given question) in the database.
pub fn create<'a>(conn: &PgConnection, q_id: i32, c_type: &'a str, c_title: &'a str) -> Choice {
    use crate::schema::choices;

    let new_choice = NewChoice {
        question_id: q_id,
        content_type: c_type,
        title: c_title,
    };

    diesel::insert_into(choices::table)
        .values(&new_choice)
        .get_result(conn)
        .expect("Error saving new choice")
}

/// Updates a choice based on optional fields it receives from a patch route.
pub fn update(conn: &PgConnection, choice_id: i32, data: UpdateChoiceData) -> Choice {
    use crate::schema::choices::dsl::*;

    diesel::update(choices.filter(id.eq(choice_id)))
        .set(&data)
        .get_result(conn)
        .expect("Error updating choice.")
}

/// Deletes a choice based on a choice id.
pub fn delete(conn: &PgConnection, choice_id: i32) -> Result<(), Box<dyn Error>> {
    use crate::schema::choices::dsl::*;

    diesel::delete(choices.filter(id.eq(choice_id))).execute(conn)?;

    Ok(())
}