INNER CODE UNIT · Go
mustConnectDB
sklinkert/at · cmd/at-ig/main.go:63
func mustConnectDB() *gorm.DB {
var err error
var db *gorm.DB
if conf.dbHost == "sqlite" {
db, err = gorm.Open(sqlite.Open("at-demo.db"), &gorm.Config{})
} else {
dsn := fmt.Sprintf("host=%s user=%s password=%s dbname=%s port=%d sslmode=require",
conf.dbHost, conf.dbUser, conf.dbPassword, conf.dbName, conf.dbPort)
db, err = gorm.Open(postgres.Open(dsn), &gorm.Config{})
}
if err != nil {
fatal("failed to connect database", err)
}
if err := db.AutoMigrate(&tick.Tick{}, &ohlc.OHLC{}, &trader.PerformanceRecord{}); err != nil {
fatal("db.AutoMigrate() failed", err)
}
return db
}