update param urls from single url to list of urls setup esClient

Co-authored-by: tarabaskara <[email protected]>
BaritoLog · Aug 30, 2019 · 72cd74b · 72cd74b
1 parent ab8a382
commit 72cd74b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/flow/elastic.go b/flow/elastic.go
@@ -43,7 +43,7 @@ func NewEsConfig(indexMethod string, bulkSize int, flushMs time.Duration, printT
 	}
 }
 
-func NewElastic(retrierFunc *ElasticRetrier, esConfig esConfig, urls ...string) (client elasticClient, err error) {
+func NewElastic(retrierFunc *ElasticRetrier, esConfig esConfig, urls []string) (client elasticClient, err error) {
 
 	c, err := elastic.NewClient(
 		elastic.SetURL(urls...),

diff --git a/flow/elastic_test.go b/flow/elastic_test.go
@@ -27,7 +27,7 @@ func TestElasticStore_CreateIndexError(t *testing.T) {
 
 	retrier := mockElasticRetrier()
 	esConfig := NewEsConfig("SingleInsert", 100, time.Duration(1000), false)
-	client, err := NewElastic(retrier, esConfig, ts.URL)
+	client, err := NewElastic(retrier, esConfig, []string{ts.URL})
 	FatalIfError(t, err)
 
 	err = client.Store(context.Background(), timber)
@@ -50,7 +50,7 @@ func TestElasticStore_CreateindexSuccess(t *testing.T) {
 
 	retrier := mockElasticRetrier()
 	esConfig := NewEsConfig("SingleInsert", 100, time.Duration(1000), false)
-	client, err := NewElastic(retrier, esConfig, ts.URL)
+	client, err := NewElastic(retrier, esConfig, []string{ts.URL})
 	FatalIfError(t, err)
 
 	appSecret := timber.Context().AppSecret
@@ -76,7 +76,7 @@ func TestElasticStoreman_store_SaveError(t *testing.T) {
 
 	retrier := mockElasticRetrier()
 	esConfig := NewEsConfig("SingleInsert", 100, time.Duration(1000), false)
-	client, err := NewElastic(retrier, esConfig, ts.URL)
+	client, err := NewElastic(retrier, esConfig, []string{ts.URL})
 	FatalIfError(t, err)
 
 	appSecret := timber.Context().AppSecret