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 std::collections::HashMap;
use std::time::SystemTime;
use sqlx::{QueryBuilder, Sqlite, SqlitePool, Transaction};
use crate::error::AppError;
use super::tag::Tag;
use super::ProductDatabaseHandler;
#[derive(Debug, serde::Serialize, sqlx::FromRow)]
pub struct Product {
pub id: i64,
pub name: String,
pub description: String,
pub uploaddate: i64,
pub updatedate: i64,
}
impl Product {
pub async fn post(tx: &mut Transaction<'static, Sqlite>, name: &str, description: &str, tags: Vec<&str>) -> Result<Product, AppError> {
let now = SystemTime::now().duration_since(std::time::UNIX_EPOCH)?.as_millis() as i64;
let product = sqlx::query_as!(
Product,
"INSERT INTO product (name, description, uploaddate, updatedate) VALUES ($1, $2, $3, $4) RETURNING id, name, description, uploaddate, updatedate",
name,
description,
now,
now)
.fetch_one(&mut **tx)
.await?;
if !tags.is_empty() {
// insert tags that do not exist yet
let mut tags_query_builder: QueryBuilder<Sqlite> = QueryBuilder::new("INSERT INTO tag (name) ");
tags_query_builder.push_values(tags.iter(), | mut b, tag | { b.push_bind(tag); });
tags_query_builder.push(" ON CONFLICT(name) DO NOTHING;");
tags_query_builder.build().execute(&mut **tx).await?;
// fetch all tags of story
let added_tags: Vec<Tag> = {
let mut tags_fetch_query_builder = QueryBuilder::new("WITH input(name) AS (");
tags_fetch_query_builder.push_values(tags.iter(), |mut b, tag| {
b.push_bind(tag);
});
tags_fetch_query_builder.push(
") \
SELECT t.id, t.name \
FROM tag t \
JOIN input i ON i.name = t.name;"
);
tags_fetch_query_builder.build_query_as::<Tag>().fetch_all(&mut **tx).await?
};
println!("{:?}", added_tags);
// bind tags to story if they are not bound yet
let mut producttags_query_builder: QueryBuilder<Sqlite> = QueryBuilder::new("INSERT INTO producttag (pid, tid) ");
producttags_query_builder.push_values(added_tags.iter(), | mut b, added_tag | {
b.push_bind(product.id).push_bind(added_tag.id);
});
producttags_query_builder.push(" ON CONFLICT(pid, tid) DO NOTHING;");
producttags_query_builder
.build().execute(&mut **tx).await?;
}
Ok(product)
}
pub async fn delete(tx: &mut Transaction<'static, Sqlite>, pid: i64) -> Result<(), AppError> {
sqlx::query!("DELETE FROM producttag WHERE pid = $1", pid).execute(&mut **tx).await?;
sqlx::query!("DELETE FROM product WHERE id = $1", pid).execute(&mut **tx).await?;
Ok(())
}
}
pub trait ProductMarker {
fn product_id(&self) -> i64;
}
pub struct SpecificProduct<T: ProductMarker> {
pub product: Product,
pub detail: T,
}
pub async fn get_all_products(pool: &SqlitePool) -> Result<Vec<Product>, AppError> {
Ok(sqlx::query_as!(Product, "SELECT * FROM product")
.fetch_all(pool)
.await?)
}
pub async fn get_all_specificproducts<T: ProductMarker + ProductDatabaseHandler>(
pool: &SqlitePool,
) -> Result<Vec<SpecificProduct<T>>, AppError> {
let product_details = T::get_all(pool).await?;
let products = get_all_products(pool).await?;
let mut details_map: HashMap<i64, T> = product_details
.into_iter()
.map(|detail| (detail.product_id(), detail))
.collect();
let specific_products = products
.into_iter()
.filter_map(|product| {
details_map
.remove(&product.id)
.map(|detail| SpecificProduct { product, detail })
})
.collect();
Ok(specific_products)
}
|
