server/database/drivers/mysqldriver/db.go

package mysqldriver

import (
	"context"
	"database/sql"
	"git.aiterp.net/stufflog/server/database/repositories"
	"github.com/jmoiron/sqlx"
	"github.com/pressly/goose"
	"strings"

	// Mysql Driver
	_ "github.com/go-sql-driver/mysql"
)

type DB struct {
	db              *sqlx.DB
	issues          *issueRepository
	items           *itemRepository
	projects        *projectRepository
	sessions        *sessionRepository
	users           *userRepository
	projectStatuses *projectStatusRepository
}

func (db *DB) Issues() repositories.IssueRepository {
	return db.issues
}

func (db *DB) Items() repositories.ItemRepository {
	return db.items
}

func (db *DB) Projects() repositories.ProjectRepository {
	return db.projects
}

func (db *DB) Session() repositories.SessionRepository {
	return db.sessions
}

func (db *DB) Users() repositories.UserRepository {
	return db.users
}

func (db *DB) ProjectStatuses() repositories.ProjectStatusRepository {
	return db.projectStatuses
}

func (db *DB) Migrate() error {
	err := goose.SetDialect("mysql")
	if err != nil {
		return err
	}

	return goose.Up(db.db.DB, "migrations/mysql")
}

func Open(connectionString string) (*DB, error) {
	// Ensure parseTime is true
	qpos := strings.LastIndexByte(connectionString, '?')
	if qpos != -1 {
		connectionString += "&parseTime=true"
	} else {
		connectionString += "?parseTime=true"
	}

	// Connect to the database
	db, err := sqlx.Connect("mysql", connectionString)
	if err != nil {
		return nil, err
	}

	// Test the connection
	err = db.Ping()
	if err != nil {
		return nil, err
	}

	issues := &issueRepository{db: db}
	items := &itemRepository{db: db}
	projects := &projectRepository{db: db}
	users := &userRepository{db: db}
	sessions := &sessionRepository{db: db}
	projectStatuses := &projectStatusRepository{db: db}

	return &DB{
		db:              db,
		issues:          issues,
		items:           items,
		projects:        projects,
		users:           users,
		sessions:        sessions,
		projectStatuses: projectStatuses,
	}, nil
}

func incCounter(ctx context.Context, tx *sqlx.Tx, kind, name string) (int, error) {
	value := 1
	err := tx.GetContext(ctx, &value, `
		SELECT value FROM counters WHERE kind=? AND name=? FOR UPDATE
	`, kind, name)
	if err != nil && err != sql.ErrNoRows {
		return -1, err
	}

	_, err = tx.ExecContext(ctx, "REPLACE INTO counters (kind, name, value) VALUES (?, ?, ?)", kind, name, value+1)
	if err != nil {
		return -1, err
	}

	return value, nil
}