diff --git a/glrdomon.go b/glrdomon.go index 51b77a9d085f4ef79815a9f0d195c1b99a9ca441..1fb284d7b1ea858db61c4dbc22764da0855d1a52 100644 --- a/glrdomon.go +++ b/glrdomon.go @@ -30,8 +30,7 @@ type tokenSource struct { var ( configPath string - logger *log.Logger - logDest string + logger *log.Logger apiKey string threshold int @@ -64,8 +63,7 @@ func initConfig() { threshold = cfg.Threshold deleteStale = cfg.DeleteStale - logDest = cfg.LogDest - setUpLogger() + setUpLogger(cfg.LogDest) logger.Printf("Starting GitLab Runner monitoring with config %s", configPath) @@ -194,7 +192,7 @@ func deleteDroplet(droplet godo.Droplet) bool { return err == nil } -func setUpLogger() { +func setUpLogger(logDest string) { logOpts := log.Ldate | log.Ltime | log.LUTC | log.Lshortfile if logDest == "os.Stdout" { diff --git a/glrdomon_test.go b/glrdomon_test.go index 3604ed9f15b3760c683a739a188d8c8e92ac6457..35ffc0a3b92cb01b67e1bb109791e6de375d083c 100644 --- a/glrdomon_test.go +++ b/glrdomon_test.go @@ -8,7 +8,7 @@ import ( ) func TestCheckDropletAge(t *testing.T) { - setUpLogger() + setUpLogger("os.stdOut") staleDroplet := godo.Droplet{ ID: 1234,