Querylog -- Implement file writing and update /querylog handler for changed structures.

coredns_plugin/coredns_plugin.go

@@ -219,6 +219,7 @@ func setup(c *caddy.Controller) error {
 		return nil
 	})
 	c.OnShutdown(p.onShutdown)
+	c.OnFinalShutdown(p.onFinalShutdown)
 
 	return nil
 }
@@ -250,6 +251,15 @@ func (p *plug) onShutdown() error {
 	return nil
 }
 
+func (p *plug) onFinalShutdown() error {
+	err := flushToFile(logBuffer)
+	if err != nil {
+		log.Printf("failed to flush to file: %s", err)
+		return err
+	}
+	return nil
+}
+
 type statsFunc func(ch interface{}, name string, text string, value float64, valueType prometheus.ValueType)
 
 func doDesc(ch interface{}, name string, text string, value float64, valueType prometheus.ValueType) {