server/database/drivers/mysqldriver/activities.go

package mysqldriver

import (
	"context"
	"database/sql"
	"errors"
	"git.aiterp.net/stufflog/server/internal/generate"
	"git.aiterp.net/stufflog/server/internal/slerrors"
	"git.aiterp.net/stufflog/server/models"
	sq "github.com/Masterminds/squirrel"
	"github.com/jmoiron/sqlx"
)

type activityRepository struct {
	db *sqlx.DB
}

func (r *activityRepository) Find(ctx context.Context, id string) (*models.Activity, error) {
	activity := models.Activity{}
	err := r.db.GetContext(ctx, &activity, "SELECT * FROM activity WHERE activity_id=?", id)
	if err != nil {
		if err == sql.ErrNoRows {
			return nil, slerrors.NotFound("Activity")
		}

		return nil, err
	}

	return &activity, nil
}

func (r *activityRepository) List(ctx context.Context, filter models.ActivityFilter) ([]*models.Activity, error) {
	q := sq.Select("*").From("activity")
	if len(filter.ActivityIDs) > 0 {
		q = q.Where(sq.Eq{"activity_id": filter.ActivityIDs})
	}
	if len(filter.ProjectIDs) > 0 {
		q = q.Where(sq.Eq{"project_id": filter.ProjectIDs})
	}

	query, args, err := q.ToSql()
	if err != nil {
		return nil, err
	}

	results := make([]*models.Activity, 0, 16)
	err = r.db.SelectContext(ctx, &results, query, args...)
	if err != nil {
		if err == sql.ErrNoRows {
			return []*models.Activity{}, nil
		}

		return nil, err
	}

	return results, nil
}

func (r *activityRepository) Insert(ctx context.Context, activity models.Activity) (*models.Activity, error) {
	if activity.ProjectID == "" {
		return nil, errors.New("missing project id")
	}

	activity.ID = generate.ActivityID()

	_, err := r.db.NamedExecContext(ctx, `
		INSERT INTO activity (
			activity_id, project_id, name, countable, 
		    unit_is_time, unit_name, unit_value, base_value
		) VALUES (
			:activity_id, :project_id, :name, :countable,
			:unit_is_time, :unit_name, :unit_value, :base_value
		)
	`, activity)
	if err != nil {
		return nil, err
	}

	return &activity, nil
}

func (r *activityRepository) Save(ctx context.Context, activity models.Activity) error {
	_, err := r.db.NamedExecContext(ctx, `
		UPDATE activity SET
			name=:name,
			countable=:countable,
			unit_is_time=:unit_is_time,
			unit_name=:unit_name,
			unit_value=:unit_value,
			base_value=:base_value
		WHERE activity_id=:activity_id
	`, activity)

	return err
}

func (r *activityRepository) Delete(ctx context.Context, activity models.Activity) error {
	_, err := r.db.ExecContext(ctx, "DELETE FROM activity WHERE activity_id=? LIMIT 1;", activity.ID)
	return err
}