diff options
Diffstat (limited to 'src/database/database.rs')
| -rw-r--r-- | src/database/database.rs | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/database/database.rs b/src/database/database.rs index ca0e018..7444779 100644 --- a/src/database/database.rs +++ b/src/database/database.rs @@ -33,7 +33,7 @@ impl Database { Ok(Database { pool }) } - pub fn insert_user(&self, email: &str, username: &str, password_hash: &str) -> Result<User, DatabaseError> { + pub fn insert_user(&self, email: &str, username: &str, password_hash: &str, is_admin: bool) -> Result<User, DatabaseError> { static QUERY: &str = include_str!("sql/insert_user.sql"); let conn = self.pool .get() @@ -42,12 +42,13 @@ impl Database { .map_err(|e| DatabaseError::Query(e.to_string()))?; Ok(statement - .query_one((email, username, password_hash), |row| { + .query_one((email, username, password_hash, is_admin), |row| { Ok(User::new( row.get("id")?, email.to_owned(), username.to_owned(), password_hash.to_owned(), + is_admin )) }) .map_err(|e| DatabaseError::Query(e.to_string()))? @@ -69,6 +70,7 @@ impl Database { row.get("email")?, row.get("username")?, row.get("password_hash")?, + row.get("is_admin")?, )) }) .optional() @@ -91,6 +93,7 @@ impl Database { row.get("email")?, row.get("username")?, row.get("password_hash")?, + row.get("is_admin")? )) }) .optional() |