You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
1.9 KiB
70 lines
1.9 KiB
package mysqldriver
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"git.aiterp.net/stufflog/server/internal/slerrors"
|
|
"git.aiterp.net/stufflog/server/models"
|
|
sq "github.com/Masterminds/squirrel"
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
type projectStatusRepository struct {
|
|
db *sqlx.DB
|
|
}
|
|
|
|
func (r *projectStatusRepository) Find(ctx context.Context, projectID string, name string) (*models.ProjectStatus, error) {
|
|
projectStatus := models.ProjectStatus{}
|
|
err := r.db.GetContext(ctx, &projectStatus, "SELECT * FROM project_status WHERE project_id=? AND name=?", projectID, name)
|
|
if err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, slerrors.NotFound("Project status")
|
|
}
|
|
|
|
return nil, err
|
|
}
|
|
|
|
return &projectStatus, nil
|
|
}
|
|
|
|
func (r *projectStatusRepository) List(ctx context.Context, filter models.ProjectStatusFilter) ([]*models.ProjectStatus, error) {
|
|
q := sq.Select("*").From("project_status")
|
|
if filter.ProjectID != nil {
|
|
q = q.Where(sq.Eq{"project_id": *filter.ProjectID})
|
|
}
|
|
if filter.MinStage != nil {
|
|
q = q.Where(sq.GtOrEq{"stage": *filter.MinStage})
|
|
}
|
|
if filter.MaxStage != nil {
|
|
q = q.Where(sq.LtOrEq{"stage": *filter.MaxStage})
|
|
}
|
|
q = q.OrderBy("project_id", "stage", "name")
|
|
|
|
query, args, err := q.ToSql()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
statuses := make([]*models.ProjectStatus, 0, 16)
|
|
err = r.db.SelectContext(ctx, &statuses, query, args...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return statuses, nil
|
|
}
|
|
|
|
func (r *projectStatusRepository) Save(ctx context.Context, status models.ProjectStatus) error {
|
|
_, err := r.db.NamedExecContext(ctx, `
|
|
REPLACE INTO project_status (project_id, name, stage, description)
|
|
VALUES (:project_id, :name, :stage, :description)
|
|
`, status)
|
|
|
|
return err
|
|
}
|
|
|
|
func (r *projectStatusRepository) Delete(ctx context.Context, status models.ProjectStatus) error {
|
|
_, err := r.db.ExecContext(ctx, "DELETE FROM project_status WHERE project_id=? AND name=?", status.ProjectID, status.Name)
|
|
|
|
return err
|
|
}
|