mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
Perform graceful shutdown during plugin disable.
Signed-off-by: Anusha Ragunathan <anusha@docker.com>
This commit is contained in:
parent
65835bfaa6
commit
766cc9b467
1 changed files with 23 additions and 18 deletions
|
@ -9,6 +9,7 @@ import (
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/Sirupsen/logrus"
|
"github.com/Sirupsen/logrus"
|
||||||
|
"github.com/docker/docker/libcontainerd"
|
||||||
"github.com/docker/docker/oci"
|
"github.com/docker/docker/oci"
|
||||||
"github.com/docker/docker/pkg/plugins"
|
"github.com/docker/docker/pkg/plugins"
|
||||||
"github.com/docker/docker/plugin/v2"
|
"github.com/docker/docker/plugin/v2"
|
||||||
|
@ -48,6 +49,25 @@ func (pm *Manager) restore(p *v2.Plugin) error {
|
||||||
return pm.containerdClient.Restore(p.GetID(), attachToLog(p.GetID()))
|
return pm.containerdClient.Restore(p.GetID(), attachToLog(p.GetID()))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func shutdownPlugin(p *v2.Plugin, containerdClient libcontainerd.Client) {
|
||||||
|
pluginID := p.GetID()
|
||||||
|
|
||||||
|
err := containerdClient.Signal(pluginID, int(syscall.SIGTERM))
|
||||||
|
if err != nil {
|
||||||
|
logrus.Errorf("Sending SIGTERM to plugin failed with error: %v", err)
|
||||||
|
} else {
|
||||||
|
select {
|
||||||
|
case <-p.ExitChan:
|
||||||
|
logrus.Debug("Clean shutdown of plugin")
|
||||||
|
case <-time.After(time.Second * 10):
|
||||||
|
logrus.Debug("Force shutdown plugin")
|
||||||
|
if err := containerdClient.Signal(pluginID, int(syscall.SIGKILL)); err != nil {
|
||||||
|
logrus.Errorf("Sending SIGKILL to plugin failed with error: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func (pm *Manager) disable(p *v2.Plugin) error {
|
func (pm *Manager) disable(p *v2.Plugin) error {
|
||||||
if !p.IsEnabled() {
|
if !p.IsEnabled() {
|
||||||
return fmt.Errorf("plugin %s is already disabled", p.Name())
|
return fmt.Errorf("plugin %s is already disabled", p.Name())
|
||||||
|
@ -55,9 +75,8 @@ func (pm *Manager) disable(p *v2.Plugin) error {
|
||||||
p.Lock()
|
p.Lock()
|
||||||
p.Restart = false
|
p.Restart = false
|
||||||
p.Unlock()
|
p.Unlock()
|
||||||
if err := pm.containerdClient.Signal(p.GetID(), int(syscall.SIGKILL)); err != nil {
|
|
||||||
logrus.Error(err)
|
shutdownPlugin(p, pm.containerdClient)
|
||||||
}
|
|
||||||
pm.pluginStore.SetState(p, false)
|
pm.pluginStore.SetState(p, false)
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
@ -71,25 +90,11 @@ func (pm *Manager) Shutdown() {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if pm.containerdClient != nil && p.IsEnabled() {
|
if pm.containerdClient != nil && p.IsEnabled() {
|
||||||
pluginID := p.GetID()
|
|
||||||
p.Lock()
|
p.Lock()
|
||||||
p.ExitChan = make(chan bool)
|
p.ExitChan = make(chan bool)
|
||||||
p.Restart = false
|
p.Restart = false
|
||||||
p.Unlock()
|
p.Unlock()
|
||||||
err := pm.containerdClient.Signal(p.PluginObj.ID, int(syscall.SIGTERM))
|
shutdownPlugin(p, pm.containerdClient)
|
||||||
if err != nil {
|
|
||||||
logrus.Errorf("Sending SIGTERM to plugin failed with error: %v", err)
|
|
||||||
} else {
|
|
||||||
select {
|
|
||||||
case <-p.ExitChan:
|
|
||||||
logrus.Debug("Clean shutdown of plugin")
|
|
||||||
case <-time.After(time.Second * 10):
|
|
||||||
logrus.Debug("Force shutdown plugin")
|
|
||||||
if err := pm.containerdClient.Signal(pluginID, int(syscall.SIGKILL)); err != nil {
|
|
||||||
logrus.Errorf("Sending SIGKILL to plugin failed with error: %v", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue