me.dinowernli.grpc.polyglot.Main Maven / Gradle / Ivy

Go to download
package me.dinowernli.grpc.polyglot;

import java.util.logging.LogManager;

import com.google.protobuf.DescriptorProtos.FileDescriptorSet;
import me.dinowernli.grpc.polyglot.command.ServiceCall;
import me.dinowernli.grpc.polyglot.command.ServiceList;
import me.dinowernli.grpc.polyglot.config.CommandLineArgs;
import me.dinowernli.grpc.polyglot.config.ConfigurationLoader;
import me.dinowernli.grpc.polyglot.io.Output;
import me.dinowernli.grpc.polyglot.protobuf.ProtocInvoker;
import me.dinowernli.grpc.polyglot.protobuf.ProtocInvoker.ProtocInvocationException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
//import org.slf4j.bridge.SLF4JBridgeHandler;
import polyglot.ConfigProto.Configuration;
import polyglot.ConfigProto.ProtoConfiguration;

public class Main {
  private static final Logger logger = LoggerFactory.getLogger(Main.class);
  private static final String VERSION = "2.0.0+dev";

  public static void main(String[] args) {
    // Fix the logging setup.
    //setupJavaUtilLogging();

    logger.info("Polyglot version: " + VERSION);

    final CommandLineArgs arguments;
    try {
      arguments = CommandLineArgs.parse(args);
    } catch (Throwable t) {
      throw new RuntimeException("Unable to parse command line flags", t);
    }

    // Catch the help case.
    if (arguments.isHelp()) {
      logger.info(CommandLineArgs.getUsage());
      return;
    }

    // Check for command
    String command = arguments.command().orElseThrow(() -> new RuntimeException("Missing command"));

    final ConfigurationLoader configLoader = arguments.configSetPath()
      .map(ConfigurationLoader::forFile).orElseGet(() -> ConfigurationLoader.forDefaultConfigSet())
      .withOverrides(arguments);
    Configuration config = arguments.configName()
      .map(configLoader::getNamedConfiguration).orElseGet(() -> configLoader.getDefaultConfiguration());
    logger.info("Loaded configuration: " + config.getName());

    try(Output commandLineOutput = Output.forConfiguration(config.getOutputConfig())) {
      switch (command) {
        case CommandLineArgs.LIST_SERVICES_COMMAND:
          FileDescriptorSet fileDescriptorSet = getFileDescriptorSet(config.getProtoConfig());
          ServiceList.listServices(
              commandLineOutput,
              fileDescriptorSet, config.getProtoConfig().getProtoDiscoveryRoot(),
              arguments.serviceFilter(), arguments.methodFilter(), arguments.withMessage());
          break;

        case CommandLineArgs.CALL_COMMAND:
          ServiceCall.callEndpoint(
              commandLineOutput,
              config.getProtoConfig(),
              arguments.endpoint(),
              arguments.fullMethod(),
              arguments.protoDiscoveryRoot(),
              arguments.configSetPath(),
              arguments.additionalProtocIncludes(),
              config.getCallConfig());
          break;

        default:
          throw new RuntimeException("Unknown command: " + arguments.command().get());
      }
    } catch (Throwable t) {
      logger.warn("Caught top-level exception during command execution", t);
      throw new RuntimeException(t);
    }
  }

  /** Invokes protoc and returns a {@link FileDescriptorSet} used for discovery. */
  private static FileDescriptorSet getFileDescriptorSet(ProtoConfiguration protoConfig) {
    try {
      return ProtocInvoker.forConfig(protoConfig).invoke();
    } catch (ProtocInvocationException e) {
      throw new RuntimeException("Failed to invoke the protoc binary", e);
    }
  }

  /** Redirects the output of standard java loggers to our slf4j handler. 
  private static void setupJavaUtilLogging() {
    LogManager.getLogManager().reset();
    SLF4JBridgeHandler.removeHandlersForRootLogger();
    SLF4JBridgeHandler.install();
  }*/
}