diff options
Diffstat (limited to 'plugin')
-rw-r--r-- | plugin/clouddns/setup.go | 18 | ||||
-rw-r--r-- | plugin/route53/route53_test.go | 6 | ||||
-rw-r--r-- | plugin/route53/setup.go | 19 |
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 } |