mirror of
https://github.com/Kugelschieber/marvinblum.git
synced 2026-01-18 14:50:27 +00:00
71 lines
1.9 KiB
Go
71 lines
1.9 KiB
Go
package tracking
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"github.com/emvi/logbuch"
|
|
"github.com/emvi/pirsch"
|
|
"github.com/jmoiron/sqlx"
|
|
"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`
|
|
)
|
|
|
|
var db *sqlx.DB
|
|
var store pirsch.Store
|
|
|
|
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)
|
|
conn, err := sql.Open("postgres", connectionStr)
|
|
|
|
if err != nil {
|
|
logbuch.Fatal("Error connecting to database", logbuch.Fields{"err": err})
|
|
return nil
|
|
}
|
|
|
|
if err := conn.Ping(); err != nil {
|
|
logbuch.Fatal("Error pinging database", logbuch.Fields{"err": err})
|
|
return nil
|
|
}
|
|
|
|
db = sqlx.NewDb(conn, "postgres")
|
|
store = pirsch.NewPostgresStore(conn)
|
|
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")
|
|
}
|