mirror of
https://github.com/genuinetools/reg.git
synced 2024-09-12 06:11:03 -04:00
71 lines
2.2 KiB
Go
71 lines
2.2 KiB
Go
![]() |
package daemon
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
|
||
|
"github.com/docker/docker/container"
|
||
|
"github.com/sirupsen/logrus"
|
||
|
)
|
||
|
|
||
|
// ContainerRestart stops and starts a container. It attempts to
|
||
|
// gracefully stop the container within the given timeout, forcefully
|
||
|
// stopping it if the timeout is exceeded. If given a negative
|
||
|
// timeout, ContainerRestart will wait forever until a graceful
|
||
|
// stop. Returns an error if the container cannot be found, or if
|
||
|
// there is an underlying error at any stage of the restart.
|
||
|
func (daemon *Daemon) ContainerRestart(name string, seconds *int) error {
|
||
|
container, err := daemon.GetContainer(name)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
if seconds == nil {
|
||
|
stopTimeout := container.StopTimeout()
|
||
|
seconds = &stopTimeout
|
||
|
}
|
||
|
if err := daemon.containerRestart(container, *seconds); err != nil {
|
||
|
return fmt.Errorf("Cannot restart container %s: %v", name, err)
|
||
|
}
|
||
|
return nil
|
||
|
|
||
|
}
|
||
|
|
||
|
// containerRestart attempts to gracefully stop and then start the
|
||
|
// container. When stopping, wait for the given duration in seconds to
|
||
|
// gracefully stop, before forcefully terminating the container. If
|
||
|
// given a negative duration, wait forever for a graceful stop.
|
||
|
func (daemon *Daemon) containerRestart(container *container.Container, seconds int) error {
|
||
|
// Avoid unnecessarily unmounting and then directly mounting
|
||
|
// the container when the container stops and then starts
|
||
|
// again
|
||
|
if err := daemon.Mount(container); err == nil {
|
||
|
defer daemon.Unmount(container)
|
||
|
}
|
||
|
|
||
|
if container.IsRunning() {
|
||
|
// set AutoRemove flag to false before stop so the container won't be
|
||
|
// removed during restart process
|
||
|
autoRemove := container.HostConfig.AutoRemove
|
||
|
|
||
|
container.HostConfig.AutoRemove = false
|
||
|
err := daemon.containerStop(container, seconds)
|
||
|
// restore AutoRemove irrespective of whether the stop worked or not
|
||
|
container.HostConfig.AutoRemove = autoRemove
|
||
|
// containerStop will write HostConfig to disk, we shall restore AutoRemove
|
||
|
// in disk too
|
||
|
if toDiskErr := daemon.checkpointAndSave(container); toDiskErr != nil {
|
||
|
logrus.Errorf("Write container to disk error: %v", toDiskErr)
|
||
|
}
|
||
|
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if err := daemon.containerStart(container, "", "", true); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
daemon.LogContainerEvent(container, "restart")
|
||
|
return nil
|
||
|
}
|