You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
3.4 KiB
Go
107 lines
3.4 KiB
Go
package lifecycle
|
|
|
|
import (
|
|
"github.com/containrrr/watchtower/pkg/container"
|
|
"github.com/containrrr/watchtower/pkg/types"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// ExecutePreChecks tries to run the pre-check lifecycle hook for all containers included by the current filter.
|
|
func ExecutePreChecks(client container.Client, params types.UpdateParams) {
|
|
containers, err := client.ListContainers(params.Filter)
|
|
if err != nil {
|
|
return
|
|
}
|
|
for _, currentContainer := range containers {
|
|
ExecutePreCheckCommand(client, currentContainer)
|
|
}
|
|
}
|
|
|
|
// ExecutePostChecks tries to run the post-check lifecycle hook for all containers included by the current filter.
|
|
func ExecutePostChecks(client container.Client, params types.UpdateParams) {
|
|
containers, err := client.ListContainers(params.Filter)
|
|
if err != nil {
|
|
return
|
|
}
|
|
for _, currentContainer := range containers {
|
|
ExecutePostCheckCommand(client, currentContainer)
|
|
}
|
|
}
|
|
|
|
// ExecutePreCheckCommand tries to run the pre-check lifecycle hook for a single container.
|
|
func ExecutePreCheckCommand(client container.Client, container types.Container) {
|
|
clog := log.WithField("container", container.Name())
|
|
command := container.GetLifecyclePreCheckCommand()
|
|
if len(command) == 0 {
|
|
clog.Debug("No pre-check command supplied. Skipping")
|
|
return
|
|
}
|
|
|
|
clog.Debug("Executing pre-check command.")
|
|
_, err := client.ExecuteCommand(container.ID(), command, 1)
|
|
if err != nil {
|
|
clog.Error(err)
|
|
}
|
|
}
|
|
|
|
// ExecutePostCheckCommand tries to run the post-check lifecycle hook for a single container.
|
|
func ExecutePostCheckCommand(client container.Client, container types.Container) {
|
|
clog := log.WithField("container", container.Name())
|
|
command := container.GetLifecyclePostCheckCommand()
|
|
if len(command) == 0 {
|
|
clog.Debug("No post-check command supplied. Skipping")
|
|
return
|
|
}
|
|
|
|
clog.Debug("Executing post-check command.")
|
|
_, err := client.ExecuteCommand(container.ID(), command, 1)
|
|
if err != nil {
|
|
clog.Error(err)
|
|
}
|
|
}
|
|
|
|
// ExecutePreUpdateCommand tries to run the pre-update lifecycle hook for a single container.
|
|
func ExecutePreUpdateCommand(client container.Client, container types.Container) (SkipUpdate bool, err error) {
|
|
timeout := container.PreUpdateTimeout()
|
|
command := container.GetLifecyclePreUpdateCommand()
|
|
clog := log.WithField("container", container.Name())
|
|
|
|
if len(command) == 0 {
|
|
clog.Debug("No pre-update command supplied. Skipping")
|
|
return false, nil
|
|
}
|
|
|
|
if !container.IsRunning() || container.IsRestarting() {
|
|
clog.Debug("Container is not running. Skipping pre-update command.")
|
|
return false, nil
|
|
}
|
|
|
|
clog.Debug("Executing pre-update command.")
|
|
return client.ExecuteCommand(container.ID(), command, timeout)
|
|
}
|
|
|
|
// ExecutePostUpdateCommand tries to run the post-update lifecycle hook for a single container.
|
|
func ExecutePostUpdateCommand(client container.Client, newContainerID types.ContainerID) {
|
|
newContainer, err := client.GetContainer(newContainerID)
|
|
timeout := newContainer.PostUpdateTimeout()
|
|
|
|
if err != nil {
|
|
log.WithField("containerID", newContainerID.ShortID()).Error(err)
|
|
return
|
|
}
|
|
clog := log.WithField("container", newContainer.Name())
|
|
|
|
command := newContainer.GetLifecyclePostUpdateCommand()
|
|
if len(command) == 0 {
|
|
clog.Debug("No post-update command supplied. Skipping")
|
|
return
|
|
}
|
|
|
|
clog.Debug("Executing post-update command.")
|
|
_, err = client.ExecuteCommand(newContainerID, command, timeout)
|
|
|
|
if err != nil {
|
|
clog.Error(err)
|
|
}
|
|
}
|