mirror of https://github.com/statping/statping
35 lines
619 B
Go
35 lines
619 B
Go
package configs
|
|
|
|
import (
|
|
"gopkg.in/yaml.v2"
|
|
"io/ioutil"
|
|
"os"
|
|
)
|
|
|
|
// Save will initially create the config.yml file
|
|
func (d *DbConfig) Save(directory string) error {
|
|
data, err := yaml.Marshal(d)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err := ioutil.WriteFile(directory+"/config.yml", data, os.ModePerm); err != nil {
|
|
return err
|
|
}
|
|
d.filename = directory + "/config.yml"
|
|
return nil
|
|
}
|
|
|
|
// defaultPort accepts a database type and returns its default port
|
|
func defaultPort(db string) int {
|
|
switch db {
|
|
case "mysql":
|
|
return 3306
|
|
case "postgres":
|
|
return 5432
|
|
case "mssql":
|
|
return 1433
|
|
default:
|
|
return 0
|
|
}
|
|
}
|