Prevent racing between `limit_stream` and logging out.

1 files changed, 15 insertions, 0 deletions

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!(