63 lines
1.7 KiB
Go
63 lines
1.7 KiB
Go
package repositories
|
|
|
|
import (
|
|
"GoMembership/internal/models"
|
|
"GoMembership/pkg/errors"
|
|
"database/sql"
|
|
)
|
|
|
|
type UserRepository interface {
|
|
CreateUser(user *models.User) (int64, error)
|
|
FindUserByID(id int) (*models.User, error)
|
|
FindUserByEmail(email string) (*models.User, error)
|
|
}
|
|
|
|
type userRepository struct {
|
|
db *sql.DB
|
|
}
|
|
|
|
func NewUserRepository(db *sql.DB) UserRepository {
|
|
return &userRepository{db}
|
|
}
|
|
|
|
func (repo *userRepository) CreateUser(user *models.User) (int64, error) {
|
|
query := "INSERT INTO users (first_name, last_name, email, password, salt, created_at, updated_at) VALUES (?, ?, ?, ?, ?, ?, ?)"
|
|
result, err := repo.db.Exec(query, user.FirstName, user.LastName, user.Email, user.Password, user.Salt, user.CreatedAt, user.UpdatedAt)
|
|
if err != nil {
|
|
return -1, err
|
|
}
|
|
|
|
lastInsertID, err := result.LastInsertId()
|
|
if err != nil {
|
|
return -1, err
|
|
}
|
|
|
|
return lastInsertID, err
|
|
}
|
|
|
|
func (repo *userRepository) FindUserByID(id int) (*models.User, error) {
|
|
var user models.User
|
|
query := "SELECT id, first_name, last_name, email FROM users WHERE id = ?"
|
|
err := repo.db.QueryRow(query, id).Scan(&user.ID, &user.FirstName, &user.LastName, &user.Email)
|
|
if err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, errors.ErrUserNotFound
|
|
}
|
|
return nil, err
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
func (repo *userRepository) FindUserByEmail(email string) (*models.User, error) {
|
|
var user models.User
|
|
query := "SELECT id, first_name, last_name, email FROM users WHERE email = ?"
|
|
err := repo.db.QueryRow(query, email).Scan(&user.ID, &user.FirstName, &user.LastName, &user.Email)
|
|
if err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, errors.ErrUserNotFound
|
|
}
|
|
return nil, err
|
|
}
|
|
return &user, nil
|
|
}
|