From 9eb04caf29ba620c2919a1fa54cf49a0e768a91c Mon Sep 17 00:00:00 2001
From: John Sun <sunforcherry@gmail.com>
Date: Tue, 14 Sep 2021 14:41:54 +0800
Subject: [PATCH] add sniff switch to es client

---
 pkg/database/es/client.go | 3 +++
 pkg/database/es/stats.go  | 6 +++---
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/pkg/database/es/client.go b/pkg/database/es/client.go
index 2dacec5..9b79086 100644
--- a/pkg/database/es/client.go
+++ b/pkg/database/es/client.go
@@ -39,6 +39,8 @@ type Config struct {
 	URLs []string
 	// Plugins enabled plugins
 	Plugins []string
+	// Sniff enabled or disabled sniffer
+	Sniff bool
 }
 
 // Client es client struct
@@ -61,6 +63,7 @@ func NewClient(config *Config) *Client {
 		es7.SetURL(config.URLs...),
 		es7.SetScheme(config.Schema),
 		es7.SetRequiredPlugins(config.Plugins...),
+		es7.SetSniff(config.Sniff),
 	)
 	client := &Client{
 		config: config,
diff --git a/pkg/database/es/stats.go b/pkg/database/es/stats.go
index dd1a7b4..de468f3 100644
--- a/pkg/database/es/stats.go
+++ b/pkg/database/es/stats.go
@@ -80,11 +80,11 @@ func (t *Transport) RoundTrip(req *http.Request) (resp *http.Response, err error
 		} else {
 			resp, err = http.DefaultTransport.RoundTrip(req)
 		}
-		
+
 		if resp != nil {
-		    	span = span.SetTag("http.status_code", resp.StatusCode)
+			span = span.SetTag("http.status_code", resp.StatusCode)
 		}
-		
+
 		if err != nil {
 			ext.Error.Set(span, true)
 			span.LogFields(log.String("event", "error"), log.String("message", err.Error()))