flibusta-apps/library_updater
1
0
Fork 0
mirror of https://github.com/flibusta-apps/library_updater.git synced 2025-12-06 07:45:35 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add year to book table

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-10-26 13:11:31 +02:00
parent c68d6041c8
commit 0492a69a39
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/types.rs
View File

@@ -106,6 +106,7 @@ pub struct Book {
pub uploaded: NaiveDate,
pub is_deleted: bool,
pub pages: u64,
pub year: u64,
}
impl FromVecExpression<Book> for Book {
@@ -143,6 +144,10 @@ impl FromVecExpression<Book> for Book {
sql_parse::Expression::Integer(v) => v.0,
_ => panic!("Book.id"),
},
year: match &value[10] {
sql_parse::Expression::Integer(v) => v.0,
_ => panic!("Book.year"),
},
}
}
}
@@ -154,17 +159,19 @@ impl Update for Book {
"
CREATE OR REPLACE FUNCTION update_book(
source_ smallint, remote_id_ int, title_ varchar, lang_ varchar,
file_type_ varchar, uploaded_ date, is_deleted_ boolean, pages_ int
file_type_ varchar, uploaded_ date, is_deleted_ boolean, pages_ int,
year_ smallint
) RETURNS void AS $$
BEGIN
IF EXISTS (SELECT * FROM books WHERE source = source_ AND remote_id = remote_id_) THEN
UPDATE books SET title = title_, lang = lang_, file_type = file_type_,
uploaded = uploaded_, is_deleted = is_deleted_, pages = pages_
uploaded = uploaded_, is_deleted = is_deleted_, pages = pages_,
year = year_
WHERE source = source_ AND remote_id = remote_id_;
RETURN;
END IF;
INSERT INTO books (source, remote_id, title, lang, file_type, uploaded, is_deleted, pages)
VALUES (source_, remote_id_, title_, lang_, file_type_, uploaded_, is_deleted_, pages_);
INSERT INTO books (source, remote_id, title, lang, file_type, uploaded, is_deleted, pages, year)
VALUES (source_, remote_id_, title_, lang_, file_type_, uploaded_, is_deleted_, pages_, year_);
END;
$$ LANGUAGE plpgsql;
"
@@ -180,8 +187,8 @@ impl Update for Book {
source_id: i16,
) -> Result<(), Box<tokio_postgres::Error>> {
match client.execute(
"SELECT update_book($1, $2, cast($3 as varchar), cast($4 as varchar), cast($5 as varchar), $6, $7, $8);",
&[&source_id, &(self.id as i32), &self.title, &self.lang, &self.file_type, &self.uploaded, &self.is_deleted, &(self.pages as i32)]
"SELECT update_book($1, $2, cast($3 as varchar), cast($4 as varchar), cast($5 as varchar), $6, $7, $8, $9);",
&[&source_id, &(self.id as i32), &self.title, &self.lang, &self.file_type, &self.uploaded, &self.is_deleted, &(self.pages as i32), &(self.year as i32)]
).await {
Ok(_) => Ok(()),
Err(err) => Err(Box::new(err)),