60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
package backend
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/compose-spec/compose-go/cli"
|
|
"github.com/docker/ecs-plugin/pkg/compose"
|
|
)
|
|
|
|
func (b *Backend) Ps(ctx context.Context, options *cli.ProjectOptions) ([]compose.ServiceStatus, error) {
|
|
projectName, err := b.projectName(options)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
parameters, err := b.api.ListStackParameters(ctx, projectName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
cluster := parameters[ParameterClusterName]
|
|
|
|
resources, err := b.api.ListStackResources(ctx, projectName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
servicesARN := []string{}
|
|
for _, r := range resources {
|
|
switch r.Type {
|
|
case "AWS::ECS::Service":
|
|
servicesARN = append(servicesARN, r.ARN)
|
|
case "AWS::ECS::Cluster":
|
|
cluster = r.ARN
|
|
}
|
|
}
|
|
if len(servicesARN) == 0 {
|
|
return nil, nil
|
|
}
|
|
status, err := b.api.DescribeServices(ctx, cluster, servicesARN)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for i, state := range status {
|
|
ports := []string{}
|
|
for _, lb := range state.LoadBalancers {
|
|
ports = append(ports, fmt.Sprintf(
|
|
"%s:%d->%d/%s",
|
|
lb.URL,
|
|
lb.PublishedPort,
|
|
lb.TargetPort,
|
|
strings.ToLower(lb.Protocol)))
|
|
}
|
|
state.Ports = ports
|
|
status[i] = state
|
|
}
|
|
return status, nil
|
|
}
|