Update worker to use environment variables

mbruntink · Aug 16, 2020 · 143275a · 143275a
1 parent 245c8f3
commit 143275a
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/worker/src/Worker/Program.cs b/worker/src/Worker/Program.cs
@@ -16,8 +16,16 @@ public static int Main(string[] args)
         {
             try
             {
-                var pgsql = OpenDbConnection("Server=db;Username=postgres;Password=postgres;");
-                var redisConn = OpenRedisConnection("redis");
+                var sdEndpoint =  Environment.GetEnvironmentVariable("COPILOT_SERVICE_DISCOVERY_ENDPOINT");
+
+                var pgUser = Environment.GetEnvironmentVariable("POSTGRES_USER");
+                var pgPassword = Environment.GetEnvironmentVariable("POSTGRES_PASSWORD");
+
+                var pgConn = "Server=db." + sdEndpoint + ";Username=" + pgUser + ";Password=" + pgPassword + ";";
+                var redisHost = "redis." + sdEndpoint; 
+
+                var pgsql = OpenDbConnection(pgConn);
+                var redisConn = OpenRedisConnection(redisHost);
                 var redis = redisConn.GetDatabase();
 
                 // Keep alive is not implemented in Npgsql yet. This workaround was recommended:
@@ -34,7 +42,7 @@ public static int Main(string[] args)
                     // Reconnect redis if down
                     if (redisConn == null || !redisConn.IsConnected) {
                         Console.WriteLine("Reconnecting Redis");
-                        redisConn = OpenRedisConnection("redis");
+                        redisConn = OpenRedisConnection(redisHost);
                         redis = redisConn.GetDatabase();
                     }
                     string json = redis.ListLeftPopAsync("votes").Result;
@@ -46,7 +54,7 @@ public static int Main(string[] args)
                         if (!pgsql.State.Equals(System.Data.ConnectionState.Open))
                         {
                             Console.WriteLine("Reconnecting DB");
-                            pgsql = OpenDbConnection("Server=db;Username=postgres;Password=postgres;");
+                            pgsql = OpenDbConnection(pgConn);
                         }
                         else
                         { // Normal +1 vote requested