com.github.tonivade.claudb.Client Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2015-2023, Antonio Gabriel Muñoz Conejo 
 * Distributed under the terms of the MIT License
 */
package com.github.tonivade.claudb;

import static com.github.tonivade.resp.protocol.RedisToken.status;

import java.util.Scanner;
import java.util.concurrent.ArrayBlockingQueue;
import java.util.concurrent.BlockingQueue;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.github.tonivade.resp.RespCallback;
import com.github.tonivade.resp.RespClient;
import com.github.tonivade.resp.protocol.RedisToken;

import joptsimple.BuiltinHelpFormatter;
import joptsimple.OptionException;
import joptsimple.OptionParser;
import joptsimple.OptionSet;
import joptsimple.OptionSpec;

public class Client implements RespCallback {

  private static final Logger LOGGER = LoggerFactory.getLogger(Client.class);

  private static final String CHARSET_NAME = "UTF-8";
  private static final String QUIT = "quit";
  private static final String PROMPT = "> ";

  private final BlockingQueue responses = new ArrayBlockingQueue<>(1);

  @Override
  public void onConnect() {
    responses.add(status("connected!"));
  }

  @Override
  public void onDisconnect() {
    responses.add(status("disconnected!"));
  }

  @Override
  public void onMessage(RedisToken token) {
    try {
      responses.put(token);
    } catch (InterruptedException e) {
      LOGGER.warn("message not processed", e);
    }
  }

  public RedisToken response() throws InterruptedException {
    return responses.take();
  }

  public static void main(String[] args) throws Exception {
    OptionParser parser = new OptionParser();
    parser.formatHelpWith(new BuiltinHelpFormatter(100, 2));
    OptionSpec help = parser.accepts("help", "print help");
    OptionSpec host = parser.accepts("h", "server name")
        .withRequiredArg()
        .defaultsTo(DBServerContext.DEFAULT_HOST)
        .describedAs("name");
    OptionSpec port = parser.accepts("p", "server port")
        .withRequiredArg()
        .ofType(Integer.class)
        .defaultsTo(DBServerContext.DEFAULT_PORT)
        .describedAs("port");

    OptionSet options;
    try {
      options = parser.parse(args);
    } catch (OptionException e) {
      System.out.println("ERROR: " + e.getMessage());
      System.out.println();
      options = parser.parse("--help");
    }

    if (options.has(help)) {
      parser.printHelpOn(System.out);
    } else {
      Client callback = new Client();

      String optionHost = options.valueOf(host);
      int optionPort = options.valueOf(port);
      RespClient client = new RespClient(optionHost, optionPort, callback);
      client.start();
      
      System.out.println(callback.response());
      prompt();
      try (Scanner scanner = new Scanner(System.in, CHARSET_NAME)) {
        for (boolean quit = false; !quit && scanner.hasNextLine(); prompt()) {
          String line = scanner.nextLine();
          if (!line.isEmpty()) {
            client.send(line.split(" "));
            System.out.println(callback.response());
            quit = line.equalsIgnoreCase(QUIT);
          }
        }
      } finally {
        client.stop();
      }
    }
  }

  private static void prompt() {
    System.out.print(PROMPT);
  }
}