talksik
109 lines
2.3 KiB
Go
109 lines
2.3 KiB
Go
package human
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"strings"
|
|
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
"go.jetify.com/typeid"
|
|
)
|
|
|
|
var errNotFound = errors.New("not found")
|
|
|
|
type humanIDPrefix struct{}
|
|
|
|
func (humanIDPrefix) Prefix() string { return "human" }
|
|
|
|
type humanID struct {
|
|
typeid.TypeID[humanIDPrefix]
|
|
}
|
|
|
|
func newHumanID() (humanID, error) {
|
|
return typeid.New[humanID]()
|
|
}
|
|
|
|
func emailPrefix(email string) string {
|
|
return strings.Split(email, "@")[0]
|
|
}
|
|
|
|
type repository interface {
|
|
getByEmail(ctx context.Context, email string) (*Human, error)
|
|
getByID(ctx context.Context, id string) (*Human, error)
|
|
create(ctx context.Context, email string) (*Human, error)
|
|
exists(ctx context.Context, email string) (bool, error)
|
|
}
|
|
|
|
type repositoryImpl struct {
|
|
pool *pgxpool.Pool
|
|
}
|
|
|
|
func newRepository(pool *pgxpool.Pool) repository {
|
|
return &repositoryImpl{pool: pool}
|
|
}
|
|
|
|
func (r *repositoryImpl) getByEmail(ctx context.Context, email string) (*Human, error) {
|
|
var h Human
|
|
err := r.pool.QueryRow(ctx,
|
|
`SELECT id, email, created_at FROM humans WHERE email = $1`,
|
|
email,
|
|
).Scan(&h.ID, &h.Email, &h.CreatedAt)
|
|
if err != nil {
|
|
if errors.Is(err, pgx.ErrNoRows) {
|
|
return nil, errNotFound
|
|
}
|
|
return nil, err
|
|
}
|
|
h.EmailPrefix = emailPrefix(h.Email)
|
|
return &h, nil
|
|
}
|
|
|
|
func (r *repositoryImpl) getByID(ctx context.Context, id string) (*Human, error) {
|
|
var h Human
|
|
err := r.pool.QueryRow(ctx,
|
|
`SELECT id, email, created_at FROM humans WHERE id = $1`,
|
|
id,
|
|
).Scan(&h.ID, &h.Email, &h.CreatedAt)
|
|
if err != nil {
|
|
if errors.Is(err, pgx.ErrNoRows) {
|
|
return nil, errNotFound
|
|
}
|
|
return nil, err
|
|
}
|
|
h.EmailPrefix = emailPrefix(h.Email)
|
|
return &h, nil
|
|
}
|
|
|
|
func (r *repositoryImpl) create(ctx context.Context, email string) (*Human, error) {
|
|
id, err := newHumanID()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var h Human
|
|
err = r.pool.QueryRow(ctx,
|
|
`INSERT INTO humans (id, email) VALUES ($1, $2)
|
|
RETURNING id, email, created_at`,
|
|
id.String(), email,
|
|
).Scan(&h.ID, &h.Email, &h.CreatedAt)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
h.EmailPrefix = emailPrefix(h.Email)
|
|
return &h, nil
|
|
}
|
|
|
|
func (r *repositoryImpl) exists(ctx context.Context, email string) (bool, error) {
|
|
var exists bool
|
|
err := r.pool.QueryRow(ctx,
|
|
`SELECT EXISTS(SELECT 1 FROM humans WHERE email = $1)`,
|
|
email,
|
|
).Scan(&exists)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
return exists, nil
|
|
}
|