Fix issues with plugin and --live-restore
Fixes an issue when starting the daemon with live-restore where previously it was not set, plugins are not running. Fixes an issue when starting the daemon with live-restore, the plugin client (for interacting with the plugins HTTP interface) is not set, causing a panic when the plugin is called. Signed-off-by: Brian Goff <cpuguy83@gmail.com>
This commit is contained in:
parent
39473dad54
commit
cb6633175c
2 changed files with 30 additions and 3 deletions
|
@ -129,7 +129,7 @@ func (s *DockerDaemonSuite) TestDaemonShutdownLiveRestoreWithPlugins(c *check.C)
|
|||
|
||||
// TestDaemonShutdownWithPlugins shuts down running plugins.
|
||||
func (s *DockerDaemonSuite) TestDaemonShutdownWithPlugins(c *check.C) {
|
||||
testRequires(c, IsAmd64, Network)
|
||||
testRequires(c, IsAmd64, Network, SameHostDaemon)
|
||||
|
||||
s.d.Start(c)
|
||||
if out, err := s.d.Cmd("plugin", "install", "--grant-all-permissions", pName); err != nil {
|
||||
|
@ -160,6 +160,11 @@ func (s *DockerDaemonSuite) TestDaemonShutdownWithPlugins(c *check.C) {
|
|||
if out, ec, err := runCommandWithOutput(cmd); ec != 1 {
|
||||
c.Fatalf("Expected exit code '1', got %d err: %v output: %s ", ec, err, out)
|
||||
}
|
||||
|
||||
s.d.Start(c, "--live-restore")
|
||||
cmd = exec.Command("pgrep", "-f", pluginProcessName)
|
||||
out, _, err := runCommandWithOutput(cmd)
|
||||
c.Assert(err, checker.IsNil, check.Commentf(out))
|
||||
}
|
||||
|
||||
// TestVolumePlugin tests volume creation using a plugin.
|
||||
|
|
|
@ -49,6 +49,10 @@ func (pm *Manager) enable(p *v2.Plugin, c *controller, force bool) error {
|
|||
return err
|
||||
}
|
||||
|
||||
return pm.pluginPostStart(p, c)
|
||||
}
|
||||
|
||||
func (pm *Manager) pluginPostStart(p *v2.Plugin, c *controller) error {
|
||||
client, err := plugins.NewClientWithTimeout("unix://"+filepath.Join(p.GetRuntimeSourcePath(), p.GetSocket()), nil, c.timeoutInSecs)
|
||||
if err != nil {
|
||||
c.restart = false
|
||||
|
@ -59,12 +63,30 @@ func (pm *Manager) enable(p *v2.Plugin, c *controller, force bool) error {
|
|||
p.SetPClient(client)
|
||||
pm.pluginStore.SetState(p, true)
|
||||
pm.pluginStore.CallHandler(p)
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (pm *Manager) restore(p *v2.Plugin) error {
|
||||
return pm.containerdClient.Restore(p.GetID(), attachToLog(p.GetID()))
|
||||
if err := pm.containerdClient.Restore(p.GetID(), attachToLog(p.GetID())); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if pm.liveRestore {
|
||||
c := &controller{}
|
||||
if pids, _ := pm.containerdClient.GetPidsForContainer(p.GetID()); len(pids) == 0 {
|
||||
// plugin is not running, so follow normal startup procedure
|
||||
return pm.enable(p, c, true)
|
||||
}
|
||||
|
||||
c.exitChan = make(chan bool)
|
||||
c.restart = true
|
||||
pm.mu.Lock()
|
||||
pm.cMap[p] = c
|
||||
pm.mu.Unlock()
|
||||
return pm.pluginPostStart(p, c)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func shutdownPlugin(p *v2.Plugin, c *controller, containerdClient libcontainerd.Client) {
|
||||
|
|
Loading…
Reference in a new issue