81 lines
1.8 KiB
Go
81 lines
1.8 KiB
Go
|
package mysqlx
|
||
|
|
|
||
|
|
import (
|
||
|
|
drivermysql "gorm.io/driver/mysql"
|
||
|
|
"gorm.io/gorm"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Client is the wrapper around a gorm DB instance.
|
||
|
|
type Client struct {
|
||
|
|
db *gorm.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
// New creates a Client using functional options.
|
||
|
|
// It requires either a DSN or a custom gorm Dialector.
|
||
|
|
func NewDB(opts ...Option) (*Client, error) {
|
||
|
|
cfg := defaultConfig()
|
||
|
|
for _, opt := range opts {
|
||
|
|
opt(cfg)
|
||
|
|
}
|
||
|
|
|
||
|
|
if cfg.Dialector == nil {
|
||
|
|
dsn, err := buildDSN(cfg)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
cfg.Dialector = drivermysql.Open(dsn)
|
||
|
|
}
|
||
|
|
|
||
|
|
gormConfig := &gorm.Config{
|
||
|
|
Logger: cfg.Logger,
|
||
|
|
NamingStrategy: cfg.NamingStrategy,
|
||
|
|
DisableForeignKeyConstraintWhenMigrating: cfg.DisableForeignKeyConstraintWhenMigrating,
|
||
|
|
SkipDefaultTransaction: cfg.SkipDefaultTransaction,
|
||
|
|
DryRun: cfg.DryRun,
|
||
|
|
}
|
||
|
|
|
||
|
|
db, err := gorm.Open(cfg.Dialector, gormConfig)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
sqlDB, err := db.DB()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
if cfg.MaxIdleConns > 0 {
|
||
|
|
sqlDB.SetMaxIdleConns(cfg.MaxIdleConns)
|
||
|
|
}
|
||
|
|
if cfg.MaxOpenConns > 0 {
|
||
|
|
sqlDB.SetMaxOpenConns(cfg.MaxOpenConns)
|
||
|
|
}
|
||
|
|
if cfg.ConnMaxIdleTime > 0 {
|
||
|
|
sqlDB.SetConnMaxIdleTime(cfg.ConnMaxIdleTime)
|
||
|
|
}
|
||
|
|
if cfg.ConnMaxLifetime > 0 {
|
||
|
|
sqlDB.SetConnMaxLifetime(cfg.ConnMaxLifetime)
|
||
|
|
}
|
||
|
|
|
||
|
|
return &Client{db: db}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// DB returns the underlying gorm DB.
|
||
|
|
func (c *Client) DB() *gorm.DB {
|
||
|
|
return c.db
|
||
|
|
}
|
||
|
|
|
||
|
|
// Close closes the underlying sql.DB connection pool.
|
||
|
|
func (c *Client) Close() error {
|
||
|
|
sqlDB, err := c.db.DB()
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return sqlDB.Close()
|
||
|
|
}
|
||
|
|
|
||
|
|
// AutoMigrate runs gorm AutoMigrate for the provided models.
|
||
|
|
func (c *Client) AutoMigrate(models ...interface{}) error {
|
||
|
|
return c.db.AutoMigrate(models...)
|
||
|
|
}
|