aboutsummaryrefslogtreecommitdiffstats
path: root/rust/src
diff options
context:
space:
mode:
authorBryan Newbold <bnewbold@robocracy.org>2018-05-15 23:05:07 -0700
committerBryan Newbold <bnewbold@robocracy.org>2018-05-15 23:05:07 -0700
commit8517c76b92d5fa7041c330438e33f2b08924ac7b (patch)
tree1d2b76814b9b18218d75df67a98c5149b5dd3384 /rust/src
parent386928f9c5860afca72b4ae00c4d5e28efc50bef (diff)
downloadfatcat-8517c76b92d5fa7041c330438e33f2b08924ac7b.tar.gz
fatcat-8517c76b92d5fa7041c330438e33f2b08924ac7b.zip
refactor error handling
Diffstat (limited to 'rust/src')
-rw-r--r--rust/src/api_server.rs199
-rw-r--r--rust/src/lib.rs12
2 files changed, 143 insertions, 68 deletions
diff --git a/rust/src/api_server.rs b/rust/src/api_server.rs
index d70c3d56..200ecdb2 100644
--- a/rust/src/api_server.rs
+++ b/rust/src/api_server.rs
@@ -8,6 +8,7 @@ use database_schema::{changelog, container_edit, container_ident, container_rev,
work_rev};
use diesel::prelude::*;
use diesel::{self, insert_into};
+use errors::*;
use fatcat_api::models;
use fatcat_api::models::*;
use fatcat_api::{Api, ApiError, ContainerIdGetResponse, ContainerLookupGetResponse,
@@ -25,40 +26,21 @@ pub struct Server {
pub db_pool: ConnectionPool,
}
-impl Api for Server {
- fn container_id_get(
- &self,
- id: String,
- _context: &Context,
- ) -> Box<Future<Item = ContainerIdGetResponse, Error = ApiError> + Send> {
+impl Server {
+
+ fn container_id_get_handler(&self, id: String) -> Result<Option<ContainerEntity>> {
let conn = self.db_pool.get().expect("db_pool error");
- let id = match uuid::Uuid::parse_str(&id) {
- Ok(some_uuid) => some_uuid,
- Err(_) => {
- return Box::new(futures::done(Ok(ContainerIdGetResponse::BadRequest(
- Error {
- message: "Failed to parse UUID".to_string(),
- },
- ))));
- }
- };
+ let id = uuid::Uuid::parse_str(&id)?;
- let res: Result<(ContainerIdentRow, ContainerRevRow), _> = container_ident::table
+ let res: ::std::result::Result<(ContainerIdentRow, ContainerRevRow), _> = container_ident::table
.find(id)
.inner_join(container_rev::table)
.first(&conn);
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(
- ContainerIdGetResponse::BadRequest(Error {
- message: "No such container".to_string(),
- }),
- )));
- }
+ Err(diesel::result::Error::NotFound) => return Ok(None),
+ Err(e) => return Err(e.into()),
};
let entity = ContainerEntity {
@@ -72,9 +54,72 @@ impl Api for Server {
redirect: ident.redirect_id.map(|u| u.to_string()),
editgroup: None,
};
- Box::new(futures::done(Ok(ContainerIdGetResponse::FoundEntity(
- entity,
- ))))
+ Ok(Some(entity))
+ }
+
+ fn creator_id_get_handler(&self, id: String) -> Result<Option<CreatorEntity>> {
+ let conn = self.db_pool.get().expect("db_pool error");
+ let id = uuid::Uuid::parse_str(&id)?;
+
+ let res: ::std::result::Result<(CreatorIdentRow, CreatorRevRow), _> = creator_ident::table
+ .find(id)
+ .inner_join(creator_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 = CreatorEntity {
+ name: rev.name,
+ orcid: rev.orcid,
+ 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 file_id_get_handler(&self, id: String) -> Result<Option<FileEntity>> {
+ Ok(None)
+ }
+
+ // TODO:
+ fn work_id_get_handler(&self, id: String) -> Result<Option<WorkEntity>> {
+ Ok(None)
+ }
+
+ // TODO:
+ fn release_id_get_handler(&self, id: String) -> Result<Option<ReleaseEntity>> {
+ Ok(None)
+ }
+}
+
+impl Api for Server {
+
+ fn container_id_get(
+ &self,
+ id: String,
+ _context: &Context,
+ ) -> Box<Future<Item = ContainerIdGetResponse, Error = ApiError> + Send> {
+ match self.container_id_get_handler(id) {
+ Ok(Some(entity)) =>
+ Box::new(futures::done(Ok(ContainerIdGetResponse::FoundEntity(entity)))),
+ Ok(None) =>
+ Box::new(futures::done(Ok(ContainerIdGetResponse::NotFound(
+ ErrorResponse { message: "No such entity".to_string() }),
+ ))),
+ Err(e) =>
+ // TODO: dig in to error type here
+ Box::new(futures::done(Ok(ContainerIdGetResponse::BadRequest(
+ ErrorResponse { message: e.to_string() },
+ )))),
+ }
}
fn container_lookup_get(
@@ -84,7 +129,7 @@ impl Api for Server {
) -> Box<Future<Item = ContainerLookupGetResponse, Error = ApiError> + Send> {
let conn = self.db_pool.get().expect("db_pool error");
- let res: Result<(ContainerIdentRow, ContainerRevRow), _> = container_ident::table
+ let res: ::std::result::Result<(ContainerIdentRow, ContainerRevRow), _> = container_ident::table
.inner_join(container_rev::table)
.first(&conn);
// XXX: actually do a filter/lookup
@@ -95,7 +140,7 @@ impl Api for Server {
return Box::new(futures::done(Ok(
// TODO: UGH, need to add 404 responses everywhere, not 400
//ContainerIdGetResponse::NotFound(
- ContainerLookupGetResponse::BadRequest(Error {
+ ContainerLookupGetResponse::BadRequest(ErrorResponse {
message: "No such container".to_string(),
}),
)));
@@ -121,7 +166,7 @@ impl Api for Server {
fn container_post(
&self,
body: models::ContainerEntity,
- context: &Context,
+ _context: &Context,
) -> Box<Future<Item = ContainerPostResponse, Error = ApiError> + Send> {
println!("{:?}", body);
//let editgroup_id: i64 = body.editgroup.expect("need editgroup_id") as i64;
@@ -166,17 +211,19 @@ impl Api for Server {
id: String,
_context: &Context,
) -> Box<Future<Item = CreatorIdGetResponse, Error = ApiError> + Send> {
- let conn = self.db_pool.get().expect("db_pool error");
- let ce = CreatorEntity {
- orcid: None,
- name: "Dummy Name".into(),
- state: None,
- ident: None,
- revision: None,
- redirect: None,
- editgroup: None,
- };
- Box::new(futures::done(Ok(CreatorIdGetResponse::FoundEntity(ce))))
+ match self.creator_id_get_handler(id) {
+ Ok(Some(entity)) =>
+ Box::new(futures::done(Ok(CreatorIdGetResponse::FoundEntity(entity)))),
+ Ok(None) =>
+ Box::new(futures::done(Ok(CreatorIdGetResponse::NotFound(
+ ErrorResponse { message: "No such entity".to_string() }),
+ ))),
+ Err(e) =>
+ // TODO: dig in to error type here
+ Box::new(futures::done(Ok(CreatorIdGetResponse::BadRequest(
+ ErrorResponse { message: e.to_string() },
+ )))),
+ }
}
fn creator_lookup_get(
@@ -210,15 +257,21 @@ impl Api for Server {
fn file_id_get(
&self,
id: String,
- context: &Context,
+ _context: &Context,
) -> Box<Future<Item = FileIdGetResponse, Error = ApiError> + Send> {
- let context = context.clone();
- println!(
- "file_id_get(\"{}\") - X-Span-ID: {:?}",
- id,
- context.x_span_id.unwrap_or(String::from("<none>")).clone()
- );
- Box::new(futures::failed("Generic failure".into()))
+ match self.file_id_get_handler(id) {
+ Ok(Some(entity)) =>
+ Box::new(futures::done(Ok(FileIdGetResponse::FoundEntity(entity)))),
+ Ok(None) =>
+ Box::new(futures::done(Ok(FileIdGetResponse::NotFound(
+ ErrorResponse { message: "No such entity".to_string() }),
+ ))),
+ Err(e) =>
+ // TODO: dig in to error type here
+ Box::new(futures::done(Ok(FileIdGetResponse::BadRequest(
+ ErrorResponse { message: e.to_string() },
+ )))),
+ }
}
fn file_lookup_get(
@@ -252,15 +305,21 @@ impl Api for Server {
fn work_id_get(
&self,
id: String,
- context: &Context,
+ _context: &Context,
) -> Box<Future<Item = WorkIdGetResponse, Error = ApiError> + Send> {
- let context = context.clone();
- println!(
- "work_id_get(\"{}\") - X-Span-ID: {:?}",
- id,
- context.x_span_id.unwrap_or(String::from("<none>")).clone()
- );
- Box::new(futures::failed("Generic failure".into()))
+ match self.work_id_get_handler(id) {
+ Ok(Some(entity)) =>
+ Box::new(futures::done(Ok(WorkIdGetResponse::FoundEntity(entity)))),
+ Ok(None) =>
+ Box::new(futures::done(Ok(WorkIdGetResponse::NotFound(
+ ErrorResponse { message: "No such entity".to_string() }),
+ ))),
+ Err(e) =>
+ // TODO: dig in to error type here
+ Box::new(futures::done(Ok(WorkIdGetResponse::BadRequest(
+ ErrorResponse { message: e.to_string() },
+ )))),
+ }
}
fn work_post(
@@ -280,15 +339,21 @@ impl Api for Server {
fn release_id_get(
&self,
id: String,
- context: &Context,
+ _context: &Context,
) -> Box<Future<Item = ReleaseIdGetResponse, Error = ApiError> + Send> {
- let context = context.clone();
- println!(
- "release_id_get(\"{}\") - X-Span-ID: {:?}",
- id,
- context.x_span_id.unwrap_or(String::from("<none>")).clone()
- );
- Box::new(futures::failed("Generic failure".into()))
+ match self.release_id_get_handler(id) {
+ Ok(Some(entity)) =>
+ Box::new(futures::done(Ok(ReleaseIdGetResponse::FoundEntity(entity)))),
+ Ok(None) =>
+ Box::new(futures::done(Ok(ReleaseIdGetResponse::NotFound(
+ ErrorResponse { message: "No such entity".to_string() }),
+ ))),
+ Err(e) =>
+ // TODO: dig in to error type here
+ Box::new(futures::done(Ok(ReleaseIdGetResponse::BadRequest(
+ ErrorResponse { message: e.to_string() },
+ )))),
+ }
}
fn release_lookup_get(
diff --git a/rust/src/lib.rs b/rust/src/lib.rs
index b59d6024..8069ad0b 100644
--- a/rust/src/lib.rs
+++ b/rust/src/lib.rs
@@ -20,9 +20,19 @@ pub mod database_models;
pub mod database_schema;
mod errors {
- error_chain!{}
+ // Create the Error, ErrorKind, ResultExt, and Result types
+ error_chain! {
+ foreign_links { Fmt(::std::fmt::Error);
+ Diesel(::diesel::result::Error);
+ Uuid(::uuid::ParseError);
+ Io(::std::io::Error) #[cfg(unix)];
+ }
+ }
}
+#[doc(hidden)]
+pub use errors::*;
+
pub use self::errors::*;
use diesel::pg::PgConnection;
use diesel::prelude::*;