mirror of
https://github.com/simon987/ws_bucket.git
synced 2025-04-04 08:12:59 +00:00
37 lines
582 B
Go
37 lines
582 B
Go
package main
|
|
|
|
import (
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/simon987/ws_bucket/api"
|
|
"os"
|
|
)
|
|
|
|
func main() {
|
|
|
|
db, err := gorm.Open(getDialect(), getConnStr())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
a := api.New(db)
|
|
a.Run()
|
|
}
|
|
|
|
func getConnStr() string {
|
|
connStr := os.Getenv("WS_BUCKET_CONNSTR")
|
|
if connStr == "" {
|
|
return "host=localhost user=ws_bucket dbname=ws_bucket password=ws_bucket sslmode=disable"
|
|
} else {
|
|
return connStr
|
|
}
|
|
}
|
|
|
|
func getDialect() string {
|
|
connStr := os.Getenv("WS_BUCKET_DIALECT")
|
|
if connStr == "" {
|
|
return "postgres"
|
|
} else {
|
|
return connStr
|
|
}
|
|
}
|