From d8997550b5694867640faaf19602c357e6c2b020 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Lauri=20V=C3=B5sandi?= Date: Sat, 8 May 2021 20:02:37 +0000 Subject: [PATCH] Use database and collection names from env vars --- main.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/main.go b/main.go index c925a68..1d10b68 100644 --- a/main.go +++ b/main.go @@ -20,6 +20,8 @@ type InventoryItem struct { } var mongoUri string = os.Getenv("MONGO_URI") +var databaseName string = os.Getenv("GOREDNS_DATABASE") +var collectionName string = os.Getenv("GOREDNS_COLLECTION") func appendResults(etype string, name string, m *dns.Msg, cur *mongo.Cursor) int { count := 0 @@ -97,7 +99,7 @@ func main() { log.Fatal(err) } - coll := client.Database("kspace_accounting").Collection("inventory") + coll := client.Database(databaseName).Collection(collectionName) defer client.Disconnect(ctx) dns.HandleFunc(".", wrapper(coll)) port := 53