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) }