Merge pull request #6 from RoundingWellOS/fix/updateServiceLBs

Actually update service load balancers

Author: BenCoffeed

lib/index.js

@@ -89,7 +89,19 @@ class ServerlessFargateTasks {
             'awslogs-stream-prefix': 'fargate'
           },
         },
-      }, container_override)
+      }, container_override);
+  
+      if(options.tasks[identifier].hasOwnProperty('loadBalancers')) {
+        let portMappings= [];
+        options.tasks[identifier]['loadBalancers'].forEach(item => {
+          const portMapping = {
+            'ContainerPort': item.containerPort,
+            'HostPort': item.hostPort
+          };
+          portMappings.push(portMapping);
+        });
+        definitions['PortMappings'] = portMappings;
+      }
 
       // create the task definition
       var task = {
@@ -165,11 +177,12 @@ class ServerlessFargateTasks {
           options.tasks[identifier]['loadBalancers'].forEach(item => {
             const loadBalancer = Object.assign({
               'ContainerName': name,
-              'ContainerPort': item.port,
+              'ContainerPort': item.containerPort,
               'TargetGroupArn': item.arn
             });
             loadBalancers.push(loadBalancer);
           });
+          service['Properties']['LoadBalancers'] = loadBalancers;
         }
         template['Resources'][normalizedIdentifier + 'Service'] = service
       }