diff --git a/README.md b/README.md new file mode 100644 index 0000000..a77115c --- /dev/null +++ b/README.md @@ -0,0 +1,15 @@ +### Environment variables + +| Name | Default | +|:---|:---| +| `WS_BUCKET_ADDR` | `0.0.0.0:3020` | +| `WS_BUCKET_WORKDIR` | `./data` | +| `WS_BUCKET_LOGLEVEL` | `trace` | +| `WS_BUCKET_CONNSTR` | `host=localhost user=ws_bucket dbname=ws_bucket password=ws_bucket sslmode=disable` | +| `WS_BUCKET_DIALECT` | `postgres` | + +### Running tests +```bash +cd test/ +go test +``` \ No newline at end of file diff --git a/main.go b/main.go index 7d5d74b..ca394ce 100644 --- a/main.go +++ b/main.go @@ -3,11 +3,12 @@ package main import ( "github.com/jinzhu/gorm" "github.com/simon987/ws_bucket/api" + "os" ) func main() { - db, err := gorm.Open("postgres", "host=localhost user=ws_bucket dbname=ws_bucket password=ws_bucket sslmode=disable") + db, err := gorm.Open(getDialect(), getConnStr()) if err != nil { panic(err) } @@ -15,3 +16,21 @@ func main() { 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 + } +}