From 7645411bcf7201e3a4927566da78080dc6a84ccf Mon Sep 17 00:00:00 2001
From: Owen Jacobson <owen@grimoire.ca>
Date: Tue, 1 Oct 2024 20:32:57 -0400
Subject: Prevent racing between `limit_stream` and logging out.

---
 src/repo/token.rs | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'src/repo/token.rs')

diff --git a/src/repo/token.rs b/src/repo/token.rs
index d96c094..1663f5e 100644
--- a/src/repo/token.rs
+++ b/src/repo/token.rs
@@ -47,6 +47,21 @@ impl<'c> Tokens<'c> {
         Ok(secret)
     }
 
+    pub async fn require(&mut self, token: &Id) -> Result<(), sqlx::Error> {
+        sqlx::query_scalar!(
+            r#"
+                select id as "id: Id"
+                from token
+                where id = $1
+            "#,
+            token,
+        )
+        .fetch_one(&mut *self.0)
+        .await?;
+
+        Ok(())
+    }
+
     // Revoke a token by its secret.
     pub async fn revoke(&mut self, token: &Id) -> Result<(), sqlx::Error> {
         sqlx::query_scalar!(
-- 
cgit v1.2.3