vault: make logger configurationvault/v0.2.4

1 files changed, 7 insertions, 3 deletions

diff --git a/vault/logger.go b/vault/logger.go
index 7e16895..a83eb33 100644
--- a/vault/logger.go
+++ b/vault/logger.go
@@ -2,13 +2,16 @@ package vault
 
 import (
         "context"
-        "log"
         "sync"
 
         "code.crute.us/mcrute/golib/service"
 )
 
-func MakeRenewalLogger(rc chan *Renewal) service.RunnerFunc {
+type VaultInfofLogger interface {
+        Infof(string, ...interface{})
+}
+
+func MakeRenewalLogger(rc chan *Renewal, log VaultInfofLogger) service.RunnerFunc {
         return func(ctx context.Context, wg *sync.WaitGroup) error {
                 wg.Add(1)
                 defer wg.Done()
@@ -16,8 +19,9 @@ func MakeRenewalLogger(rc chan *Renewal) service.RunnerFunc {
                 for {
                         select {
                         case r := <-rc:
-                                log.Printf("Renewing credential %s at %s", r.Name, r.RenewedAt)
+                                log.Infof("Renewing credential %s at %s", r.Name, r.RenewedAt)
                         case <-ctx.Done():
+                                log.Infof("Shutting down vault renewal logger")
                                 return nil
                         }
                 }