113 lines
2.8 KiB
Go
113 lines
2.8 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
dbgen "Yimaru-Backend/gen/db"
|
|
"Yimaru-Backend/internal/domain"
|
|
|
|
"github.com/jackc/pgx/v5"
|
|
"github.com/jackc/pgx/v5/pgtype"
|
|
)
|
|
|
|
func programToDomain(p dbgen.Program) domain.Program {
|
|
out := domain.Program{
|
|
ID: p.ID,
|
|
Name: p.Name,
|
|
}
|
|
out.Description = fromPgText(p.Description)
|
|
out.Thumbnail = fromPgText(p.Thumbnail)
|
|
out.CreatedAt = p.CreatedAt.Time
|
|
if p.UpdatedAt.Valid {
|
|
t := p.UpdatedAt.Time
|
|
out.UpdatedAt = &t
|
|
}
|
|
return out
|
|
}
|
|
|
|
func (s *Store) CreateProgram(ctx context.Context, input domain.CreateProgramInput) (domain.Program, error) {
|
|
p, err := s.queries.CreateProgram(ctx, dbgen.CreateProgramParams{
|
|
Name: input.Name,
|
|
Description: toPgText(input.Description),
|
|
Thumbnail: toPgText(input.Thumbnail),
|
|
})
|
|
if err != nil {
|
|
return domain.Program{}, err
|
|
}
|
|
return programToDomain(p), nil
|
|
}
|
|
|
|
func (s *Store) GetProgramByID(ctx context.Context, id int64) (domain.Program, error) {
|
|
p, err := s.queries.GetProgramByID(ctx, id)
|
|
if err != nil {
|
|
if errors.Is(err, pgx.ErrNoRows) {
|
|
return domain.Program{}, pgx.ErrNoRows
|
|
}
|
|
return domain.Program{}, err
|
|
}
|
|
return programToDomain(p), nil
|
|
}
|
|
|
|
func (s *Store) ListPrograms(ctx context.Context, limit, offset int32) ([]domain.Program, int64, error) {
|
|
rows, err := s.queries.ListPrograms(ctx, dbgen.ListProgramsParams{
|
|
Limit: limit,
|
|
Offset: offset,
|
|
})
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
if len(rows) == 0 {
|
|
return []domain.Program{}, 0, nil
|
|
}
|
|
var total int64
|
|
out := make([]domain.Program, 0, len(rows))
|
|
for i, r := range rows {
|
|
if i == 0 {
|
|
total = r.TotalCount
|
|
}
|
|
out = append(out, programToDomain(dbgen.Program{
|
|
ID: r.ID,
|
|
Name: r.Name,
|
|
Description: r.Description,
|
|
Thumbnail: r.Thumbnail,
|
|
CreatedAt: r.CreatedAt,
|
|
UpdatedAt: r.UpdatedAt,
|
|
}))
|
|
}
|
|
return out, total, nil
|
|
}
|
|
|
|
func optionalTextUpdate(val *string) pgtype.Text {
|
|
if val == nil {
|
|
return pgtype.Text{Valid: false}
|
|
}
|
|
return pgtype.Text{String: *val, Valid: true}
|
|
}
|
|
|
|
func (s *Store) UpdateProgram(ctx context.Context, id int64, input domain.UpdateProgramInput) (domain.Program, error) {
|
|
var nameText pgtype.Text
|
|
if input.Name != nil {
|
|
nameText = pgtype.Text{String: *input.Name, Valid: true}
|
|
} else {
|
|
nameText = pgtype.Text{Valid: false}
|
|
}
|
|
p, err := s.queries.UpdateProgram(ctx, dbgen.UpdateProgramParams{
|
|
ID: id,
|
|
Name: nameText,
|
|
Description: optionalTextUpdate(input.Description),
|
|
Thumbnail: optionalTextUpdate(input.Thumbnail),
|
|
})
|
|
if err != nil {
|
|
if errors.Is(err, pgx.ErrNoRows) {
|
|
return domain.Program{}, pgx.ErrNoRows
|
|
}
|
|
return domain.Program{}, err
|
|
}
|
|
return programToDomain(p), nil
|
|
}
|
|
|
|
func (s *Store) DeleteProgram(ctx context.Context, id int64) error {
|
|
return s.queries.DeleteProgram(ctx, id)
|
|
}
|