mirror of
https://github.com/coredns/coredns.git
synced 2025-10-30 17:53:21 -04:00
34 lines
570 B
Go
34 lines
570 B
Go
|
package setup
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/miekg/coredns/middleware"
|
||
|
|
"github.com/miekg/coredns/middleware/health"
|
||
|
|
)
|
||
|
|
|
||
|
|
func Health(c *Controller) (middleware.Middleware, error) {
|
||
|
|
addr, err := parseHealth(c)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
h := health.Health{Addr: addr}
|
||
|
|
c.Startup = append(c.Startup, h.ListenAndServe)
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func parseHealth(c *Controller) (string, error) {
|
||
|
|
addr := ""
|
||
|
|
for c.Next() {
|
||
|
|
args := c.RemainingArgs()
|
||
|
|
|
||
|
|
switch len(args) {
|
||
|
|
case 0:
|
||
|
|
case 1:
|
||
|
|
addr = args[0]
|
||
|
|
default:
|
||
|
|
return "", c.ArgErr()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return addr, nil
|
||
|
|
}
|