diff --git a/caddy.go b/caddy.go index d2f60001d..1adf01134 100644 --- a/caddy.go +++ b/caddy.go @@ -111,6 +111,7 @@ type Instance struct { onFirstStartup []func() error // starting, not as part of a restart onStartup []func() error // starting, even as part of a restart onRestart []func() error // before restart commences + onRestartFailed []func() error // if restart failed onShutdown []func() error // stopping, even as part of a restart onFinalShutdown []func() error // stopping, not as part of a restart @@ -186,9 +187,26 @@ func (i *Instance) Restart(newCaddyfile Input) (*Instance, error) { i.wg.Add(1) defer i.wg.Done() + var err error + // if something went wrong on restart then run onRestartFailed callbacks + defer func() { + r := recover() + if err != nil || r != nil { + for _, fn := range i.onRestartFailed { + err = fn() + if err != nil { + log.Printf("[ERROR] restart failed: %v", err) + } + } + if r != nil { + panic(r) + } + } + }() + // run restart callbacks for _, fn := range i.onRestart { - err := fn() + err = fn() if err != nil { return i, err } @@ -224,7 +242,7 @@ func (i *Instance) Restart(newCaddyfile Input) (*Instance, error) { newInst := &Instance{serverType: newCaddyfile.ServerType(), wg: i.wg, Storage: make(map[interface{}]interface{})} // attempt to start new instance - err := startWithListenerFds(newCaddyfile, newInst, restartFds) + err = startWithListenerFds(newCaddyfile, newInst, restartFds) if err != nil { return i, err } diff --git a/caddy_test.go b/caddy_test.go index 76cd2ec25..0a18100f6 100644 --- a/caddy_test.go +++ b/caddy_test.go @@ -15,9 +15,14 @@ package caddy import ( + "fmt" "net" + "reflect" "strconv" + "sync" "testing" + + "github.com/mholt/caddy/caddyfile" ) /* @@ -48,6 +53,70 @@ func TestCaddyStartStop(t *testing.T) { } */ +// CallbackTestContext implements Context interface +type CallbackTestContext struct { + // If MakeServersFail is set to true then MakeServers returns an error + MakeServersFail bool +} + +func (h *CallbackTestContext) InspectServerBlocks(name string, sblock []caddyfile.ServerBlock) ([]caddyfile.ServerBlock, error) { + return sblock, nil +} +func (h *CallbackTestContext) MakeServers() ([]Server, error) { + if h.MakeServersFail { + return make([]Server, 0), fmt.Errorf("MakeServers failed") + } + return make([]Server, 0), nil +} + +func TestCaddyRestartCallbacks(t *testing.T) { + for i, test := range []struct { + restartFail bool + expectedCalls []string + }{ + {false, []string{"OnRestart", "OnShutdown"}}, + {true, []string{"OnRestart", "OnRestartFailed"}}, + } { + serverName := fmt.Sprintf("%v", i) + // RegisterServerType to make successful restart possible + RegisterServerType(serverName, ServerType{ + Directives: func() []string { return []string{} }, + // If MakeServersFail is true then the restart will fail due to context failure + NewContext: func(inst *Instance) Context { return &CallbackTestContext{MakeServersFail: test.restartFail} }, + }) + c := NewTestController(serverName, "") + c.instance = &Instance{ + serverType: serverName, + wg: new(sync.WaitGroup), + } + + // Register callbacks which save the calls order + calls := make([]string, 0) + c.OnRestart(func() error { + calls = append(calls, "OnRestart") + return nil + }) + c.OnRestartFailed(func() error { + calls = append(calls, "OnRestartFailed") + return nil + }) + c.OnShutdown(func() error { + calls = append(calls, "OnShutdown") + return nil + }) + + c.instance.Restart(CaddyfileInput{Contents: []byte(""), ServerTypeName: serverName}) + + if !reflect.DeepEqual(calls, test.expectedCalls) { + t.Errorf("Test %d: Callbacks expected: %v, got: %v", i, test.expectedCalls, calls) + } + + c.instance.Stop() + c.instance.Wait() + } + +} + func TestIsLoopback(t *testing.T) { for i, test := range []struct { input string diff --git a/controller.go b/controller.go index 6015d210f..f63cebe00 100644 --- a/controller.go +++ b/controller.go @@ -86,6 +86,12 @@ func (c *Controller) OnRestart(fn func() error) { c.instance.onRestart = append(c.instance.onRestart, fn) } +// OnRestartFailed adds fn to the list of callback functions to execute +// if the server failed to restart. +func (c *Controller) OnRestartFailed(fn func() error) { + c.instance.onRestartFailed = append(c.instance.onRestartFailed, fn) +} + // OnShutdown adds fn to the list of callback functions to execute // when the server is about to be shut down (including restarts). func (c *Controller) OnShutdown(fn func() error) {