add activities.

database/drivers/mysqldriver/activities.go

@@ -0,0 +1,100 @@
+package mysqldriver
+
+import (
+	"context"
+	"database/sql"
+	"errors"
+	"git.aiterp.net/stufflog/server/internal/generate"
+	"git.aiterp.net/stufflog/server/internal/xlerrors"
+	"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, xlerrors.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
+}

database/drivers/mysqldriver/db.go

@@ -14,6 +14,7 @@ import (
 
 type DB struct {
 	db              *sqlx.DB
+	activities      *activityRepository
 	issues          *issueRepository
 	items           *itemRepository
 	projects        *projectRepository
@@ -22,6 +23,10 @@ type DB struct {
 	projectStatuses *projectStatusRepository
 }
 
+func (db *DB) Activities() repositories.ActivityRepository {
+	return db.activities
+}
+
 func (db *DB) Issues() repositories.IssueRepository {
 	return db.issues
 }
@@ -76,6 +81,7 @@ func Open(connectionString string) (*DB, error) {
 		return nil, err
 	}
 
+	activities := &activityRepository{db: db}
 	issues := &issueRepository{db: db}
 	items := &itemRepository{db: db}
 	projects := &projectRepository{db: db}
@@ -85,6 +91,7 @@ func Open(connectionString string) (*DB, error) {
 
 	return &DB{
 		db:              db,
+		activities:      activities,
 		issues:          issues,
 		items:           items,
 		projects:        projects,