60 lines
1.7 KiB
Go
60 lines
1.7 KiB
Go
package auth
|
|
|
|
import "gorm.io/gorm"
|
|
|
|
type Repository interface {
|
|
CreateRefreshToken(token *RefreshToken) error
|
|
FindByToken(token string) (*RefreshToken, error)
|
|
DeleteByToken(token string) error
|
|
DeleteByUserID(userID uint) error
|
|
ReplaceUserRefreshToken(userID uint, newToken *RefreshToken) error
|
|
RotateRefreshToken(oldToken string, newToken *RefreshToken) error
|
|
}
|
|
|
|
type repository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewRepository(db *gorm.DB) Repository {
|
|
return &repository{db: db}
|
|
}
|
|
|
|
func (r *repository) CreateRefreshToken(token *RefreshToken) error {
|
|
return r.db.Create(token).Error
|
|
}
|
|
|
|
func (r *repository) FindByToken(token string) (*RefreshToken, error) {
|
|
var rt RefreshToken
|
|
err := r.db.Select("id", "user_id", "token", "expires_at", "created_at").Where("token = ?", token).First(&rt).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &rt, nil
|
|
}
|
|
|
|
func (r *repository) DeleteByToken(token string) error {
|
|
return r.db.Where("token = ?", token).Delete(&RefreshToken{}).Error
|
|
}
|
|
|
|
func (r *repository) DeleteByUserID(userID uint) error {
|
|
return r.db.Where("user_id = ?", userID).Delete(&RefreshToken{}).Error
|
|
}
|
|
|
|
func (r *repository) ReplaceUserRefreshToken(userID uint, newToken *RefreshToken) error {
|
|
return r.db.Transaction(func(tx *gorm.DB) error {
|
|
if err := tx.Where("user_id = ?", userID).Delete(&RefreshToken{}).Error; err != nil {
|
|
return err
|
|
}
|
|
return tx.Create(newToken).Error
|
|
})
|
|
}
|
|
|
|
func (r *repository) RotateRefreshToken(oldToken string, newToken *RefreshToken) error {
|
|
return r.db.Transaction(func(tx *gorm.DB) error {
|
|
if err := tx.Where("token = ?", oldToken).Delete(&RefreshToken{}).Error; err != nil {
|
|
return err
|
|
}
|
|
return tx.Create(newToken).Error
|
|
})
|
|
}
|