diff --git a/main.go b/main.go index 6688bee..9ba1080 100644 --- a/main.go +++ b/main.go @@ -18,13 +18,13 @@ import ( ) const ( - defaultHost = "0.0.0.0" - defaultPort = "50051" - defaultEsHost = "http://localhost" - defaultEsPort = "9200" + defaultHost = "0.0.0.0" + defaultPort = "50051" + defaultEsHost = "http://localhost" + defaultEsIndex = "claims" + defaultEsPort = "9200" ) - func GetEnvironment(data []string, getkeyval func(item string) (key, val string)) map[string]string { items := make(map[string]string) for _, item := range data { @@ -53,8 +53,8 @@ func parseArgs(searchRequest *pb.SearchRequest) *server.Args { host := parser.String("", "rpchost", &argparse.Options{Required: false, Help: "host", Default: defaultHost}) port := parser.String("", "rpcport", &argparse.Options{Required: false, Help: "port", Default: defaultPort}) esHost := parser.String("", "eshost", &argparse.Options{Required: false, Help: "host", Default: defaultEsHost}) + esIndex := parser.String("", "esindex", &argparse.Options{Required: false, Help: "host", Default: defaultEsIndex}) esPort := parser.String("", "esport", &argparse.Options{Required: false, Help: "port", Default: defaultEsPort}) - dev := parser.Flag("", "dev", &argparse.Options{Required: false, Help: "port", Default: false}) text := parser.String("", "text", &argparse.Options{Required: false, Help: "text query"}) name := parser.String("", "name", &argparse.Options{Required: false, Help: "name"}) @@ -72,14 +72,13 @@ func parseArgs(searchRequest *pb.SearchRequest) *server.Args { log.Fatalln(parser.Usage(err)) } - args := &server.Args{ - Serve: false, - Host: *host, - Port: ":" + *port, - EsHost: *esHost, - EsPort: *esPort, - Dev: *dev, + Serve: false, + Host: *host, + Port: ":" + *port, + EsHost: *esHost, + EsPort: *esPort, + EsIndex: *esIndex, } if esHost, ok := environment["ELASTIC_HOST"]; ok { @@ -95,8 +94,8 @@ func parseArgs(searchRequest *pb.SearchRequest) *server.Args { } /* - Verify no invalid argument combinations - */ + Verify no invalid argument combinations + */ if len(*channelIds) > 0 && *channelId != "" { log.Fatal("Cannot specify both channel_id and channel_ids") } @@ -108,7 +107,7 @@ func parseArgs(searchRequest *pb.SearchRequest) *server.Args { if *text != "" { searchRequest.Text = *text } - if *name!= "" { + if *name != "" { searchRequest.Name = []string{*name} } if *claimType != "" { @@ -174,7 +173,6 @@ func main() { ctx, cancel := context.WithTimeout(context.Background(), time.Second) defer cancel() - r, err := c.Search(ctx, searchRequest) if err != nil { log.Fatal(err) diff --git a/server/search.go b/server/search.go index 5d8ef22..ad97080 100644 --- a/server/search.go +++ b/server/search.go @@ -153,27 +153,13 @@ func (s *Server) Search(ctx context.Context, in *pb.SearchRequest) (*pb.Outputs, var pageSize = 10 var orderBy []orderField var searchIndices = []string{} + searchIndices = make([]string, 0, 1) + searchIndices = append(searchIndices, s.Args.EsIndex) q := elastic.NewBoolQuery() q = s.setupEsQuery(q, in, &pageSize, &from, &orderBy) - if s.Args.Dev && len(in.SearchIndices) == 0 { - // If we're running in dev mode ignore the mainnet claims index - indices, err := client.IndexNames() - if err != nil { - log.Fatalln(err) - } - var numIndices = len(indices) - searchIndices = make([]string, 0, numIndices) - for i := 0; i < numIndices; i++ { - if indices[i] == "claims" { - continue - } - searchIndices = append(searchIndices, indices[i]) - } - } - if len(in.SearchIndices) > 0 { searchIndices = in.SearchIndices } @@ -185,7 +171,6 @@ func (s *Server) Search(ctx context.Context, in *pb.SearchRequest) (*pb.Outputs, Query(q). // specify the query From(0).Size(DefaultSearchSize) - for _, x := range orderBy { search = search.Sort(x.Field, x.IsAsc) } diff --git a/server/server.go b/server/server.go index 7d50181..248abc5 100644 --- a/server/server.go +++ b/server/server.go @@ -11,7 +11,7 @@ import ( type Server struct { GrpcServer *grpc.Server - Args *Args + Args *Args MultiSpaceRe *regexp.Regexp WeirdCharsRe *regexp.Regexp EsClient *elastic.Client @@ -19,12 +19,12 @@ type Server struct { } type Args struct { - Serve bool - Host string - Port string - EsHost string - EsPort string - Dev bool + Serve bool + Host string + Port string + EsHost string + EsPort string + EsIndex string } /* @@ -79,12 +79,12 @@ func MakeHubServer(args *Args) *Server { log.Fatal(err) } - s := &Server { - GrpcServer: grpcServer, - Args: args, + s := &Server{ + GrpcServer: grpcServer, + Args: args, MultiSpaceRe: multiSpaceRe, WeirdCharsRe: weirdCharsRe, } return s -} \ No newline at end of file +}