fieldproject/fieldproject_com/db/db.go

package db

import (
	"context"
	"database/sql"
	"time"

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

var (
	_db *sql.DB
)

type dbCfg struct {
	DriveName         string
	Dsn               string
	MaxIdleConnection int
	MaxOpenConnection int
	MaxQueryTime      time.Duration
}

// GetConn from db conn pool
func GetConn(ctx context.Context) (*sql.Conn, error) {
	return _db.Conn(ctx)
}

// Option is function option
type Option func(*dbCfg)

// Drivename  Set mysql or sqlite
func Drivename(drivename string) Option {
	return func(dbcfg *dbCfg) {
		dbcfg.DriveName = drivename
	}
}

// Dsn set dbCfg conn addr
func Dsn(dsn string) func(*dbCfg) {
	return func(dbcfg *dbCfg) {
		dbcfg.Dsn = dsn
	}
}

// MaxIdleConnection set sql db max idle conn
func MaxIdleConnection(idle int) Option {
	return func(dbcfg *dbCfg) {
		dbcfg.MaxIdleConnection = idle
	}
}

// MaxOpenConnection  set sql db  max open conn
func MaxOpenConnection(open int) Option {
	return func(dbcfg *dbCfg) {
		dbcfg.MaxOpenConnection = open
	}
}

// MaxQueryTime set sql conn exec max timeout
func MaxQueryTime(query time.Duration) Option {
	return func(dbcfg *dbCfg) {
		dbcfg.MaxQueryTime = query
	}
}

// NewDb create new db
func NewDb(opts ...Option) (err error) {

	dbcfg := &dbCfg{}
	for _, opt := range opts {
		opt(dbcfg)
	}
	_db, err = sql.Open(dbcfg.DriveName, dbcfg.Dsn)
	if err != nil {
		return err
	}
	_db.SetMaxOpenConns(dbcfg.MaxOpenConnection)
	_db.SetMaxIdleConns(dbcfg.MaxIdleConnection)
	err = _db.Ping()
	return err
}