43 lines
893 B
Go
43 lines
893 B
Go
package task
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync/atomic"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type Data struct {
|
|
gormDB *reloadableDB
|
|
databaseReady atomic.Bool
|
|
}
|
|
|
|
func (d *Data) DB() *gorm.DB {
|
|
if d == nil || d.gormDB == nil {
|
|
return nil
|
|
}
|
|
return d.gormDB.DB()
|
|
}
|
|
|
|
func (d *Data) DatabaseReady() bool { return d != nil && d.databaseReady.Load() }
|
|
|
|
type reloadableDB struct{ db *gorm.DB }
|
|
|
|
func newReloadableDB(db *gorm.DB) *reloadableDB { return &reloadableDB{db: db} }
|
|
func (r *reloadableDB) DB() *gorm.DB {
|
|
if r == nil {
|
|
return nil
|
|
}
|
|
return r.db
|
|
}
|
|
func (r *reloadableDB) WithContext(ctx context.Context) *gorm.DB { return r.db.WithContext(ctx) }
|
|
|
|
func openWithDriver(driver, dsn string) (*gorm.DB, error) {
|
|
if driver != "sqlite" {
|
|
return nil, fmt.Errorf("unsupported test database driver %q", driver)
|
|
}
|
|
return gorm.Open(sqlite.Open(dsn), &gorm.Config{})
|
|
}
|