#387 fix: switch to image id map and add additional tests

This commit is contained in:
Simon Aronsson 2020-01-11 00:28:27 +01:00
parent 4e000fa89c
commit 7b8b8e8ad9
9 changed files with 244 additions and 102 deletions

View file

@ -73,17 +73,18 @@ func stopStaleContainer(container container.Container, client container.Client,
}
func restartContainersInSortedOrder(containers []container.Container, client container.Client, params UpdateParams) {
toDelete := make(map[container.Container]bool)
imageIDs := make(map[string]bool)
for _, container := range containers {
if !container.Stale {
continue
}
restartStaleContainer(container, client, params)
toDelete[container] = true
imageIDs[container.ImageID()] = true
}
if params.Cleanup {
for cont := range toDelete {
if err := client.RemoveImage(cont); err != nil {
for imageID := range imageIDs {
if err := client.RemoveImageByID(imageID); err != nil {
log.Error(err)
}
}