gogmagog/store/postgres.go

package store

import (
	"gitea.deepak.science/deepak/gogmagog/models"
	"gitea.deepak.science/deepak/gogmagog/util"
	"github.com/jmoiron/sqlx"
)

type postgresStore struct {
	db *sqlx.DB
}

// GetPostgresStore provides a store to back the model based on a postgres connection.
func GetPostgresStore(db *sqlx.DB) (models.Store, error) {

	db.MapperFunc(util.ToSnake)
	return &postgresStore{db: db}, nil
}

func (store *postgresStore) SelectActions() ([]*models.Action, error) {
	actions := make([]*models.Action, 0)
	err := store.db.Select(&actions, "SELECT action_id, action_description, estimated_chunks, completed_chunks, completed_on, created_at, updated_at, plan_id FROM actions")
	if err != nil {
		return nil, err
	}
	return actions, nil
}

func (store *postgresStore) SelectActionsByPlanID(plan *models.Plan) ([]*models.Action, error) {
	queryString := store.db.Rebind("SELECT action_id, action_description, estimated_chunks, completed_chunks, completed_on, created_at, updated_at, plan_id FROM actions WHERE plan_id = ?")
	actions := make([]*models.Action, 0)
	err := store.db.Select(&actions, queryString, plan.PlanID)
	if err != nil {
		return nil, err
	}
	return actions, nil
}

func (store *postgresStore) SelectActionByID(id int) (*models.Action, error) {
	action := models.Action{}
	err := store.db.Get(&action, store.db.Rebind("SELECT action_id, action_description, estimated_chunks, completed_chunks, completed_on, created_at, updated_at, plan_id FROM actions WHERE action_id = ?"), id)
	if err != nil {
		return nil, err
	}
	return &action, nil
}

func (store *postgresStore) InsertAction(action *models.Action) (int, error) {
	queryString := store.db.Rebind(
		`INSERT INTO actions (action_description,
			estimated_chunks,
			completed_chunks,
			completed_on,
			plan_id) VALUES (?, ?, ?, ?, ?) RETURNING action_id`,
	)
	tx := store.db.MustBegin()
	var id int
	err := tx.Get(
		&id,
		queryString,
		action.ActionDescription,
		action.EstimatedChunks,
		action.CompletedChunks,
		action.CompletedOn,
		action.PlanID,
	)
	if err != nil {
		tx.Rollback()
		return -1, err
	}
	err = tx.Commit()
	if err != nil {
		return -1, err
	}
	return id, nil
}

func (store *postgresStore) UpdateAction(action *models.Action) error {
	query := `UPDATE actions SET
		action_description = :action_description,
		estimated_chunks = :estimated_chunks,
		completed_chunks = :completed_chunks,
		completed_on = :completed_on,
		plan_id = :plan_id
		WHERE action_id = :action_id`
	tx := store.db.MustBegin()
	_, err := store.db.NamedExec(query, action)
	if err != nil {
		tx.Rollback()
		return err
	}
	err = tx.Commit()
	if err != nil {
		return err
	}
	return nil

}

func (store *postgresStore) SelectPlans() ([]*models.Plan, error) {
	plans := make([]*models.Plan, 0)
	err := store.db.Select(&plans, "SELECT plan_id, plan_date FROM plans")
	if err != nil {
		return nil, err
	}
	return plans, nil
}

func (store *postgresStore) SelectPlanByID(id int) (*models.Plan, error) {
	plan := models.Plan{}
	err := store.db.Get(&plan, store.db.Rebind("SELECT plan_id, plan_date FROM plans WHERE plan_id = ?"), id)
	if err != nil {
		return nil, err
	}
	return &plan, nil
}

func (store *postgresStore) InsertPlan(plan *models.Plan) (int, error) {
	queryString := store.db.Rebind("INSERT INTO plans (plan_date) VALUES (?) RETURNING plan_id")
	tx := store.db.MustBegin()
	var id int
	err := tx.Get(&id, queryString, plan.PlanDate)
	if err != nil {
		tx.Rollback()
		return -1, err
	}
	err = tx.Commit()
	if err != nil {
		return -1, err
	}
	return id, nil
}

func (store *postgresStore) ConnectionLive() error {
	return store.db.Ping()
}

func (store *postgresStore) SelectUserByUsername(username string) (*models.User, error) {
	user := models.User{}
	err := store.db.Get(&user, store.db.Rebind("SELECT user_id, username, display_name, password FROM users WHERE username = ?"), username)
	if err != nil {
		return nil, err
	}
	return &user, nil
}

func (store *postgresStore) InsertUser(user *models.User) (int, error) {
	queryString := store.db.Rebind("INSERT INTO users (username, display_name, password) VALUES (?, ?, ?) RETURNING user_id")
	tx := store.db.MustBegin()
	var id int
	err := tx.Get(&id, queryString, user.Username, user.DisplayName, user.Password)
	if err != nil {
		tx.Rollback()
		return -1, err
	}
	err = tx.Commit()
	if err != nil {
		return -1, err
	}
	return id, nil
}