tyler.durden/compose
1
0
Fork 0
mirror of https://github.com/docker/compose.git synced 2025-05-03 06:00:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

add all service security groups to LB

Browse Source 
Signed-off-by: aiordache <anca.iordache@docker.com>
Signed-off-by: Nicolas De Loof <nicolas.deloof@gmail.com>
This commit is contained in:
aiordache 2020-06-03 17:40:39 +02:00 committed by Nicolas De Loof
parent 92173eaf35
commit e7f77ca3ef
No known key found for this signature in database
GPG Key ID: 9858809D6F8F6E7E
1 changed files with 36 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
ecs/pkg/amazon/cloudformation.go
View File

@ -97,6 +97,8 @@ func (c client) Convert(project *compose.Project) (*cloudformation.Template, err
Name: fmt.Sprintf("%s.local", project.Name), Name: fmt.Sprintf("%s.local", project.Name),
Vpc: cloudformation.Ref(ParameterVPCId), Vpc: cloudformation.Ref(ParameterVPCId),
} }
//map LB type to security groups list
loadBalancers := map[string][]string{}
for _, service := range project.Services { for _, service := range project.Services {
definition, err := Convert(project, service) definition, err := Convert(project, service)
@ -168,7 +170,7 @@ func (c client) Convert(project *compose.Project) (*cloudformation.Template, err
} }
dependsOn := []string{} dependsOn := []string{}
loadBalancers := []ecs.Service_LoadBalancer{} serviceLB := []ecs.Service_LoadBalancer{}
if len(service.Ports) > 0 { if len(service.Ports) > 0 {
for _, port := range service.Ports { for _, port := range service.Ports {
loadBalancerType := "network" loadBalancerType := "network"
@ -191,26 +193,12 @@ func (c client) Convert(project *compose.Project) (*cloudformation.Template, err
strings.Title(project.Name), strings.Title(project.Name),
strings.ToUpper(loadBalancerType[0:1]), strings.ToUpper(loadBalancerType[0:1]),
) )
// create load baalncer if it doesn't exist // create load balancer if it doesn't exist
if _, ok := template.Resources[loadBalancerName]; !ok { if _, ok := loadBalancers[loadBalancerType]; !ok {
loadBalancers[loadBalancerType] = []string{}
}
loadBalancers[loadBalancerType] = append(loadBalancers[loadBalancerType], loadBalancerSecGroups...)
template.Resources[loadBalancerName] = &elasticloadbalancingv2.LoadBalancer{
Name: loadBalancerName,
Scheme: "internet-facing",
SecurityGroups: loadBalancerSecGroups,
Subnets: []string{
cloudformation.Ref(ParameterSubnet1Id),
cloudformation.Ref(ParameterSubnet2Id),
},
Tags: []tags.Tag{
{
Key: ProjectTag,
Value: project.Name,
},
},
Type: loadBalancerType,
}
}
targetGroupName := fmt.Sprintf( targetGroupName := fmt.Sprintf(
"%s%s%sTargetGroup", "%s%s%sTargetGroup",
normalizeResourceName(service.Name), normalizeResourceName(service.Name),
@ -257,7 +245,7 @@ func (c client) Convert(project *compose.Project) (*cloudformation.Template, err
Port: int(port.Published), Port: int(port.Published),
} }
loadBalancers = append(loadBalancers, ecs.Service_LoadBalancer{ serviceLB = append(serviceLB, ecs.Service_LoadBalancer{
ContainerName: service.Name, ContainerName: service.Name,
ContainerPort: int(port.Published), ContainerPort: int(port.Published),
TargetGroupArn: cloudformation.Ref(targetGroupName), TargetGroupArn: cloudformation.Ref(targetGroupName),
@ -278,7 +266,7 @@ func (c client) Convert(project *compose.Project) (*cloudformation.Template, err
Cluster: cluster, Cluster: cluster,
DesiredCount: desiredCount, DesiredCount: desiredCount,
LaunchType: ecsapi.LaunchTypeFargate, LaunchType: ecsapi.LaunchTypeFargate,
LoadBalancers: loadBalancers, LoadBalancers: serviceLB,
NetworkConfiguration: &ecs.Service_NetworkConfiguration{ NetworkConfiguration: &ecs.Service_NetworkConfiguration{
AwsvpcConfiguration: &ecs.Service_AwsVpcConfiguration{ AwsvpcConfiguration: &ecs.Service_AwsVpcConfiguration{
AssignPublicIp: ecsapi.AssignPublicIpEnabled, AssignPublicIp: ecsapi.AssignPublicIpEnabled,
@ -305,6 +293,32 @@ func (c client) Convert(project *compose.Project) (*cloudformation.Template, err
TaskDefinition: cloudformation.Ref(normalizeResourceName(taskDefinition)), TaskDefinition: cloudformation.Ref(normalizeResourceName(taskDefinition)),
} }
} }
// create LBs
for lbType, lbSecGroups := range loadBalancers {
loadBalancerName := fmt.Sprintf(
"%s%sLB",
strings.Title(project.Name),
strings.ToUpper(lbType[0:1]),
)
template.Resources[loadBalancerName] = &elasticloadbalancingv2.LoadBalancer{
Name: loadBalancerName,
Scheme: "internet-facing",
SecurityGroups: lbSecGroups,
Subnets: []string{
cloudformation.Ref(ParameterSubnet1Id),
cloudformation.Ref(ParameterSubnet2Id),
},
Tags: []tags.Tag{
{
Key: ProjectTag,
Value: project.Name,
},
},
Type: lbType,
}
}
return template, nil return template, nil
} }