Kiu/Semaphore
Kiu/Semaphore
1
0
Fork 0
mirror of https://github.com/semaphoreui/semaphore.git synced 2025-01-20 15:29:28 +01:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

refactor(be): add field dialect to config file

Browse Source
This commit is contained in:
Denis Gukov 2021-08-28 17:04:56 +05:00
parent 22e422e15a
commit 31a71abf4a
1 changed files with 41 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
util/config.go
View File

@ -20,16 +20,17 @@ var Cookie *securecookie.SecureCookie
// WebHostURL is the public route to the semaphore server
var WebHostURL *url.URL
type DbDriver int
type DbDriver string
const (
DbDriverMySQL DbDriver = iota
DbDriverBolt
DbDriverPostgres
DbDriverMySQL DbDriver = "mysql"
DbDriverBolt DbDriver = "bolt"
DbDriverPostgres DbDriver = "postgres"
)
type DbConfig struct {
Dialect DbDriver `json:"-"`
Hostname string `json:"host"`
Username string `json:"user"`
Password string `json:"pass"`
@ -49,6 +50,8 @@ type ConfigType struct {
BoltDb DbConfig `json:"bolt"`
Postgres DbConfig `json:"pgsql"`
Dialect DbDriver `json:"dialect"`
// Format `:port_num` eg, :3000
// if : is missing it will be corrected
Port string `json:"port"`
@ -194,13 +197,8 @@ func decodeConfig(file io.Reader) {
}
// String returns dialect name for GORP.
// TODO: It should be moved to sql package.
func (d DbDriver) String() string {
return [...]string{
"mysql",
"", // didn't support by BoltDB
"postgres",
}[d]
return string(d)
}
func (d *DbConfig) IsPresent() bool {
@ -251,17 +249,42 @@ func (d *DbConfig) GetConnectionString(includeDbName bool) (connectionString str
return
}
func (conf *ConfigType) GetDialect() (dialect DbDriver, err error) {
if conf.Dialect == "" {
switch {
case conf.MySQL.IsPresent():
dialect = DbDriverMySQL
case conf.BoltDb.IsPresent():
dialect = DbDriverBolt
case conf.Postgres.IsPresent():
dialect = DbDriverPostgres
default:
err = errors.New("database configuration not found")
}
return
}
dialect = conf.Dialect
return
}
func (conf *ConfigType) GetDBConfig() (dbConfig DbConfig, err error) {
switch {
case conf.MySQL.IsPresent():
dbConfig = conf.MySQL
dbConfig.Dialect = DbDriverMySQL
case conf.BoltDb.IsPresent():
var dialect DbDriver
dialect, err = conf.GetDialect()
if err != nil {
return
}
dbConfig.Dialect = dialect
switch dialect {
case DbDriverBolt:
dbConfig = conf.BoltDb
dbConfig.Dialect = DbDriverBolt
case conf.Postgres.IsPresent():
case DbDriverPostgres:
dbConfig = conf.Postgres
dbConfig.Dialect = DbDriverPostgres
case DbDriverMySQL:
dbConfig = conf.MySQL
default:
err = errors.New("database configuration not found")
}