forked from mirrors/homebox
28 lines
567 B
Go
28 lines
567 B
Go
|
package config
|
||
|
|
|
||
|
|
const (
|
||
|
|
DriverSqlite3 = "sqlite3"
|
||
|
|
DriverPostgres = "postgres"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Database struct {
|
||
|
|
Driver string `yaml:"driver" conf:"default:sqlite3"`
|
||
|
|
SqliteUrl string `yaml:"sqlite-url" conf:"default:file:ent?mode=memory&cache=shared&_fk=1"`
|
||
|
|
PostgresUrl string `yaml:"postgres-url" conf:""`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *Database) GetDriver() string {
|
||
|
|
return d.Driver
|
||
|
|
}
|
||
|
|
|
||
|
|
func (d *Database) GetUrl() string {
|
||
|
|
switch d.Driver {
|
||
|
|
case DriverSqlite3:
|
||
|
|
return d.SqliteUrl
|
||
|
|
case DriverPostgres:
|
||
|
|
return d.PostgresUrl
|
||
|
|
default:
|
||
|
|
panic("unknown database driver")
|
||
|
|
}
|
||
|
|
}
|