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.
140 lines
3.5 KiB
140 lines
3.5 KiB
package postgres
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"database/sql/driver"
|
|
"encoding/json"
|
|
"errors"
|
|
"github.com/gissleh/stufflog/internal/slerrors"
|
|
"github.com/gissleh/stufflog/models"
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
type projectGroupRepository struct {
|
|
db *sqlx.DB
|
|
}
|
|
|
|
func (r *projectGroupRepository) Find(ctx context.Context, id string) (*models.ProjectGroup, error) {
|
|
res := projectGroupDBO{}
|
|
err := r.db.GetContext(ctx, &res, "SELECT * FROM project_group WHERE project_group_id=$1", id)
|
|
if err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return nil, slerrors.NotFound("Project group")
|
|
}
|
|
|
|
return nil, err
|
|
}
|
|
|
|
return res.ToProjectGroup(), nil
|
|
}
|
|
|
|
func (r *projectGroupRepository) List(ctx context.Context, filter models.ProjectGroupFilter) ([]*models.ProjectGroup, error) {
|
|
res := make([]*projectGroupDBO, 0, 16)
|
|
err := r.db.SelectContext(ctx, &res, "SELECT * FROM project_group WHERE user_id=$1", filter.UserID)
|
|
if err != nil {
|
|
if err == sql.ErrNoRows {
|
|
return []*models.ProjectGroup{}, nil
|
|
}
|
|
|
|
return nil, err
|
|
}
|
|
|
|
res2 := make([]*models.ProjectGroup, 0, len(res))
|
|
for _, pdo := range res {
|
|
res2 = append(res2, pdo.ToProjectGroup())
|
|
}
|
|
|
|
return res2, nil
|
|
}
|
|
|
|
func (r *projectGroupRepository) Insert(ctx context.Context, group models.ProjectGroup) error {
|
|
_, err := r.db.NamedExecContext(ctx, `
|
|
INSERT INTO project_group (
|
|
project_group_id, user_id, name, abbreviation, description, category_names
|
|
) VALUES (
|
|
:project_group_id, :user_id, :name, :abbreviation, :description, :category_names
|
|
)
|
|
`, toProjectGroupDBO(group))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *projectGroupRepository) Update(ctx context.Context, group models.ProjectGroup) error {
|
|
if group.CategoryNames == nil {
|
|
group.CategoryNames = make(map[string]string)
|
|
}
|
|
|
|
_, err := r.db.NamedExecContext(ctx, `
|
|
UPDATE project_group SET
|
|
name = :name,
|
|
abbreviation = :abbreviation,
|
|
description = :description,
|
|
category_names = :category_names
|
|
WHERE project_group_id = :project_group_id
|
|
`, toProjectGroupDBO(group))
|
|
return err
|
|
}
|
|
|
|
func (r *projectGroupRepository) Delete(ctx context.Context, group models.ProjectGroup) error {
|
|
_, err := r.db.ExecContext(ctx, `DELETE FROM project_group WHERE project_group_id=$1`, group.ID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = r.db.ExecContext(ctx, "UPDATE project SET project_group_id=NULL where project_group_id=$1", group.ID)
|
|
if err != nil && err != sql.ErrNoRows {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
type projectGroupDBO struct {
|
|
ID string `db:"project_group_id"`
|
|
UserID string `db:"user_id"`
|
|
Name string `db:"name"`
|
|
Description string `db:"description"`
|
|
Abbreviation string `db:"abbreviation"`
|
|
CategoryNames jsonMap `db:"category_names"`
|
|
}
|
|
|
|
func (dbo *projectGroupDBO) ToProjectGroup() *models.ProjectGroup {
|
|
return &models.ProjectGroup{
|
|
ID: dbo.ID,
|
|
UserID: dbo.UserID,
|
|
Name: dbo.Name,
|
|
Description: dbo.Description,
|
|
Abbreviation: dbo.Abbreviation,
|
|
CategoryNames: dbo.CategoryNames,
|
|
}
|
|
}
|
|
|
|
func toProjectGroupDBO(group models.ProjectGroup) projectGroupDBO {
|
|
return projectGroupDBO{
|
|
ID: group.ID,
|
|
UserID: group.UserID,
|
|
Name: group.Name,
|
|
Description: group.Description,
|
|
Abbreviation: group.Abbreviation,
|
|
CategoryNames: group.CategoryNames,
|
|
}
|
|
}
|
|
|
|
type jsonMap map[string]string
|
|
|
|
func (m jsonMap) Value() (driver.Value, error) {
|
|
return json.Marshal(m)
|
|
}
|
|
|
|
func (m *jsonMap) Scan(value interface{}) error {
|
|
b, ok := value.([]byte)
|
|
if !ok {
|
|
return errors.New("type assertion to []byte failed")
|
|
}
|
|
|
|
return json.Unmarshal(b, m)
|
|
}
|