stufflog3/ports/mysql/db.go

package mysql

import (
	"context"
	"database/sql"
	"fmt"
	"git.aiterp.net/stufflog3/stufflog3/ports/mysql/mysqlcore"
	"git.aiterp.net/stufflog3/stufflog3/usecases/items"
	"git.aiterp.net/stufflog3/stufflog3/usecases/projects"
	"git.aiterp.net/stufflog3/stufflog3/usecases/scopes"
	"git.aiterp.net/stufflog3/stufflog3/usecases/stats"
	"time"

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

type Database struct {
	db *sql.DB
	q  *mysqlcore.Queries
}

func (db *Database) Scopes() scopes.Repository {
	return &scopeRepository{
		db: db.db,
		q:  db.q,
	}
}

func (db *Database) Projects() projects.Repository {
	return &projectRepository{
		db: db.db,
		q:  db.q,
	}
}

func (db *Database) Stats() stats.Repository {
	return &statsRepository{
		db: db.db,
		q:  db.q,
	}
}

func (db *Database) Items() items.Repository {
	return &itemRepository{
		db: db.db,
		q:  db.q,
	}
}

func Connect(host string, port int, username, password, database string) (*Database, error) {
	db, err := sql.Open("mysql", fmt.Sprintf(
		"%s:%s@(%s:%d)/%s?parseTime=true", username, password, host, port, database,
	))
	if err != nil {
		return nil, err
	}

	db.SetMaxOpenConns(10)
	db.SetMaxIdleConns(10)
	db.SetConnMaxIdleTime(time.Minute)

	err = db.Ping()
	if err != nil {
		return nil, err
	}

	q, err := mysqlcore.Prepare(context.Background(), db)
	if err != nil {
		return nil, err
	}

	return &Database{db: db, q: q}, nil
}

func timePtr(nullTime sql.NullTime) *time.Time {
	if nullTime.Valid {
		return &nullTime.Time
	} else {
		return nil
	}
}

func intPtr(nullInt32 sql.NullInt32) *int {
	if nullInt32.Valid {
		v := int(nullInt32.Int32)
		return &v
	} else {
		return nil
	}
}