add sniff switch to es client

UnderTreeTech · Sep 14, 2021 · 9eb04ca · 9eb04ca
1 parent a7da9d5
commit 9eb04ca
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git 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
@@ -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()))