69 lines
1.4 KiB
Go
69 lines
1.4 KiB
Go
package timeouts
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/coredns/caddy"
|
|
"github.com/coredns/coredns/core/dnsserver"
|
|
"github.com/coredns/coredns/plugin"
|
|
"github.com/coredns/coredns/plugin/pkg/durations"
|
|
)
|
|
|
|
func init() { plugin.Register("timeouts", setup) }
|
|
|
|
func setup(c *caddy.Controller) error {
|
|
err := parseTimeouts(c)
|
|
if err != nil {
|
|
return plugin.Error("timeouts", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func parseTimeouts(c *caddy.Controller) error {
|
|
config := dnsserver.GetConfig(c)
|
|
|
|
for c.Next() {
|
|
args := c.RemainingArgs()
|
|
if len(args) > 0 {
|
|
return plugin.Error("timeouts", c.ArgErr())
|
|
}
|
|
|
|
b := 0
|
|
for c.NextBlock() {
|
|
block := c.Val()
|
|
timeoutArgs := c.RemainingArgs()
|
|
if len(timeoutArgs) != 1 {
|
|
return c.ArgErr()
|
|
}
|
|
|
|
timeout, err := durations.NewDurationFromArg(timeoutArgs[0])
|
|
if err != nil {
|
|
return c.Err(err.Error())
|
|
}
|
|
|
|
if timeout < (1*time.Second) || timeout > (24*time.Hour) {
|
|
return c.Errf("timeout provided '%s' needs to be between 1 second and 24 hours", timeout)
|
|
}
|
|
|
|
switch block {
|
|
case "read":
|
|
config.ReadTimeout = timeout
|
|
|
|
case "write":
|
|
config.WriteTimeout = timeout
|
|
|
|
case "idle":
|
|
config.IdleTimeout = timeout
|
|
|
|
default:
|
|
return c.Errf("unknown option: '%s'", block)
|
|
}
|
|
b++
|
|
}
|
|
|
|
if b == 0 {
|
|
return plugin.Error("timeouts", c.Err("timeouts block with no timeouts specified"))
|
|
}
|
|
}
|
|
return nil
|
|
}
|