com.yelp.nrtsearch.tools.cli.NrtsearchClientCommand Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of server Show documentation
GRPC lucene server using near-real-time replication
The newest version!
/*
 * Copyright 2020 Yelp Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.yelp.nrtsearch.tools.cli;

import com.yelp.nrtsearch.server.Version;
import com.yelp.nrtsearch.server.grpc.NrtsearchClient;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import picocli.CommandLine;

/*Each new command needs to be added here*/
@CommandLine.Command(
    name = "nrtsearch_client",
    synopsisSubcommandLabel = "COMMAND",
    subcommands = {
      AddDocumentsCommand.class,
      BackupWarmingQueriesCommand.class,
      CommitCommand.class,
      CreateIndexCommand.class,
      GetCurrentSearcherVersion.class,
      CustomCommand.class,
      DeleteByQueryCommand.class,
      DeleteDocumentsCommand.class,
      DeleteAllDocumentsCommand.class,
      DeleteIndexCommand.class,
      ForceMergeCommand.class,
      ForceMergeDeletesCommand.class,
      GlobalStateCommand.class,
      IndexStateCommand.class,
      IndicesCommand.class,
      LiveSettingsV2Command.class,
      MetricsCommand.class,
      NodeInfoCommand.class,
      ReadyCommand.class,
      RefreshCommand.class,
      RegisterFieldsCommand.class,
      ReloadStateCommand.class,
      SearchCommand.class,
      SettingsV2Command.class,
      StartIndexCommand.class,
      StartIndexV2Command.class,
      StatsCommand.class,
      StatusCommand.class,
      StopIndexCommand.class,
      WriteNRTPointCommand.class,
      CommandLine.HelpCommand.class
    })
public class NrtsearchClientCommand implements Runnable {
  public static final Logger logger =
      LoggerFactory.getLogger(NrtsearchClientCommand.class.getName());

  @CommandLine.Option(
      names = {"-p", "--port"},
      description = "Port number of server to connect to (default: ${DEFAULT-VALUE})",
      defaultValue = "6000")
  private String port;

  public int getPort() {
    return Integer.parseInt(port);
  }

  @CommandLine.Option(
      names = {"-h", "--hostname", "--host"},
      description = "Host name of server to connect to (default: ${DEFAULT-VALUE})",
      defaultValue = "localhost")
  private String hostname;

  @CommandLine.Option(
      names = {"-V", "--version"},
      description = "Print version information and exit")
  private boolean printVersion;

  public String getHostname() {
    return hostname;
  }

  public NrtsearchClient getClient() {
    return new NrtsearchClient(getHostname(), getPort());
  }

  public static void main(String[] args) {
    System.exit(new CommandLine(new NrtsearchClientCommand()).execute(args));
  }

  @Override
  public void run() {
    if (printVersion) {
      System.out.println(Version.CURRENT);
    } else {
      // if only the base command is run, just print the usage
      new CommandLine(this).execute("help");
    }
  }
}