diff options
author | Bryan Newbold <bnewbold@robocracy.org> | 2018-05-16 20:55:29 -0700 |
---|---|---|
committer | Bryan Newbold <bnewbold@robocracy.org> | 2018-05-16 20:55:29 -0700 |
commit | 2c9edb5c394d9fc75cf3be49a157763506e1265c (patch) | |
tree | bb6deca3941722ccaaf21bd000bc73c03cbb2f40 /rust | |
parent | 3fce9b2ff8a0daa043357facbbd961a231bd3c02 (diff) | |
download | fatcat-2c9edb5c394d9fc75cf3be49a157763506e1265c.tar.gz fatcat-2c9edb5c394d9fc75cf3be49a157763506e1265c.zip |
implement simplifications and more GETs
Diffstat (limited to 'rust')
-rw-r--r-- | rust/src/api_server.rs | 113 | ||||
-rw-r--r-- | rust/src/database_models.rs | 6 |
2 files changed, 91 insertions, 28 deletions
diff --git a/rust/src/api_server.rs b/rust/src/api_server.rs index 47139363..b081ded9 100644 --- a/rust/src/api_server.rs +++ b/rust/src/api_server.rs @@ -46,7 +46,6 @@ impl Server { let entity = ContainerEntity { issn: rev.issn, publisher: rev.publisher, - parent: None, // TODO: name: rev.name, state: None, // TODO: ident: Some(ident.id.to_string()), @@ -84,19 +83,92 @@ impl Server { Ok(Some(entity)) } - // TODO: fn file_id_get_handler(&self, id: String) -> Result<Option<FileEntity>> { - Ok(None) + let conn = self.db_pool.get().expect("db_pool error"); + let id = uuid::Uuid::parse_str(&id)?; + + let res: ::std::result::Result<(FileIdentRow, FileRevRow), _> = file_ident::table + .find(id) + .inner_join(file_rev::table) + .first(&conn); + + let (ident, rev) = match res { + Ok(r) => r, + Err(diesel::result::Error::NotFound) => return Ok(None), + Err(e) => return Err(e.into()), + }; + + let entity = FileEntity { + sha1: rev.sha1, + size: rev.size.map(|v| v as isize), + url: rev.url, + state: None, // TODO: + ident: Some(ident.id.to_string()), + revision: ident.rev_id.map(|v| v as isize), + redirect: ident.redirect_id.map(|u| u.to_string()), + editgroup: None, + }; + Ok(Some(entity)) } - // TODO: fn work_id_get_handler(&self, id: String) -> Result<Option<WorkEntity>> { - Ok(None) + let conn = self.db_pool.get().expect("db_pool error"); + let id = uuid::Uuid::parse_str(&id)?; + + let res: ::std::result::Result<(WorkIdentRow, WorkRevRow), _> = work_ident::table + .find(id) + .inner_join(work_rev::table) + .first(&conn); + + let (ident, rev) = match res { + Ok(r) => r, + Err(diesel::result::Error::NotFound) => return Ok(None), + Err(e) => return Err(e.into()), + }; + + let entity = WorkEntity { + work_type: rev.work_type, + state: None, // TODO: + ident: Some(ident.id.to_string()), + revision: ident.rev_id.map(|v| v as isize), + redirect: ident.redirect_id.map(|u| u.to_string()), + editgroup: None, + }; + Ok(Some(entity)) } - // TODO: fn release_id_get_handler(&self, id: String) -> Result<Option<ReleaseEntity>> { - Ok(None) + let conn = self.db_pool.get().expect("db_pool error"); + let id = uuid::Uuid::parse_str(&id)?; + + let res: ::std::result::Result<(ReleaseIdentRow, ReleaseRevRow), _> = release_ident::table + .find(id) + .inner_join(release_rev::table) + .first(&conn); + + let (ident, rev) = match res { + Ok(r) => r, + Err(diesel::result::Error::NotFound) => return Ok(None), + Err(e) => return Err(e.into()), + }; + + let entity = ReleaseEntity { + title: rev.title, + release_type: rev.release_type, + //date: rev.date, + doi: rev.doi, + volume: rev.volume, + pages: rev.pages, + issue: rev.issue, + container_id: None, // TODO + work_id: None, // TODO + state: None, // TODO: + ident: Some(ident.id.to_string()), + revision: ident.rev_id.map(|v| v as isize), + redirect: ident.redirect_id.map(|u| u.to_string()), + editgroup: None, + }; + Ok(Some(entity)) } } @@ -137,20 +209,17 @@ impl Api for Server { let (ident, rev) = match res { Ok(r) => r, Err(_) => { - return Box::new(futures::done(Ok( - // TODO: UGH, need to add 404 responses everywhere, not 400 - //ContainerIdGetResponse::NotFound( - ContainerLookupGetResponse::BadRequest(ErrorResponse { + return Box::new(futures::done(Ok(ContainerLookupGetResponse::BadRequest( + ErrorResponse { message: "No such container".to_string(), - }), - ))); + }, + )))); } }; let entity = ContainerEntity { issn: rev.issn, publisher: rev.publisher, - parent: None, // TODO: name: rev.name, state: None, // TODO: ident: Some(ident.id.to_string()), @@ -168,28 +237,24 @@ impl Api for Server { body: models::ContainerEntity, _context: &Context, ) -> Box<Future<Item = ContainerPostResponse, Error = ApiError> + Send> { - println!("{:?}", body); //let editgroup_id: i64 = body.editgroup.expect("need editgroup_id") as i64; // TODO: or find/create let editgroup_id = 1; let conn = self.db_pool.get().expect("db_pool error"); - let name = body.name; - let issn = body.issn; - println!("name={} issn={:?}", name, issn); - let edit: Vec<ContainerEditRow> = diesel::sql_query( - "WITH rev AS ( INSERT INTO container_rev (name, issn) - VALUES ($1, $2) + "WITH rev AS ( INSERT INTO container_rev (name, publisher, issn) + VALUES ($1, $2, $3) RETURNING id ), ident AS ( INSERT INTO container_ident (rev_id) VALUES ((SELECT rev.id FROM rev)) RETURNING id ) INSERT INTO container_edit (editgroup_id, ident_id, rev_id) VALUES - ($3, (SELECT ident.id FROM ident), (SELECT rev.id FROM rev)) + ($4, (SELECT ident.id FROM ident), (SELECT rev.id FROM rev)) RETURNING *", - ).bind::<diesel::sql_types::Text, _>(name) - .bind::<diesel::sql_types::Nullable<diesel::sql_types::Text>, _>(issn) + ).bind::<diesel::sql_types::Text, _>(body.name) + .bind::<diesel::sql_types::Nullable<diesel::sql_types::Text>, _>(body.publisher) + .bind::<diesel::sql_types::Nullable<diesel::sql_types::Text>, _>(body.issn) .bind::<diesel::sql_types::BigInt, _>(editgroup_id) .load(&conn) .unwrap(); diff --git a/rust/src/database_models.rs b/rust/src/database_models.rs index 0995304f..a1f7fc04 100644 --- a/rust/src/database_models.rs +++ b/rust/src/database_models.rs @@ -37,7 +37,6 @@ pub struct ContainerRevRow { pub id: i64, //pub extra_json: Option<serde_json::Value>, pub name: String, - pub parent_ident_id: Option<i64>, pub publisher: Option<String>, pub issn: Option<String>, } @@ -82,10 +81,9 @@ entity_structs!("file_edit", FileEditRow, "file_ident", FileIdentRow); pub struct ReleaseRevRow { pub id: i64, //extra_json: Option<Json>, - pub work_ident_id: Option<Uuid>, + pub work_ident_id: Uuid, pub container_ident_id: Option<Uuid>, - pub title: Option<String>, - pub license: Option<String>, + pub title: String, pub release_type: Option<String>, pub date: Option<String>, pub doi: Option<String>, |