114 lines
2.0 KiB
Go
114 lines
2.0 KiB
Go
package logging
|
|
|
|
/**
|
|
* This database type is just for,
|
|
* - debugging without a influxconn
|
|
* - example for other developers for new databases
|
|
*/
|
|
import (
|
|
|
|
"log"
|
|
"time"
|
|
"context"
|
|
|
|
"github.com/FreifunkBremen/yanic/database"
|
|
"github.com/FreifunkBremen/yanic/runtime"
|
|
"gopkg.in/olivere/elastic.v5"
|
|
|
|
|
|
)
|
|
|
|
type Connection struct {
|
|
database.Connection
|
|
config Config
|
|
client *elastic.Client
|
|
}
|
|
|
|
type Config map[string]interface{}
|
|
|
|
func (c Config) Enable() bool {
|
|
return c["enable"].(bool)
|
|
}
|
|
func (c Config) Path() string {
|
|
return c["path"].(string)
|
|
}
|
|
|
|
func init() {
|
|
database.RegisterAdapter("elasticsearch", Connect)
|
|
}
|
|
|
|
func Connect(configuration interface{}) (database.Connection, error) {
|
|
var config Config
|
|
config = configuration.(map[string]interface{})
|
|
if !config.Enable() {
|
|
return nil, nil
|
|
}
|
|
|
|
|
|
// Create a client
|
|
client, err := elastic.NewClient(
|
|
elastic.SetURL("http://127.0.0.1:9200", "http://127.0.0.1:9201"),
|
|
elastic.SetBasicAuth("user", "secret"))
|
|
|
|
|
|
|
|
if err != nil {
|
|
// Handle error
|
|
panic(err)
|
|
}
|
|
|
|
|
|
return &Connection{config: config, client: client}, nil
|
|
}
|
|
|
|
func (conn *Connection) InsertNode(node *runtime.Node) {
|
|
log.Print("InsertNode: [", node.Statistics.NodeID, "] clients: ", node.Statistics.Clients.Total)
|
|
|
|
_, err := conn.client.Index().
|
|
Index("ffhb").
|
|
Type("node").
|
|
BodyJson(node).
|
|
Refresh("true").
|
|
Do(context.Background())
|
|
|
|
|
|
|
|
if err != nil {
|
|
// Handle error
|
|
panic(err)
|
|
}
|
|
|
|
|
|
}
|
|
|
|
func (conn *Connection) InsertGlobals(stats *runtime.GlobalStats, time time.Time) {
|
|
log.Print("InsertGlobals: [", time.String(), "] nodes: ", stats.Nodes, ", clients: ", stats.Clients, " models: ", len(stats.Models))
|
|
|
|
_, err := conn.client.Index().
|
|
Index("ffhb").
|
|
Type("globals").
|
|
BodyJson(stats).
|
|
Refresh("true").
|
|
Do(context.Background())
|
|
|
|
|
|
if err != nil {
|
|
// Handle error
|
|
panic(err)
|
|
}
|
|
|
|
|
|
}
|
|
|
|
func (conn *Connection) PruneNodes(deleteAfter time.Duration) {
|
|
log.Print("PruneNodes")
|
|
|
|
// TODO
|
|
}
|
|
|
|
func (conn *Connection) Close() {
|
|
|
|
log.Print("Closing connection, stop client")
|
|
conn.client.Stop()
|
|
}
|