63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
|
// Copyright (C) 2018 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
"sync"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
_ "github.com/lib/pq"
|
||
|
|
_ "github.com/mattn/go-sqlite3"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
sqlupsert = map[string]string{
|
||
|
|
"sqlite3": "INSERT INTO metrics (metric, instance, val, timestamp) " +
|
||
|
|
"VALUES (?, ?, ?, ?) ON CONFLICT(metric, instance) DO UPDATE SET " +
|
||
|
|
"val=excluded.val, timestamp=excluded.timestamp;",
|
||
|
|
"postgres": "INSERT INTO metrics (metric, instance, val, timestamp) " +
|
||
|
|
"VALUES ($1, $2, $3, $4) ON CONFLICT(metric, instance) DO UPDATE SET " +
|
||
|
|
"val=EXCLUDED.val, timestamp=EXCLUDED.timestamp;",
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
// DBDest is a database metric destination. It stores the latest value given
|
||
|
|
// a metric key and application per instance.
|
||
|
|
type DBDest struct {
|
||
|
|
mtx sync.Mutex
|
||
|
|
driver, address string
|
||
|
|
db *sql.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewDBDest creates a DBDest
|
||
|
|
func NewDBDest(driver, address string) *DBDest {
|
||
|
|
if _, found := sqlupsert[driver]; !found {
|
||
|
|
panic(fmt.Sprintf("driver %s not supported", driver))
|
||
|
|
}
|
||
|
|
return &DBDest{
|
||
|
|
driver: driver,
|
||
|
|
address: address,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Metric implements the MetricDest interface
|
||
|
|
func (db *DBDest) Metric(application, instance string, key []byte, val float64,
|
||
|
|
ts time.Time) error {
|
||
|
|
db.mtx.Lock()
|
||
|
|
if db.db == nil {
|
||
|
|
conn, err := sql.Open(db.driver, db.address)
|
||
|
|
if err != nil {
|
||
|
|
db.mtx.Unlock()
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
db.db = conn
|
||
|
|
}
|
||
|
|
db.mtx.Unlock()
|
||
|
|
_, err := db.db.Exec(sqlupsert[db.driver], application+"."+string(key),
|
||
|
|
instance, val, ts.Unix())
|
||
|
|
return err
|
||
|
|
}
|