From b4782ce6ac878d32ad27af55afe6a762d7d11047 Mon Sep 17 00:00:00 2001
From: Victor Shyba <victor.shyba@gmail.com>
Date: Fri, 13 Aug 2021 18:36:43 -0300
Subject: [PATCH] add --debug to show queries and error details

---
 main.go          | 2 ++
 server/search.go | 7 ++++++-
 server/server.go | 1 +
 3 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index 0754178..ef684c6 100644
--- a/main.go
+++ b/main.go
@@ -49,6 +49,7 @@ func parseArgs(searchRequest *pb.SearchRequest) *server.Args {
 	parser := argparse.NewParser("hub", "hub server and client")
 
 	serveCmd := parser.NewCommand("serve", "start the hub server")
+	debug := parser.Flag("", "debug", &argparse.Options{Required: false, Help: "enable debug logging", Default: false})
 
 	host := parser.String("", "rpchost", &argparse.Options{Required: false, Help: "RPC host", Default: defaultHost})
 	port := parser.String("", "rpcport", &argparse.Options{Required: false, Help: "RPC port", Default: defaultPort})
@@ -79,6 +80,7 @@ func parseArgs(searchRequest *pb.SearchRequest) *server.Args {
 		EsHost:  *esHost,
 		EsPort:  *esPort,
 		EsIndex: *esIndex,
+		Debug:   *debug,
 	}
 
 	if esHost, ok := environment["ELASTIC_HOST"]; ok {
diff --git a/server/search.go b/server/search.go
index 1511110..6487c03 100644
--- a/server/search.go
+++ b/server/search.go
@@ -8,6 +8,7 @@ import (
 	"fmt"
 	"log"
 	"math"
+	"os"
 	"reflect"
 	"strings"
 	"time"
@@ -146,7 +147,11 @@ func (s *Server) Search(ctx context.Context, in *pb.SearchRequest) (*pb.Outputs,
 	var client *elastic.Client = nil
 	if s.EsClient == nil {
 		esUrl := s.Args.EsHost + ":" + s.Args.EsPort
-		tmpClient, err := elastic.NewClient(elastic.SetURL(esUrl), elastic.SetSniff(false))
+		opts := []elastic.ClientOptionFunc{elastic.SetSniff(false), elastic.SetURL(esUrl)}
+		if s.Args.Debug {
+			opts = append(opts, elastic.SetTraceLog(log.New(os.Stderr, "[[ELASTIC]]", 0)))
+		}
+		tmpClient, err := elastic.NewClient(opts...)
 		if err != nil {
 			log.Println(err)
 			return nil, err
diff --git a/server/server.go b/server/server.go
index 248abc5..167d14c 100644
--- a/server/server.go
+++ b/server/server.go
@@ -25,6 +25,7 @@ type Args struct {
 	EsHost  string
 	EsPort  string
 	EsIndex string
+	Debug   bool
 }
 
 /*