diff --git a/src/tables.rs b/src/tables.rs index 834ffa4..59013a9 100644 --- a/src/tables.rs +++ b/src/tables.rs @@ -82,9 +82,21 @@ impl User { } } } - pub async fn get(mut db: Connection, id: i32) -> User { + pub async fn get_by_id(mut db: Connection, id: i32) -> User { let res = db - .fetch_one(sqlx::query("SELECT * FROM posts WHERE id = $1;").bind(id)) + .fetch_one(sqlx::query("SELECT * FROM users WHERE id = $1;").bind(id)) + .await + .unwrap(); + User { + id: res.get::("id"), + username: res.get::("username"), + password: res.get::("password"), + token: res.get::, _>("token"), + } + } + pub async fn get_by_token(mut db: Connection, token: String) -> User { + let res = db + .fetch_one(sqlx::query("SELECT * FROM users WHERE token = $1;").bind(token)) .await .unwrap(); User {