fix(main): use port dynamically

adhocore · Mar 10, 2021 · 04e7272 · 04e7272
1 parent baa12e0
commit 04e7272
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/main.go b/main.go
@@ -9,12 +9,20 @@ import (
     "github.com/adhocore/urlsh/router"
 )
 
-func main() {
-    port := os.Getenv("APP_PORT")
-    if port == "" {
-        port = "1000"
+func getPort() string {
+    if port := os.Getenv("PORT"); port != "" {
+        return port
+    }
+    if port := os.Getenv("APP_PORT"); port != "" {
+        return port
     }
 
+    return "1000"
+}
+
+func main() {
+    port := getPort()
+
     server := &http.Server{
         Addr:         ":" + port,
         Handler:      router.RegisterHandlers(),