From a2dac0365ed3522c2a88388cca0480c61c908231 Mon Sep 17 00:00:00 2001 From: Martin Geno Date: Fri, 21 Apr 2017 00:45:45 +0200 Subject: [PATCH] add config to elasiticsearch database type --- database/elasticsearch/elasticsearch.go | 37 +++++++++++++------------ 1 file changed, 19 insertions(+), 18 deletions(-) diff --git a/database/elasticsearch/elasticsearch.go b/database/elasticsearch/elasticsearch.go index 3533ff6..d1fe5c7 100644 --- a/database/elasticsearch/elasticsearch.go +++ b/database/elasticsearch/elasticsearch.go @@ -6,16 +6,14 @@ package logging * - example for other developers for new databases */ import ( - + "context" "log" "time" - "context" + + elastic "gopkg.in/olivere/elastic.v5" "github.com/FreifunkBremen/yanic/database" "github.com/FreifunkBremen/yanic/runtime" - "gopkg.in/olivere/elastic.v5" - - ) type Connection struct { @@ -29,8 +27,20 @@ type Config map[string]interface{} func (c Config) Enable() bool { return c["enable"].(bool) } -func (c Config) Path() string { - return c["path"].(string) +func (c Config) Host() string { + return c["host"].(string) +} +func (c Config) Username() string { + return c["username"].(string) +} +func (c Config) Password() string { + return c["password"].(string) +} +func (c Config) IndexPrefix() string { + return c["index_prefix"].(string) +} +func (c Config) UpdateTemplates() bool { + return c["update_templates"].(bool) } func init() { @@ -44,20 +54,16 @@ func Connect(configuration interface{}) (database.Connection, error) { 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")) - - + elastic.SetURL(config.Host()), + elastic.SetBasicAuth(config.Username(), config.Password())) if err != nil { // Handle error panic(err) } - return &Connection{config: config, client: client}, nil } @@ -71,14 +77,11 @@ func (conn *Connection) InsertNode(node *runtime.Node) { Refresh("true"). Do(context.Background()) - - if err != nil { // Handle error panic(err) } - } func (conn *Connection) InsertGlobals(stats *runtime.GlobalStats, time time.Time) { @@ -91,13 +94,11 @@ func (conn *Connection) InsertGlobals(stats *runtime.GlobalStats, time time.Time Refresh("true"). Do(context.Background()) - if err != nil { // Handle error panic(err) } - } func (conn *Connection) PruneNodes(deleteAfter time.Duration) {