74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
package amazon
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sort"
|
|
"time"
|
|
|
|
"github.com/aws/aws-sdk-go/aws"
|
|
"github.com/aws/aws-sdk-go/service/cloudformation"
|
|
"github.com/docker/ecs-plugin/pkg/console"
|
|
)
|
|
|
|
func (c *client) WaitStackCompletion(ctx context.Context, name string, operation int) error {
|
|
w := console.NewProgressWriter()
|
|
knownEvents := map[string]struct{}{}
|
|
|
|
// Get the unique Stack ID so we can collect events without getting some from previous deployments with same name
|
|
stackID, err := c.api.GetStackID(ctx, name)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
ticker := time.NewTicker(1 * time.Second)
|
|
done := make(chan error)
|
|
|
|
go func() {
|
|
err := c.api.WaitStackComplete(ctx, name, operation)
|
|
ticker.Stop()
|
|
done <- err
|
|
}()
|
|
|
|
var completed bool
|
|
var waitErr error
|
|
for !completed {
|
|
select {
|
|
case err := <-done:
|
|
completed = true
|
|
waitErr = err
|
|
case <-ticker.C:
|
|
}
|
|
events, err := c.api.DescribeStackEvents(ctx, stackID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
sort.Slice(events, func(i, j int) bool {
|
|
return events[i].Timestamp.Before(*events[j].Timestamp)
|
|
})
|
|
|
|
for _, event := range events {
|
|
if _, ok := knownEvents[*event.EventId]; ok {
|
|
continue
|
|
}
|
|
knownEvents[*event.EventId] = struct{}{}
|
|
|
|
resource := fmt.Sprintf("%s %q", aws.StringValue(event.ResourceType), aws.StringValue(event.LogicalResourceId))
|
|
w.ResourceEvent(resource, aws.StringValue(event.ResourceStatus), aws.StringValue(event.ResourceStatusReason))
|
|
}
|
|
}
|
|
return waitErr
|
|
}
|
|
|
|
type waitAPI interface {
|
|
GetStackID(ctx context.Context, name string) (string, error)
|
|
WaitStackComplete(ctx context.Context, name string, operation int) error
|
|
DescribeStackEvents(ctx context.Context, stackID string) ([]*cloudformation.StackEvent, error)
|
|
}
|
|
|
|
const (
|
|
StackCreate = iota
|
|
StackDelete
|
|
)
|