Small refactoring and improved legal page and more whitespace.

tracking/tracking.go

@@ -0,0 +1,66 @@
+package tracking
+
+import (
+	"database/sql"
+	"fmt"
+	"github.com/emvi/logbuch"
+	"github.com/emvi/pirsch"
+	"os"
+	"strconv"
+	"time"
+)
+
+const (
+	connectionString = `host=%s port=%s user=%s password=%s dbname=%s sslmode=%s sslcert=%s sslkey=%s sslrootcert=%s connectTimeout=%s timezone=%s`
+)
+
+func NewTracker() *pirsch.Tracker {
+	logbuch.Info("Connecting to database...")
+	host := os.Getenv("MB_DB_HOST")
+	port := os.Getenv("MB_DB_PORT")
+	user := os.Getenv("MB_DB_USER")
+	password := os.Getenv("MB_DB_PASSWORD")
+	schema := os.Getenv("MB_DB_SCHEMA")
+	sslMode := os.Getenv("MB_DB_SSLMODE")
+	sslCert := os.Getenv("MB_DB_SSLCERT")
+	sslKey := os.Getenv("MB_DB_SSLKEY")
+	sslRootCert := os.Getenv("MB_DB_SSLROOTCERT")
+	zone, offset := time.Now().Zone()
+	timezone := zone + strconv.Itoa(-offset/3600)
+	logbuch.Info("Setting time zone", logbuch.Fields{"timezone": timezone})
+	connectionStr := fmt.Sprintf(connectionString, host, port, user, password, schema, sslMode, sslCert, sslKey, sslRootCert, "30", timezone)
+	db, err := sql.Open("postgres", connectionStr)
+
+	if err != nil {
+		logbuch.Fatal("Error connecting to database", logbuch.Fields{"err": err})
+		return nil
+	}
+
+	if err := db.Ping(); err != nil {
+		logbuch.Fatal("Error pinging database", logbuch.Fields{"err": err})
+		return nil
+	}
+
+	store := pirsch.NewPostgresStore(db)
+	tracker := pirsch.NewTracker(store, nil)
+	processor := pirsch.NewProcessor(store)
+	processTrackingData(processor)
+	pirsch.RunAtMidnight(func() {
+		processTrackingData(processor)
+	})
+
+	return tracker
+}
+
+func processTrackingData(processor *pirsch.Processor) {
+	logbuch.Info("Processing tracking data...")
+
+	defer func() {
+		if err := recover(); err != nil {
+			logbuch.Error("Error processing tracking data", logbuch.Fields{"err": err})
+		}
+	}()
+
+	processor.Process()
+	logbuch.Info("Done processing tracking data")
+}