jimeh/ozu.io
1
0
Fork 0
mirror of https://github.com/jimeh/ozu.io.git synced 2026-02-19 08:06:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update main.go according to latest changes

Browse Source
This commit is contained in:
Jim Myhrberg
2016-07-17 16:18:01 +01:00
parent 5d3c8ef3ca
commit 351be0b59d
1 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
main.go
View File

@@ -7,9 +7,17 @@ import (
"github.com/jimeh/ozu.io/shortener" "github.com/jimeh/ozu.io/shortener"
"github.com/jimeh/ozu.io/storage/goleveldbstore" "github.com/jimeh/ozu.io/storage/goleveldbstore"
"github.com/jimeh/ozu.io/web" "github.com/jimeh/ozu.io/web"
"github.com/valyala/fasthttp"
) )
func getPort() string {
port := os.Getenv("PORT")
if port == "" {
port = "8080"
}
return port
}
func main() { func main() {
store, err := goleveldbstore.New("ozuio_database") store, err := goleveldbstore.New("ozuio_database")
if err != nil { if err != nil {
@@ -18,12 +26,7 @@ func main() {
defer store.Close() defer store.Close()
s := shortener.New(store) s := shortener.New(store)
router := web.NewRouter(s) server := web.NewServer(s)
port := os.Getenv("PORT") log.Fatal(server.ListenAndServe(":" + getPort()))
if port == "" {
port = "8080"
}
log.Fatal(fasthttp.ListenAndServe(":"+port, router.HandleRequest))
} }