model refactor
This commit is contained in:
parent
42b53012dc
commit
0fdc00e2cc
3 changed files with 13 additions and 9 deletions
|
|
@ -6,8 +6,8 @@ import (
|
|||
)
|
||||
|
||||
type Repository interface {
|
||||
CreateRefreshToken(token *RefreshToken) error
|
||||
ReadRefreshToken(userUuid, tokenUuid, sessionUuid string) (RefreshToken, error)
|
||||
CreateRefreshToken(token *Session) error
|
||||
ReadRefreshToken(userUuid, tokenUuid, sessionUuid string) (Session, error)
|
||||
InvalidateRefreshToken(userUuid, refreshUuid, sessionUuid string) error
|
||||
}
|
||||
|
||||
|
|
@ -19,12 +19,12 @@ func newRepository(db *gorm.DB) *repo {
|
|||
return &repo{db: db}
|
||||
}
|
||||
|
||||
func (r *repo) CreateRefreshToken(token *RefreshToken) error {
|
||||
func (r *repo) CreateRefreshToken(token *Session) error {
|
||||
return r.db.Create(token).Error
|
||||
}
|
||||
|
||||
func (r *repo) ReadRefreshToken(userUuid, tokenUuid, sessionUuid string) (RefreshToken, error) {
|
||||
var tokenData RefreshToken
|
||||
func (r *repo) ReadRefreshToken(userUuid, tokenUuid, sessionUuid string) (Session, error) {
|
||||
var tokenData Session
|
||||
|
||||
if err := r.db.
|
||||
Where("user_uuid = ?", userUuid).
|
||||
|
|
@ -32,7 +32,7 @@ func (r *repo) ReadRefreshToken(userUuid, tokenUuid, sessionUuid string) (Refres
|
|||
Where("session_uuid = ?", sessionUuid).
|
||||
Where("deleted_at IS NULL").
|
||||
First(&tokenData).Error; err != nil {
|
||||
return RefreshToken{}, err
|
||||
return Session{}, err
|
||||
}
|
||||
|
||||
return tokenData, nil
|
||||
|
|
@ -40,7 +40,7 @@ func (r *repo) ReadRefreshToken(userUuid, tokenUuid, sessionUuid string) (Refres
|
|||
|
||||
func (r *repo) InvalidateRefreshToken(userUuid, refreshUuid, sessionUuid string) error {
|
||||
return r.db.
|
||||
Model(&RefreshToken{}).
|
||||
Model(&Session{}).
|
||||
Where("user_uuid = ?", userUuid).
|
||||
Where("refresh_uuid = ?", refreshUuid).
|
||||
Where("session_uuid = ?", sessionUuid).
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue