synapsecns/sanguine

package mysql

import (
    "context"
    "fmt"
    "time"

    "github.com/synapsecns/sanguine/contrib/screener-api/db"
    "github.com/synapsecns/sanguine/contrib/screener-api/db/sql/base"

    "github.com/ipfs/go-log"
    common_base "github.com/synapsecns/sanguine/core/dbcommon"
    "github.com/synapsecns/sanguine/core/metrics"
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
    "gorm.io/gorm/schema"
)

// Logger is the mysql logger.
var logger = log.Logger("screener-mysql")

// NewMysqlStore creates a new mysql store for a given data store.
func NewMysqlStore(ctx context.Context, dbURL string, handler metrics.Handler) (*Store, error) {
    logger.Debug("create mysql store")

    gdb, err := gorm.Open(mysql.Open(dbURL), &gorm.Config{
        Logger:               common_base.GetGormLogger(logger),
        FullSaveAssociations: true,
        NamingStrategy:       NamingStrategy,
        NowFunc:              time.Now,
    })

    if err != nil {
        return nil, fmt.Errorf("could not create mysql connection: %w", err)
    }

    sqlDB, err := gdb.DB()
    if err != nil {
        return nil, fmt.Errorf("could not get sql db: %w", err)
    }

    // fixes a timeout issue https://stackoverflow.com/a/42146536
    sqlDB.SetMaxIdleConns(MaxIdleConns)
    sqlDB.SetConnMaxLifetime(time.Hour)

    handler.AddGormCallbacks(gdb)

    err = gdb.WithContext(ctx).AutoMigrate(base.GetAllModels()...)
    if err != nil {
        return nil, fmt.Errorf("could not migrate on mysql: %w", err)
    }

    return &Store{base.NewStore(gdb, handler)}, nil
}

// Store is the mysql store. It extends the bsae store for mysql queries.
type Store struct {
    *base.Store
}

// MaxIdleConns is exported here for testing. Tests execute too slowly with a reconnect each time.
var MaxIdleConns = 10

// NamingStrategy is for table prefixes.
var NamingStrategy = schema.NamingStrategy{}

var _ db.BlacklistedAddressDB = &Store{}