synapsecns/sanguine

package mysql

import (
    "context"
    "fmt"
    "github.com/synapsecns/sanguine/core/metrics"
    scribeLogger "github.com/synapsecns/sanguine/services/scribe/logger"
    gormLogger "gorm.io/gorm/logger"

    "time"

    "github.com/synapsecns/sanguine/services/scribe/db/datastore/sql/base"
    "gorm.io/driver/mysql"
    "gorm.io/gorm"
    "gorm.io/gorm/schema"
)

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

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

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

// NamingStrategy is exported here for testing.
var NamingStrategy = schema.NamingStrategy{
    TablePrefix: "v3_",
}

// NewMysqlStore creates a new mysql store for a given data store.
func NewMysqlStore(parentCtx context.Context, dbURL string, handler metrics.Handler, skipMigrations bool) (_ *Store, err error) {
    logger.Debug("creating mysql store")
    scribeLogger.ReportScribeState(0, 0, nil, scribeLogger.CreatingSQLStore)
    ctx, span := handler.Tracer().Start(parentCtx, "start-mysql")
    defer func() {
        metrics.EndSpanWithErr(span, err)
    }()

    gdb, err := gorm.Open(mysql.Open(dbURL), &gorm.Config{
        Logger:                 gormLogger.Default.LogMode(gormLogger.Silent),
        FullSaveAssociations:   true,
        NamingStrategy:         NamingStrategy,
        NowFunc:                time.Now,
        SkipDefaultTransaction: true,
    })

    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(30 * time.Minute)
    sqlDB.SetMaxOpenConns(MaxOpenConns)

    handler.AddGormCallbacks(gdb)

    if !skipMigrations {
        // migrate in a transaction since we skip this by default
        err = gdb.Transaction(func(tx *gorm.DB) error {
            //nolint: wrapcheck
            return 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
}

// var _ db.Service = &Store{}