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
|
use crate::models;
use diesel::insert_into;
use diesel::prelude::*;
use diesel::result::Error;
use diesel::update;
use entities::*;
pub mod entities;
type DBConn = SqliteConnection;
pub fn store_photo_set(
conn: &DBConn,
mut photo_set: models::PhotoSet,
) -> Result<models::PhotoSet, Error> {
use crate::schema::photo_sets::dsl::*;
if photo_set.id.is_some() {
// update
let form = PhotoSetForm::from(photo_set.clone());
update(photo_sets).set(&form).execute(conn)?;
} else {
// insert
photo_set.id = Some(find_next_photo_set_id(conn)?);
let form = PhotoSetForm::from(photo_set.clone());
insert_into(photo_sets).values(&form).execute(conn)?;
}
Ok(photo_set)
}
pub fn store_product(
conn: &DBConn,
mut product: models::Product,
) -> Result<models::Product, Error> {
use crate::schema::products::dsl::*;
if product.id.is_some() {
// update
let form = ProductForm::from(product.clone());
update(products).set(&form).execute(conn)?;
} else {
// insert
product.id = Some(find_next_product_id(conn)?);
let form = ProductForm::from(product.clone());
insert_into(products).values(&form).execute(conn)?;
}
Ok(product)
}
pub fn find_all(conn: &DBConn) -> Result<Vec<models::Product>, Error> {
use crate::schema::*;
let query = products::table.inner_join(photo_sets::table).select((
products::id,
products::name,
products::description,
products::quantity,
products::cents,
products::featured,
photo_sets::id,
photo_sets::original,
photo_sets::fullsize,
photo_sets::base,
photo_sets::thumbnail,
));
Ok(query
.load::<entities::Product>(conn)?
.into_iter()
.map(|p| p.into())
.collect::<Vec<models::Product>>())
}
pub fn find(conn: &DBConn, id: i32) -> Result<Option<models::Product>, Error> {
use crate::schema::*;
let query = products::table
.inner_join(photo_sets::table)
.filter(products::id.eq(id))
.select((
products::id,
products::name,
products::description,
products::quantity,
products::cents,
products::featured,
photo_sets::id,
photo_sets::original,
photo_sets::fullsize,
photo_sets::base,
photo_sets::thumbnail,
));
let product = query.first::<Product>(conn).map(|p| p.into());
match product {
Ok(p) => Ok(Some(p)),
Err(e) => {
if e == Error::NotFound {
Ok(None)
} else {
Err(e)
}
}
}
}
fn find_next_product_id(conn: &DBConn) -> Result<i32, Error> {
use crate::schema::products::dsl::*;
Ok(products.select(id).order(id.desc()).first::<i32>(conn)? + 1)
}
fn find_next_photo_set_id(conn: &DBConn) -> Result<i32, Error> {
use crate::schema::photo_sets::dsl::*;
Ok(photo_sets.select(id).order(id.desc()).first::<i32>(conn)? + 1)
}
|
