add log to influx

cache.go

@@ -4,7 +4,7 @@ import (
 	"image"
 
 	"github.com/fogleman/gg"
-	log "github.com/sirupsen/logrus"
+	"github.com/sirupsen/logrus"
 )
 
 var cacheImages = map[int]image.Image{}
@@ -15,7 +15,7 @@ func getBaseImage(week Days) image.Image {
 	if !ok || refreshCache {
 		img = generateBaseImage(week)
 
-		log.Infof("miss cache")
+		logrus.Infof("miss cache")
 
 		cacheImages[int(week)] = img
 	}

config.json.dist

@@ -71,5 +71,12 @@
     },
     "http": {
         "port": 9090
+    },
+    "influxdb": {
+        "host": "",
+        "username": "",
+        "password": "",
+        "database": "",
+        "measurement": "cards-generator"
     }
 }

generate.go

@@ -6,12 +6,16 @@ import (
 	"strconv"
 	"sync"
 
+	"time"
+
 	"github.com/fogleman/gg"
+	"github.com/sirupsen/logrus"
 )
 
 var mutex = &sync.Mutex{}
 
 func generateStoptimeCards(w http.ResponseWriter, r *http.Request) {
+	t := time.Now()
 	v := r.URL.Query()
 
 	mission := v.Get("mission")
@@ -20,9 +24,17 @@ func generateStoptimeCards(w http.ResponseWriter, r *http.Request) {
 	schedule := v.Get("schedule")
 	days, err := strconv.Atoi(v.Get("days"))
 
+	logger := logrus.WithFields(logrus.Fields{
+		"action": "generate",
+		"type":   "stop_time",
+		"query":  r.URL.RawQuery,
+	})
+
 	if mission == "" || origin == "" || terminus == "" || schedule == "" || days == 0 || days > 127 || err != nil {
 		w.WriteHeader(http.StatusBadRequest)
 		w.Write([]byte("{\"status\": \"bad request\"}"))
+
+		logger.WithField("status", "bad_request").Warn("bad_request")
 		return
 	}
 
@@ -41,7 +53,17 @@ func generateStoptimeCards(w http.ResponseWriter, r *http.Request) {
 	dc.DrawStringAnchored(origin, xOrigin, yOrigin, 0, 1)
 	dc.DrawStringAnchored(terminus, xTerminus, yTerminus, 0, 1)
 
-	dc.EncodePNG(w)
+	if err := dc.EncodePNG(w); err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+
+		logrus.WithField("status", "fail").Error(err.Error())
+		return
+	}
+
+	logger.WithFields(logrus.Fields{
+		"status":        "success",
+		"generate_time": time.Since(t).Seconds(),
+	}).Info("success")
 }
 
 func formatStation(name string) string {

init.go

@@ -4,8 +4,10 @@ import (
 	"flag"
 	"image/color"
 
+	"github.com/Abramovic/logrus_influxdb"
 	"github.com/fogleman/gg"
-	log "github.com/sirupsen/logrus"
+	client "github.com/influxdata/influxdb/client/v2"
+	"github.com/sirupsen/logrus"
 	"github.com/spf13/viper"
 	"golang.org/x/image/font"
 )
@@ -42,6 +44,7 @@ var (
 
 func init() {
 	initConfig()
+	initLog()
 	initFont()
 	initVar()
 	initDays()
@@ -56,7 +59,7 @@ func initConfig() {
 
 	// If a config file is found, read it in.
 	if err := viper.ReadInConfig(); err != nil {
-		log.Fatal(err)
+		logrus.Fatal(err)
 	}
 }
 
@@ -65,12 +68,12 @@ func initFont() {
 
 	fontRegular, err = gg.LoadFontFace(viper.GetString("font.regular.path"), viper.GetFloat64("font.regular.size"))
 	if err != nil {
-		log.Fatal(err)
+		logrus.Fatal(err)
 	}
 
 	fontBold, err = gg.LoadFontFace(viper.GetString("font.bold.path"), viper.GetFloat64("font.bold.size"))
 	if err != nil {
-		log.Fatal(err)
+		logrus.Fatal(err)
 	}
 }
 
@@ -139,3 +142,26 @@ func initDays() {
 		{Sunday, "D", widthPerCase*6 + widthPerCase/2},
 	}
 }
+
+func initLog() {
+	config := &logrus_influxdb.Config{
+		Database:    viper.GetString("influxdb.database"),
+		Measurement: viper.GetString("influxdb.measurement"),
+		Tags:        []string{"action", "status", "type"},
+	}
+
+	// Connect to InfluxDB using the standard client.
+	influxClient, _ := client.NewHTTPClient(client.HTTPConfig{
+		Addr:     viper.GetString("influxdb.host"),
+		Username: viper.GetString("influxdb.username"),
+		Password: viper.GetString("influxdb.password"),
+	})
+
+	hook, err := logrus_influxdb.NewInfluxDB(config, influxClient)
+
+	if err != nil {
+		logrus.Fatal(err)
+	}
+
+	logrus.AddHook(hook)
+}

main.go

@@ -8,7 +8,7 @@ import (
 	"os/signal"
 	"time"
 
-	log "github.com/sirupsen/logrus"
+	"github.com/sirupsen/logrus"
 	"github.com/spf13/viper"
 )
 
@@ -25,24 +25,24 @@ func main() {
 	}
 
 	go func() {
-		log.Infof("HTTP Server listening on %s", h.Addr)
+		logrus.Infof("HTTP Server listening on %s", h.Addr)
 
 		if err := h.ListenAndServe(); err != nil {
-			log.Fatal(err.Error())
+			logrus.Fatal(err.Error())
 		}
 	}()
 
 	<-stop
 
-	log.Info("Graceful shutdown...")
+	logrus.Info("Graceful shutdown...")
 
 	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
 	defer cancel()
 
 	err := h.Shutdown(ctx)
 
 	if err != nil {
-		log.Fatal(err.Error())
+		logrus.Fatal(err.Error())
 	}
 }