Feat: finish login and add test
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user