first commit

src/db/url.rs

@@ -0,0 +1,29 @@
+use self::db::models::Urls;
+use crate::db::schema::urls::dsl::*;
+use crate::*;
+use chrono::NaiveDateTime;
+use diesel::{delete, dsl::insert_into, prelude::*, result::Error};
+
+pub fn get_entry(path: &str) -> Result<Urls, Error> {
+    let connection = &mut establish_connection();
+    urls.find(path).select(Urls::as_select()).first(connection)
+}
+
+pub fn upsert_entry(
+    path: &str,
+    dest: &str,
+    time_to_live: Option<NaiveDateTime>,
+) -> Result<usize, Error> {
+    let connection = &mut establish_connection();
+    insert_into(urls)
+        .values((url.eq(path), destination_url.eq(dest), ttl.eq(time_to_live)))
+        .on_conflict(url)
+        .do_update()
+        .set((destination_url.eq(dest), ttl.eq(time_to_live)))
+        .execute(connection)
+}
+
+pub fn delete_entry(path: &str) -> Result<usize, Error> {
+    let connection = &mut establish_connection();
+    delete(urls.filter(url.eq(path))).execute(connection)
+}