70 lines
1.3 KiB
Go
70 lines
1.3 KiB
Go
package setup
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/miekg/coredns/middleware"
|
|
prom "github.com/miekg/coredns/middleware/prometheus"
|
|
)
|
|
|
|
const (
|
|
path = "/metrics"
|
|
addr = "localhost:9153"
|
|
)
|
|
|
|
var once sync.Once
|
|
|
|
func Prometheus(c *Controller) (middleware.Middleware, error) {
|
|
metrics, err := parsePrometheus(c)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if metrics.Addr == "" {
|
|
metrics.Addr = addr
|
|
}
|
|
once.Do(func() {
|
|
c.Startup = append(c.Startup, metrics.Start)
|
|
})
|
|
|
|
return func(next middleware.Handler) middleware.Handler {
|
|
metrics.Next = next
|
|
return metrics
|
|
}, nil
|
|
}
|
|
|
|
func parsePrometheus(c *Controller) (*prom.Metrics, error) {
|
|
var (
|
|
metrics *prom.Metrics
|
|
err error
|
|
)
|
|
|
|
for c.Next() {
|
|
if metrics != nil {
|
|
return nil, c.Err("prometheus: can only have one metrics module per server")
|
|
}
|
|
metrics = &prom.Metrics{ZoneNames: c.ServerBlockHosts}
|
|
args := c.RemainingArgs()
|
|
|
|
switch len(args) {
|
|
case 0:
|
|
case 1:
|
|
metrics.Addr = args[0]
|
|
default:
|
|
return nil, c.ArgErr()
|
|
}
|
|
for c.NextBlock() {
|
|
switch c.Val() {
|
|
case "address":
|
|
args = c.RemainingArgs()
|
|
if len(args) != 1 {
|
|
return nil, c.ArgErr()
|
|
}
|
|
metrics.Addr = args[0]
|
|
default:
|
|
return nil, c.Errf("prometheus: unknown item: %s", c.Val())
|
|
}
|
|
|
|
}
|
|
}
|
|
return metrics, err
|
|
}
|