47 lines
1.2 KiB
Go
Executable File
47 lines
1.2 KiB
Go
Executable File
package application
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"github.com/codegangsta/martini"
|
|
"github.com/tiburon-777/OTUS_HighLoad/internal/models"
|
|
"github.com/tiburon-777/modules/core/config"
|
|
)
|
|
|
|
type App struct {
|
|
Config *models.Configuration
|
|
DB *sql.DB
|
|
martini.ClassicMartini
|
|
}
|
|
|
|
func New(configFile, envPrefix string) (App, error) {
|
|
conf, err := configure(configFile, envPrefix)
|
|
if err != nil{
|
|
return App{}, fmt.Errorf("can't apply config: %w\n",err)
|
|
}
|
|
|
|
db, err := sql.Open("mysql", conf.DSN.User+":"+conf.DSN.Pass+"@tcp("+conf.DSN.Host+":"+conf.DSN.Port+")/"+conf.DSN.Base)
|
|
if err != nil {
|
|
panic(err.Error())
|
|
}
|
|
|
|
return App{Config: conf, DB: db}, nil
|
|
}
|
|
|
|
func configure(fileName string, envPrefix string) (*models.Configuration,error) {
|
|
var conf models.Configuration
|
|
s := config.New(&conf)
|
|
if fileName != "" {
|
|
fmt.Printf("try to apply config from file %s...\n", fileName)
|
|
if err := s.SetFromFile(fileName); err != nil {
|
|
return &models.Configuration{}, fmt.Errorf("can't apply config from file: %w", err)
|
|
}
|
|
}
|
|
if envPrefix != "" {
|
|
fmt.Printf("try to apply config from environment...\n")
|
|
if err := s.SetFromEnv(envPrefix); err != nil {
|
|
return &models.Configuration{}, fmt.Errorf("can't apply envvars to config:%w", err)
|
|
}
|
|
}
|
|
return &conf, nil
|
|
} |