Added loggly support

env/config.go

@@ -23,4 +23,6 @@ type Flags struct {
 
 	YubiCloudID  string
 	YubiCloudKey string
+
+	LogglyToken string
 }

main.go

@@ -66,6 +66,8 @@ var (
 	// YubiCloud params
 	yubiCloudID  = flag.String("yubicloud_id", "", "YubiCloud API id")
 	yubiCloudKey = flag.String("yubicloud_key", "", "YubiCloud API key")
+	// Loggly URL
+	logglyToken = flag.String("loggly_token", "", "Loggly token")
 )
 
 func main() {
@@ -95,6 +97,8 @@ func main() {
 
 		YubiCloudID:  *yubiCloudID,
 		YubiCloudKey: *yubiCloudKey,
+
+		LogglyToken: *logglyToken,
 	}
 
 	// Generate a mux

setup/loggly.go

@@ -0,0 +1,24 @@
+package setup
+
+import (
+	"github.com/Sirupsen/logrus"
+	"github.com/segmentio/go-loggly"
+)
+
+type logglyHook struct {
+	Loggly *loggly.Client
+}
+
+func (h *logglyHook) Fire(entry *logrus.Entry) error {
+	entry.Data["message"] = entry.Message
+	return h.Loggly.Send(map[string]interface{}(entry.Data))
+}
+
+func (h *logglyHook) Levels() []logrus.Level {
+	return []logrus.Level{
+		logrus.WarnLevel,
+		logrus.ErrorLevel,
+		logrus.FatalLevel,
+		logrus.PanicLevel,
+	}
+}

setup/setup.go

@@ -15,6 +15,7 @@ import (
 	"github.com/dancannon/gorethink"
 	"github.com/googollee/go-socket.io"
 	"github.com/rs/cors"
+	"github.com/segmentio/go-loggly"
 	"github.com/zenazn/goji/web"
 	"github.com/zenazn/goji/web/middleware"
 	"gopkg.in/igm/sockjs-go.v2/sockjs"
@@ -47,6 +48,13 @@ func PrepareMux(flags *env.Flags) *web.Mux {
 		log.Formatter = &logrus.JSONFormatter{}
 	}
 
+	// Install logrus hooks
+	if flags.LogglyToken != "" {
+		log.Hooks.Add(&logglyHook{
+			Loggly: loggly.New(flags.LogglyToken),
+		})
+	}
+
 	// Pass it to the environment package
 	env.Log = log