Feat: finish login and add test

This commit is contained in:
2025-12-07 17:42:28 +08:00
parent 2eb8a18c40
commit 3970dd5ebd
15 changed files with 110 additions and 41 deletions

View File

@@ -57,19 +57,18 @@ func (self *BunDatabase) UpdateRefreshToken(
return models.Session{}, err
}
err = self.db.NewUpdate().
_, err = self.db.NewUpdate().
Model((*models.Session)(nil)).
Set("refresh_token = ?", ret.RefreshToken).
Where("user_id = ?", ret.UserId).
Returning("*").
Scan(ctx, &ret)
Where("user_id = ?", userId).
Exec(ctx)
if err != nil {
return models.Session{}, err
}
return ret, nil
}
func (self *BunDatabase) UpdateLoginToken(
func (self *BunDatabase) UpsertLoginToken(
ctx context.Context,
userId string,
) (string, error) {
@@ -78,10 +77,14 @@ func (self *BunDatabase) UpdateLoginToken(
return "", err
}
_, err = self.db.NewUpdate().
Model((*models.Session)(nil)).
Set("login_token = ?", token).
Where("user_id = ?", userId).
session := models.Session{
UserId: userId,
LoginToken: token,
}
_, err = self.db.NewInsert().
Model(&session).
On("CONFLICT (user_id) DO UPDATE").
Set("login_token = EXCLUDED.login_token").
Exec(ctx)
if err != nil {
return "", err