Work with env variables

cmd/dor-insert-mongo/dor-insert-mongo.go

@@ -1,7 +1,7 @@
 package main
 
 import (
-	"flag"
+	"os"
 	"time"
 
 	"github.com/ilyaglow/dor"
@@ -10,10 +10,9 @@ import (
 const updatePeriod = time.Hour * 24
 
 func main() {
-	mongoURL := flag.String("mongo", "", "MongoDB URL")
-	flag.Parse()
+	mongoURL := os.Getenv("DOR_MONGO_URL")
 
-	d, err := dor.New("mongodb", *mongoURL, false)
+	d, err := dor.New("mongodb", mongoURL, false)
 	if err != nil {
 		panic(err)
 	}

service/dor-web-mongodb/dor-web-mongodb.go

@@ -1,24 +1,26 @@
 package main
 
 import (
-	"flag"
 	"fmt"
+	"os"
 
 	"github.com/ilyaglow/dor"
 	"github.com/ilyaglow/dor/web"
 )
 
 func main() {
-	bindAddr := flag.String("host", "127.0.0.1", "IP-address to bind")
-	bindPort := flag.String("port", "8080", "Port to bind")
-	mongoURL := flag.String("mongo", "", "MongoDB URL")
-	flag.Parse()
-	hp := fmt.Sprintf("%s:%s", *bindAddr, *bindPort)
+	var port string
+	if port = os.Getenv("DOR_PORT"); port == "" {
+		port = "8080"
+	}
+	port = fmt.Sprintf(":%s", port)
+
+	mongoURL := os.Getenv("DOR_MONGO_URL")
 
-	d, err := dor.New("mongodb", *mongoURL, false)
+	d, err := dor.New("mongodb", mongoURL, false)
 	if err != nil {
 		panic(err)
 	}
 
-	dorweb.Serve(hp, d)
+	dorweb.Serve(port, d)
 }