From 693320eb9d6642f211b38e2e3160b9a4e447e41e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Se=C3=A1n=20C=20McCord?= <ulexus@gmail.com>
Date: Wed, 16 May 2018 15:43:25 -0400
Subject: [PATCH] use oklog when in k8s

---
 app/init.go | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/app/init.go b/app/init.go
index 2305d73..66f7c29 100644
--- a/app/init.go
+++ b/app/init.go
@@ -1,6 +1,11 @@
 package app
 
-import "github.com/revel/revel"
+import (
+	"os"
+
+	"github.com/revel/log15"
+	"github.com/revel/revel"
+)
 
 func init() {
 	// Filters is the default set of global filters.
@@ -23,8 +28,10 @@ func init() {
 	// ( order dependent )
 	// revel.OnAppStart(InitDB)
 	// revel.OnAppStart(FillCache)
+	revel.OnAppStart(InitLogger)
 }
 
+// HeaderFilter is a default set of headers to be added
 // TODO turn this into revel.HeaderFilter
 // should probably also have a filter for CSRF
 // not sure if it can go in the same filter or not
@@ -36,3 +43,17 @@ var HeaderFilter = func(c *revel.Controller, fc []revel.Filter) {
 
 	fc[0](c, fc[1:]) // Execute the next filter stage.
 }
+
+// InitLogger initializes the revel logger
+func InitLogger() {
+
+	// If we are running inside kubernetes, use the oklog logger
+	if os.Getenv("KUBERNETES_SERVICE_HOST") != "" {
+		h, err := log15.NetHandler("tcp", "oklog.log:7651", log15.JsonFormat())
+		if err != nil {
+			revel.RootLog.Error("failed to construct network log handler", "error", err)
+			return
+		}
+		revel.RootLog.SetHandler(h)
+	}
+}