Fix: rotate access token and add test

This commit is contained in:
2025-12-07 22:06:32 +08:00
parent 8d3cd0260e
commit 1ce2174bdc
7 changed files with 46 additions and 12 deletions

View File

@@ -18,7 +18,7 @@ func NewBunDatabase(db *bun.DB) *BunDatabase {
return &BunDatabase{db: db}
}
func (self *BunDatabase) GetSession(
func (self *BunDatabase) GetSessionByLoginToken(
ctx context.Context,
loginToken string,
) (models.Session, error) {
@@ -35,6 +35,23 @@ func (self *BunDatabase) GetSession(
return ret, nil
}
func (self *BunDatabase) GetSessionByUserId(
ctx context.Context,
userId string,
) (models.Session, error) {
ret := models.Session{
UserId: userId,
}
err := self.db.NewSelect().
Model(&ret).
Where("user_id = ?", userId).
Scan(ctx)
if err != nil {
return models.Session{}, err
}
return ret, nil
}
func (self *BunDatabase) UpdateRefreshToken(
ctx context.Context,
userId string,
@@ -88,6 +105,7 @@ func (self *BunDatabase) UpsertLoginToken(
session := models.Session{
UserId: userId,
LoginToken: token,
IsValid: true,
}
_, err = self.db.NewInsert().
Model(&session).