46 lines
999 B
Go
46 lines
999 B
Go
package auth
|
|
|
|
import (
|
|
"gorm.io/gorm"
|
|
"time"
|
|
)
|
|
|
|
type Repository interface {
|
|
CreateRefreshToken(token *Session) error
|
|
ReadRefreshToken(userUuid, tokenUuid string) (Session, error)
|
|
InvalidateRefreshToken(userUuid, refreshUuid string) error
|
|
}
|
|
|
|
type repo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func newRepository(db *gorm.DB) *repo {
|
|
return &repo{db: db}
|
|
}
|
|
|
|
func (r *repo) CreateRefreshToken(token *Session) error {
|
|
return r.db.Create(token).Error
|
|
}
|
|
|
|
func (r *repo) ReadRefreshToken(userUuid, tokenUuid string) (Session, error) {
|
|
var tokenData Session
|
|
|
|
if err := r.db.
|
|
Where("user_uuid = ?", userUuid).
|
|
Where("refresh_uuid = ?", tokenUuid).
|
|
Where("deleted_at IS NULL").
|
|
First(&tokenData).Error; err != nil {
|
|
return Session{}, err
|
|
}
|
|
|
|
return tokenData, nil
|
|
}
|
|
|
|
func (r *repo) InvalidateRefreshToken(userUuid, refreshUuid string) error {
|
|
return r.db.
|
|
Model(&Session{}).
|
|
Where("user_uuid = ?", userUuid).
|
|
Where("refresh_uuid = ?", refreshUuid).
|
|
Update("deleted_at", time.Now().UTC()).Error
|
|
}
|