diff options
Diffstat (limited to 'middleware/metrics/setup.go')
| -rw-r--r-- | middleware/metrics/setup.go | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/middleware/metrics/setup.go b/middleware/metrics/setup.go index de82a1f1b..1af41222e 100644 --- a/middleware/metrics/setup.go +++ b/middleware/metrics/setup.go @@ -28,9 +28,10 @@ func setup(c *caddy.Controller) error { return m }) + // During restarts we will keep this handler running. metricsOnce.Do(func() { - c.OnStartup(m.OnStartup) - c.OnShutdown(m.OnShutdown) + c.OncePerServerBlock(m.OnStartup) + c.OnFinalShutdown(m.OnShutdown) }) return nil |