Next round of `db.DefaultContext` refactor (#27089)

Part of #27065

1 files changed, 8 insertions, 7 deletions

diff --git a/modules/session/db.go b/modules/session/db.go
index f86f7d1e9c..9909f2dc1e 100644
--- a/modules/session/db.go
+++ b/modules/session/db.go
@@ -8,6 +8,7 @@ import (
 	"sync"
 
 	"code.gitea.io/gitea/models/auth"
+	"code.gitea.io/gitea/models/db"
 	"code.gitea.io/gitea/modules/timeutil"
 
 	"gitea.com/go-chi/session"
@@ -71,7 +72,7 @@ func (s *DBStore) Release() error {
 		return err
 	}
 
-	return auth.UpdateSession(s.sid, data)
+	return auth.UpdateSession(db.DefaultContext, s.sid, data)
 }
 
 // Flush deletes all session data.
@@ -97,7 +98,7 @@ func (p *DBProvider) Init(maxLifetime int64, connStr string) error {
 
 // Read returns raw session store by session ID.
 func (p *DBProvider) Read(sid string) (session.RawStore, error) {
-	s, err := auth.ReadSession(sid)
+	s, err := auth.ReadSession(db.DefaultContext, sid)
 	if err != nil {
 		return nil, err
 	}
@@ -117,7 +118,7 @@ func (p *DBProvider) Read(sid string) (session.RawStore, error) {
 
 // Exist returns true if session with given ID exists.
 func (p *DBProvider) Exist(sid string) bool {
-	has, err := auth.ExistSession(sid)
+	has, err := auth.ExistSession(db.DefaultContext, sid)
 	if err != nil {
 		panic("session/DB: error checking existence: " + err.Error())
 	}
@@ -126,12 +127,12 @@ func (p *DBProvider) Exist(sid string) bool {
 
 // Destroy deletes a session by session ID.
 func (p *DBProvider) Destroy(sid string) error {
-	return auth.DestroySession(sid)
+	return auth.DestroySession(db.DefaultContext, sid)
 }
 
 // Regenerate regenerates a session store from old session ID to new one.
 func (p *DBProvider) Regenerate(oldsid, sid string) (_ session.RawStore, err error) {
-	s, err := auth.RegenerateSession(oldsid, sid)
+	s, err := auth.RegenerateSession(db.DefaultContext, oldsid, sid)
 	if err != nil {
 		return nil, err
 	}
@@ -151,7 +152,7 @@ func (p *DBProvider) Regenerate(oldsid, sid string) (_ session.RawStore, err err
 
 // Count counts and returns number of sessions.
 func (p *DBProvider) Count() int {
-	total, err := auth.CountSessions()
+	total, err := auth.CountSessions(db.DefaultContext)
 	if err != nil {
 		panic("session/DB: error counting records: " + err.Error())
 	}
@@ -160,7 +161,7 @@ func (p *DBProvider) Count() int {
 
 // GC calls GC to clean expired sessions.
 func (p *DBProvider) GC() {
-	if err := auth.CleanupSessions(p.maxLifetime); err != nil {
+	if err := auth.CleanupSessions(db.DefaultContext, p.maxLifetime); err != nil {
 		log.Printf("session/DB: error garbage collecting: %v", err)
 	}
 }