ostiwe-com/status
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases 1 Wiki Activity

refactor: Provide context to user repository

Browse Source
This commit is contained in:
Nikolay Anoshin
2025-08-20 11:17:39 +03:00
parent 32780303a3
commit 2a9dfcf998
4 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
repository/user.go
View File

@@ -1,13 +1,15 @@
package repository
import (
"context"
"git.ostiwe.com/ostiwe-com/status/model"
"git.ostiwe.com/ostiwe-com/status/modules/db"
)
type User interface {
FindByLogin(login string) (*model.User, error)
FindByID(ID uint64) (*model.User, error)
FindByLogin(ctx context.Context, login string) (*model.User, error)
FindByID(ctx context.Context, ID uint64) (*model.User, error)
}
type userRepository struct {
@@ -18,14 +20,14 @@ func NewUserRepository() User {
return &userRepository{repository{db: db.Global}}
}
func (u userRepository) FindByLogin(login string) (*model.User, error) {
func (u userRepository) FindByLogin(ctx context.Context, login string) (*model.User, error) {
var user *model.User
return user, u.db.Find(&user, "login = ?", login).Error
return user, u.db.WithContext(ctx).Find(&user, "login = ?", login).Error
}
func (u userRepository) FindByID(ID uint64) (*model.User, error) {
func (u userRepository) FindByID(ctx context.Context, ID uint64) (*model.User, error) {
var user *model.User
return user, u.db.Find(&user, "id = ?", ID).Error
return user, u.db.WithContext(ctx).Find(&user, "id = ?", ID).Error
}