106 lines
2.5 KiB
Go
106 lines
2.5 KiB
Go
package status
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"log"
|
|
"madsky.ru/go-finance/internal/database"
|
|
"madsky.ru/go-finance/internal/status"
|
|
)
|
|
|
|
type Repository interface {
|
|
Find(ctx context.Context) ([]*status.Status, error)
|
|
FindOne(ctx context.Context, id uint64) (*status.Status, error)
|
|
Create(ctx context.Context, dto *status.CreateStatusDTO) (*status.Status, error)
|
|
Update(ctx context.Context, id uint64, issue *status.Status) error
|
|
Remove(ctx context.Context, id uint64) (uint64, error)
|
|
}
|
|
|
|
type repository struct {
|
|
client database.Client
|
|
}
|
|
|
|
func NewRepository(client database.Client) Repository {
|
|
return &repository{
|
|
client: client,
|
|
}
|
|
}
|
|
|
|
func (r *repository) Find(ctx context.Context) ([]*status.Status, error) {
|
|
query := "select id, name, description, position from statuses order by id asc"
|
|
|
|
rows, err := r.client.Query(ctx, query)
|
|
if err != nil {
|
|
log.Println("rows", err)
|
|
return nil, err
|
|
}
|
|
|
|
statuses := make([]*status.Status, 0)
|
|
|
|
for rows.Next() {
|
|
var n status.Status
|
|
|
|
err = rows.Scan(&n.ID, &n.Name, &n.Description, &n.Position)
|
|
if err != nil {
|
|
log.Println("scan", err)
|
|
return nil, err
|
|
}
|
|
statuses = append(statuses, &n)
|
|
}
|
|
|
|
if err = rows.Err(); err != nil {
|
|
log.Println("rows err", err)
|
|
return nil, err
|
|
}
|
|
|
|
return statuses, nil
|
|
}
|
|
|
|
func (r *repository) FindOne(ctx context.Context, id uint64) (*status.Status, error) {
|
|
query := "select id, name, description, position from statuses where id = $1"
|
|
|
|
var s status.Status
|
|
if err := r.client.QueryRow(ctx, query, id).Scan(&s.ID, &s.Name, &s.Description, &s.Position); err != nil {
|
|
fmt.Println(err)
|
|
return nil, err
|
|
}
|
|
|
|
return &s, nil
|
|
}
|
|
|
|
func (r *repository) Create(ctx context.Context, dto *status.CreateStatusDTO) (*status.Status, error) {
|
|
q := "insert into statuses (name, description) values ($1, $2) returning id, name, description, position"
|
|
|
|
var s status.Status
|
|
|
|
if err := r.client.QueryRow(ctx, q, dto.Name, dto.Description).Scan(&s.ID, &s.Name, &s.Description, &s.Position); err != nil {
|
|
fmt.Println(fmt.Sprintf("error %v", err))
|
|
return nil, err
|
|
}
|
|
return &s, nil
|
|
}
|
|
|
|
func (r *repository) Update(ctx context.Context, id uint64, issue *status.Status) error {
|
|
//TODO implement me
|
|
panic("implement me")
|
|
}
|
|
|
|
func (r *repository) Remove(ctx context.Context, id uint64) (uint64, error) {
|
|
q := "delete from statuses where id=$1"
|
|
|
|
tag, err := r.client.Exec(ctx, q, id)
|
|
if err != nil {
|
|
log.Println("exec error", err)
|
|
return 0, err
|
|
}
|
|
|
|
rowsAffected := tag.RowsAffected()
|
|
|
|
if rowsAffected == 0 {
|
|
return 0, errors.New("status not found")
|
|
}
|
|
|
|
return uint64(rowsAffected), nil
|
|
}
|