From 13c03a89096ae81f171689e4fdb3c3a6f373d6af Mon Sep 17 00:00:00 2001
From: tobi <tobi.smethurst@protonmail.com>
Date: Mon, 3 Mar 2025 14:39:11 +0100
Subject: [PATCH] wrap UpdateToken with cache

---
 internal/db/bundb/application.go | 19 ++++++++-----------
 1 file changed, 8 insertions(+), 11 deletions(-)

diff --git a/internal/db/bundb/application.go b/internal/db/bundb/application.go
index e266a8ec6..d94c984d0 100644
--- a/internal/db/bundb/application.go
+++ b/internal/db/bundb/application.go
@@ -199,18 +199,15 @@ func (a *applicationDB) PutToken(ctx context.Context, token *gtsmodel.Token) err
 }
 
 func (a *applicationDB) UpdateToken(ctx context.Context, token *gtsmodel.Token, columns ...string) error {
-	_, err := a.db.
-		NewUpdate().
-		Model(token).
-		Column(columns...).
-		Where("? = ?", bun.Ident("id"), token.ID).
-		Exec(ctx)
-	if err != nil {
+	return a.state.Caches.DB.Token.Store(token, func() error {
+		_, err := a.db.
+			NewUpdate().
+			Model(token).
+			Column(columns...).
+			Where("? = ?", bun.Ident("id"), token.ID).
+			Exec(ctx)
 		return err
-	}
-
-	a.state.Caches.DB.Token.Invalidate("ID", token.ID)
-	return nil
+	})
 }
 
 func (a *applicationDB) DeleteTokenByID(ctx context.Context, id string) error {