diff options
Diffstat (limited to 'src/db/models/endpoint.rs')
-rw-r--r-- | src/db/models/endpoint.rs | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/src/db/models/endpoint.rs b/src/db/models/endpoint.rs index 73009f6..d7a7986 100644 --- a/src/db/models/endpoint.rs +++ b/src/db/models/endpoint.rs @@ -13,6 +13,7 @@ use anyhow::Result; use diesel::prelude::*; use diesel::PgConnection; +use crate::config::EndpointName; use crate::schema::endpoints; use crate::schema::endpoints::*; @@ -29,8 +30,8 @@ struct NewEndpoint<'a> { } impl Endpoint { - pub fn create_or_fetch(database_connection: &PgConnection, ep_name: &str) -> Result<Endpoint> { - let new_ep = NewEndpoint { name: ep_name }; + pub fn create_or_fetch(database_connection: &PgConnection, ep_name: &EndpointName) -> Result<Endpoint> { + let new_ep = NewEndpoint { name: ep_name.as_ref() }; diesel::insert_into(endpoints::table) .values(&new_ep) @@ -38,7 +39,7 @@ impl Endpoint { .execute(database_connection)?; dsl::endpoints - .filter(name.eq(ep_name)) + .filter(name.eq(ep_name.as_ref())) .first::<Endpoint>(database_connection) .map_err(Error::from) } |