migrate orion repo into monorepo structure

go/internal/human/repository.go

@@ -0,0 +1,108 @@
+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
+}