30 lines
850 B
Go
30 lines
850 B
Go
|
package config
|
||
|
|
|
||
|
|
import (
|
||
|
|
"code.icb4dc0.de/prskr/searcherside/core/ports"
|
||
|
|
"code.icb4dc0.de/prskr/searcherside/infrastructure/db/migrate"
|
||
|
|
"code.icb4dc0.de/prskr/searcherside/internal/ent"
|
||
|
|
"code.icb4dc0.de/prskr/searcherside/scripts/migrations"
|
||
|
|
)
|
||
|
|
|
||
|
|
type DB struct {
|
||
|
|
Driver ports.Driver `env:"DB_DRIVER" name:"driver" default:"sqlite" help:"DB driver, either postgres or sqlite"`
|
||
|
|
URL string `env:"DB_URL" name:"url" default:"sqlite://file::memory:?cache=shared" help:"Connection URL"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d DB) Client() (*ent.Client, error) {
|
||
|
|
return ent.Open(d.Driver.String(), d.URL)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d DB) Migrator() (ports.Migrator, *ent.Client, error) {
|
||
|
|
client, err := d.Client()
|
||
|
|
if err != nil {
|
||
|
|
return nil, nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
return migrate.AtlasMigrator{
|
||
|
|
MigrationsFS: migrations.Fs,
|
||
|
|
RevisionRW: migrate.NewDBRevisionRW(client),
|
||
|
|
}, client, nil
|
||
|
|
}
|