diff options
-rw-r--r-- | echo/vault.go | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/echo/vault.go b/echo/vault.go index 13a7407..68de60b 100644 --- a/echo/vault.go +++ b/echo/vault.go | |||
@@ -2,6 +2,7 @@ package echo | |||
2 | 2 | ||
3 | import ( | 3 | import ( |
4 | "context" | 4 | "context" |
5 | "sync" | ||
5 | 6 | ||
6 | "code.crute.us/mcrute/golib/secrets" | 7 | "code.crute.us/mcrute/golib/secrets" |
7 | "code.crute.us/mcrute/golib/service" | 8 | "code.crute.us/mcrute/golib/service" |
@@ -51,6 +52,16 @@ func MakeVaultSecretsClient(ctx context.Context) (secrets.ClientManager, error) | |||
51 | // failure handler will terminate the application if a critical | 52 | // failure handler will terminate the application if a critical |
52 | // credential renewal failure occurs. | 53 | // credential renewal failure occurs. |
53 | func AttachSecretsClient(c secrets.ClientManager, cancel func(), run *service.AppRunner, log echo.Logger) { | 54 | func AttachSecretsClient(c secrets.ClientManager, cancel func(), run *service.AppRunner, log echo.Logger) { |
54 | run.AddBackgroundJob(c.Run) | 55 | run.AddBackgroundJob(func(ctx context.Context, wg *sync.WaitGroup) error { |
55 | run.AddBackgroundJob(secrets.MakeRenewalLogger(c, log, cancel)) | 56 | log.Info("Starting credential renewal handler") |
57 | err := c.Run(ctx, wg) | ||
58 | log.Info("Shutting down credential renewal handler") | ||
59 | return err | ||
60 | }) | ||
61 | run.AddBackgroundJob(func(ctx context.Context, wg *sync.WaitGroup) error { | ||
62 | log.Info("Starting credential renewal logger") | ||
63 | err := secrets.MakeRenewalLogger(c, log, cancel)(ctx, wg) | ||
64 | log.Info("Shutting down credential renewal logger") | ||
65 | return err | ||
66 | }) | ||
56 | } | 67 | } |