diff options
Diffstat (limited to 'src/db/models/endpoint.rs')
-rw-r--r-- | src/db/models/endpoint.rs | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/src/db/models/endpoint.rs b/src/db/models/endpoint.rs new file mode 100644 index 0000000..ab30192 --- /dev/null +++ b/src/db/models/endpoint.rs @@ -0,0 +1,36 @@ +use anyhow::Error; +use anyhow::Result; +use diesel::PgConnection; +use diesel::prelude::*; + +use crate::schema::endpoints::*; +use crate::schema::endpoints; + +#[derive(Queryable)] +pub struct Endpoint { + pub id: i32, + pub name: String, +} + +#[derive(Insertable)] +#[table_name="endpoints"] +struct NewEndpoint<'a> { + pub name: &'a str, +} + +impl Endpoint { + pub fn create_or_fetch(database_connection: &PgConnection, ep_name: &str) -> Result<Endpoint> { + let new_ep = NewEndpoint { name: ep_name }; + + diesel::insert_into(endpoints::table) + .values(&new_ep) + .on_conflict_do_nothing() + .execute(database_connection)?; + + dsl::endpoints + .filter(name.eq(ep_name)) + .first::<Endpoint>(database_connection) + .map_err(Error::from) + } +} + |