Регистрация и тесты эндпоинтов

src/database/driver.rs

@@ -1,9 +1,8 @@
 pub mod users {
     use crate::database::models::User;
-    use crate::database::schema::fcm::user_id;
     use crate::database::schema::users::dsl::users;
     use crate::database::schema::users::dsl::*;
-    use diesel::{ExpressionMethods, QueryResult};
+    use diesel::{insert_into, ExpressionMethods, QueryResult};
     use diesel::{PgConnection, SelectableHelper};
     use diesel::{QueryDsl, RunQueryDsl};
     use std::ops::DerefMut;
@@ -31,4 +30,42 @@ pub mod users {
             .select(User::as_select())
             .first(con)
     }
+
+    pub fn contains_by_username(connection: &Mutex<PgConnection>, _username: String) -> bool {
+        let mut lock = connection.lock().unwrap();
+        let con = lock.deref_mut();
+
+        match users
+            .filter(username.eq(_username))
+            .count()
+            .get_result::<i64>(con)
+        {
+            Ok(count) => count > 0,
+            Err(_) => false,
+        }
+    }
+
+    pub fn delete_by_username(connection: &Mutex<PgConnection>, _username: String) -> bool {
+        let mut lock = connection.lock().unwrap();
+        let con = lock.deref_mut();
+
+        match diesel::delete(users.filter(username.eq(_username))).execute(con) {
+            Ok(count) => count > 0,
+            Err(_) => false,
+        }
+    }
+
+    pub fn insert(connection: &Mutex<PgConnection>, user: &User) -> QueryResult<usize> {
+        let mut lock = connection.lock().unwrap();
+        let con = lock.deref_mut();
+
+        insert_into(users).values(user).execute(con)
+    }
+    
+    pub fn insert_or_ignore(connection: &Mutex<PgConnection>, user: &User) -> QueryResult<usize> {
+        let mut lock = connection.lock().unwrap();
+        let con = lock.deref_mut();
+
+        insert_into(users).values(user).on_conflict_do_nothing().execute(con)
+    }
 }