From 487d97cb019ef1a37d3ef90c6b051ba0389c6d15 Mon Sep 17 00:00:00 2001
From: Tyler Murphy <tylermurphy534@gmail.com>
Date: Sat, 28 Jan 2023 11:52:32 -0500
Subject: tracing

---
 src/database/sessions.rs | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'src/database/sessions.rs')

diff --git a/src/database/sessions.rs b/src/database/sessions.rs
index 7866d76..a2a2f6e 100644
--- a/src/database/sessions.rs
+++ b/src/database/sessions.rs
@@ -1,8 +1,10 @@
 use rusqlite::OptionalExtension;
+use tracing::instrument;
 
 use crate::{database, types::session::Session};
 
 pub fn init() -> Result<(), rusqlite::Error> {
+    tracing::trace!("Retrieving posts page");
     let sql = "
         CREATE TABLE IF NOT EXISTS sessions (
             user_id        INTEGER PRIMARY KEY NOT NULL,
@@ -15,7 +17,9 @@ pub fn init() -> Result<(), rusqlite::Error> {
     Ok(())
 }
 
+#[instrument()]
 pub fn get_session(token: &str) -> Result<Option<Session>, rusqlite::Error> {
+    tracing::trace!("Retrieving session");
     let conn = database::connect()?;
     let mut stmt = conn.prepare("SELECT * FROM sessions WHERE token = ?")?;
     let row = stmt.query_row([token], |row| {
@@ -27,14 +31,18 @@ pub fn get_session(token: &str) -> Result<Option<Session>, rusqlite::Error> {
     Ok(row)
 }
 
+#[instrument()]
 pub fn set_session(user_id: u64, token: &str) -> Result<(), Box<dyn std::error::Error>> {
+    tracing::trace!("Setting new session");
     let conn = database::connect()?;
     let sql = "INSERT OR REPLACE INTO sessions (user_id, token) VALUES (?, ?);";
     conn.execute(sql, (user_id, token))?;
     Ok(())
 }
 
+#[instrument()]
 pub fn delete_session(user_id: u64) -> Result<(), Box<dyn std::error::Error>> {
+    tracing::trace!("Deleting session");
     let conn = database::connect()?;
     let sql = "DELETE FROM sessions WHERE user_id = ?;";
     conn.execute(sql, [user_id])?;
-- 
cgit v1.2.3-freya