aboutsummaryrefslogtreecommitdiff
path: root/plugin
diff options
context:
space:
mode:
Diffstat (limited to 'plugin')
-rw-r--r--plugin/clouddns/setup.go18
-rw-r--r--plugin/route53/route53_test.go6
-rw-r--r--plugin/route53/setup.go19
3 files changed, 22 insertions, 21 deletions
diff --git a/plugin/clouddns/setup.go b/plugin/clouddns/setup.go
index 7de8efe89..3b6bcba5f 100644
--- a/plugin/clouddns/setup.go
+++ b/plugin/clouddns/setup.go
@@ -46,14 +46,14 @@ func setup(c *caddy.Controller) error {
for i := 0; i < len(args); i++ {
parts := strings.SplitN(args[i], ":", 3)
if len(parts) != 3 {
- return c.Errf("invalid zone '%s'", args[i])
+ return plugin.Error("clouddns", c.Errf("invalid zone %q", args[i]))
}
dnsName, projectName, hostedZone := parts[0], parts[1], parts[2]
if dnsName == "" || projectName == "" || hostedZone == "" {
- return c.Errf("invalid zone '%s'", args[i])
+ return plugin.Error("clouddns", c.Errf("invalid zone %q", args[i]))
}
if _, ok := keyPairs[args[i]]; ok {
- return c.Errf("conflict zone '%s'", args[i])
+ return plugin.Error("clouddns", c.Errf("conflict zone %q", args[i]))
}
keyPairs[args[i]] = struct{}{}
@@ -64,18 +64,17 @@ func setup(c *caddy.Controller) error {
for c.NextBlock() {
switch c.Val() {
case "upstream":
- c.RemainingArgs() // eats args
- // if filepath is provided in the Corefile use it to authenticate the dns client
+ c.RemainingArgs()
case "credentials":
if c.NextArg() {
opt = option.WithCredentialsFile(c.Val())
} else {
- return c.ArgErr()
+ return plugin.Error("clouddns", c.ArgErr())
}
case "fallthrough":
fall.SetZonesFromArgs(c.RemainingArgs())
default:
- return c.Errf("unknown property '%s'", c.Val())
+ return plugin.Error("clouddns", c.Errf("unknown property %q", c.Val()))
}
}
@@ -87,18 +86,19 @@ func setup(c *caddy.Controller) error {
h, err := New(ctx, client, keys, up)
if err != nil {
- return c.Errf("failed to create Cloud DNS plugin: %v", err)
+ return plugin.Error("clouddns", c.Errf("failed to create plugin: %v", err))
}
h.Fall = fall
if err := h.Run(ctx); err != nil {
- return c.Errf("failed to initialize Cloud DNS plugin: %v", err)
+ return plugin.Error("clouddns", c.Errf("failed to initialize plugin: %v", err))
}
dnsserver.GetConfig(c).AddPlugin(func(next plugin.Handler) plugin.Handler {
h.Next = next
return h
})
+ c.OnShutdown(func() error { ctx.Done(); return nil })
}
return nil
diff --git a/plugin/route53/route53_test.go b/plugin/route53/route53_test.go
index cc5540b6b..057313fa6 100644
--- a/plugin/route53/route53_test.go
+++ b/plugin/route53/route53_test.go
@@ -81,7 +81,7 @@ func TestRoute53(t *testing.T) {
r, err := New(ctx, fakeRoute53{}, map[string][]string{"bad.": {"0987654321"}}, time.Minute)
if err != nil {
- t.Fatalf("Failed to create Route53: %v", err)
+ t.Fatalf("Failed to create route53: %v", err)
}
if err = r.Run(ctx); err == nil {
t.Fatalf("Expected errors for zone bad.")
@@ -89,7 +89,7 @@ func TestRoute53(t *testing.T) {
r, err = New(ctx, fakeRoute53{}, map[string][]string{"org.": {"1357986420", "1234567890"}, "gov.": {"Z098765432", "1234567890"}}, 90*time.Second)
if err != nil {
- t.Fatalf("Failed to create Route53: %v", err)
+ t.Fatalf("Failed to create route53: %v", err)
}
r.Fall = fall.Zero
r.Fall.SetZonesFromArgs([]string{"gov."})
@@ -117,7 +117,7 @@ func TestRoute53(t *testing.T) {
})
err = r.Run(ctx)
if err != nil {
- t.Fatalf("Failed to initialize Route53: %v", err)
+ t.Fatalf("Failed to initialize route53: %v", err)
}
tests := []struct {
diff --git a/plugin/route53/setup.go b/plugin/route53/setup.go
index c285bee2d..1d1eaee95 100644
--- a/plugin/route53/setup.go
+++ b/plugin/route53/setup.go
@@ -53,14 +53,14 @@ func setup(c *caddy.Controller) error {
for i := 0; i < len(args); i++ {
parts := strings.SplitN(args[i], ":", 2)
if len(parts) != 2 {
- return plugin.Error("route53", c.Errf("invalid zone '%s'", args[i]))
+ return plugin.Error("route53", c.Errf("invalid zone %q", args[i]))
}
dns, hostedZoneID := parts[0], parts[1]
if dns == "" || hostedZoneID == "" {
- return plugin.Error("route53", c.Errf("invalid zone '%s'", args[i]))
+ return plugin.Error("route53", c.Errf("invalid zone %q", args[i]))
}
if _, ok := keyPairs[args[i]]; ok {
- return plugin.Error("route53", c.Errf("conflict zone '%s'", args[i]))
+ return plugin.Error("route53", c.Errf("conflict zone %q", args[i]))
}
keyPairs[args[i]] = struct{}{}
@@ -72,7 +72,7 @@ func setup(c *caddy.Controller) error {
case "aws_access_key":
v := c.RemainingArgs()
if len(v) < 2 {
- return plugin.Error("route53", c.Errf("invalid access key '%v'", v))
+ return plugin.Error("route53", c.Errf("invalid access key: '%v'", v))
}
providers = append(providers, &credentials.StaticProvider{
Value: credentials.Value{
@@ -102,16 +102,16 @@ func setup(c *caddy.Controller) error {
}
refresh, err = time.ParseDuration(refreshStr)
if err != nil {
- return plugin.Error("route53", c.Errf("Unable to parse duration: '%v'", err))
+ return plugin.Error("route53", c.Errf("Unable to parse duration: %v", err))
}
if refresh <= 0 {
- return plugin.Error("route53", c.Errf("refresh interval must be greater than 0: %s", refreshStr))
+ return plugin.Error("route53", c.Errf("refresh interval must be greater than 0: %q", refreshStr))
}
} else {
return plugin.Error("route53", c.ArgErr())
}
default:
- return plugin.Error("route53", c.Errf("unknown property '%s'", c.Val()))
+ return plugin.Error("route53", c.Errf("unknown property %q", c.Val()))
}
}
@@ -127,16 +127,17 @@ func setup(c *caddy.Controller) error {
ctx := context.Background()
h, err := New(ctx, client, keys, refresh)
if err != nil {
- return plugin.Error("route53", c.Errf("failed to create Route53 plugin: %v", err))
+ return plugin.Error("route53", c.Errf("failed to create route53 plugin: %v", err))
}
h.Fall = fall
if err := h.Run(ctx); err != nil {
- return plugin.Error("route53", c.Errf("failed to initialize Route53 plugin: %v", err))
+ return plugin.Error("route53", c.Errf("failed to initialize route53 plugin: %v", err))
}
dnsserver.GetConfig(c).AddPlugin(func(next plugin.Handler) plugin.Handler {
h.Next = next
return h
})
+ c.OnShutdown(func() error { ctx.Done(); return nil })
}
return nil
}