Exemplo n.º 1
0
  private static void iterateCommands(CommandManager mgr) throws IOException {
    Map<String, CommandTarget> commandMap = mgr.getCommands();
    StringBuilder serviceString = new StringBuilder("include \"");
    serviceString.append(MGMT_STRUCT_THRIFT_FILE).append("\"\n");
    serviceString.append("include \"");
    serviceString.append(MGMT_CMDRESULT_THRIFT_FILE).append("\"\n");
    serviceString.append("namespace java ").append(MGMT_SERVICE_PACKAGE).append("\n");
    serviceString.append(EXCEPTION_DECLARATION);
    serviceString.append("\nservice GeodeCommandService {\n");

    StringBuilder structString = new StringBuilder();
    structString.append("namespace java ").append(MGMT_STRUCT_PACKAGE).append("\n");

    int size = commandMap.size();
    for (String command : commandMap.keySet()) {
      CommandTarget tgr = commandMap.get(command);
      boolean argsStructRequired = argsStructRequired(tgr);
      boolean commandSupported = commandSupported(tgr);
      if (commandSupported) {
        if (argsStructRequired) {
          boolean commandProcessed = processCommand(tgr, structString);
          if (commandProcessed) {
            serviceString
                .append("\n\tcommandResult.CommandResult ")
                .append(getProperName(tgr.getCommandName(), METHOD_NAME));
            serviceString
                .append("(")
                .append("1:structs.")
                .append(getProperName(tgr.getCommandName(), STRUCT_NAME));
            serviceString.append(" arguments)");
            serviceString.append(EXCEPTION_METHOD_SIGNATURE);
            if (size > 1) serviceString.append(",");
          }
        } else {
          serviceString
              .append("\n\tcommandResult.CommandResult ")
              .append(getProperName(tgr.getCommandName(), METHOD_NAME));
          serviceString.append("()");
          serviceString.append(EXCEPTION_METHOD_SIGNATURE);
          if (size > 1) serviceString.append(",");
        }
      }
      size--;
    }
    serviceString.append("\n}");
    writeToFile(structString, MGMT_STRUCT_THRIFT_FILE);
    writeToFile(serviceString, MGMT_SERVICE_THRIFT_FILE);
  }
Exemplo n.º 2
0
 /**
  * Some commands can not be supported over wire but only work thorugh shell. Add them here.
  *
  * @param tgr
  * @return
  */
 private static boolean commandSupported(CommandTarget tgr) {
   if (tgr.getCommandName().contains("connect")
       || tgr.getCommandName().startsWith("start")
       || tgr.getCommandName().startsWith("exit")
       || tgr.getCommandName().startsWith("disconnect")
       || tgr.getCommandName().startsWith("debug")
       || tgr.getCommandName()
           .startsWith("deploy") // deploy need byte[] manual changes. will be supported later
       || tgr.getCommandName().startsWith("describe client")
       || tgr.getCommandName().startsWith("echo")
       || tgr.getCommandName().startsWith("import") // import command have file structures again
       || tgr.getCommandName().equals("sh")
       || tgr.getCommandName().equals("sleep")) return false;
   else return true;
 }
Exemplo n.º 3
0
  private static boolean processCommand(CommandTarget commandTarget, StringBuilder structFile) {

    StringBuilder structString = new StringBuilder();
    try {
      int paramNumber = 1;
      String command = commandTarget.getCommandName();
      List<Argument> arguments = commandTarget.getOptionParser().getArguments();
      List<Option> options = commandTarget.getOptionParser().getOptions();

      structString.append("\nstruct " + getProperName(command, STRUCT_NAME) + " {\n");

      // This is to ensure at least one non-optional fields
      structString
          .append("\t")
          .append(paramNumber++)
          .append(":")
          .append("i16 argsId=")
          .append(argsCounter.incrementAndGet())
          .append(",\n");

      for (int i = 0; i < arguments.size(); i++) {
        Argument argument = arguments.get(i);
        structString
            .append("\t")
            .append(paramNumber++)
            .append(":")
            .append(" string ")
            .append(getProperName(argument.getArgumentName(), STRUCT_FIELD_NAME));
        if (options.size() > 0) structString.append(",\n");
        else structString.append("\n");
      }

      for (int i = 0; i < options.size(); i++) {
        Option option = options.get(i);
        String thriftType = getThriftDataType(option.getDataType());
        String thriftName = getProperName(option.getLongOption(), STRUCT_FIELD_NAME);
        structString
            .append("\t")
            .append(paramNumber++)
            .append(":")
            .append((option.isRequired() ? "" : "optional "))
            .append(thriftType)
            .append(" ")
            .append(thriftName);

        String defaultValue = option.getUnspecifiedDefaultValue();
        if (defaultValue != null && isNumericStringOrBoolean(option.getDataType())) {
          // Check for numeric or string and boolean dont allow collections
          structString
              .append(" = ")
              .append(quoteStringOptionValue(defaultValue, option.getDataType()));
        }

        if (i < options.size() - 1) structString.append(",\n");
        else structString.append("\n");

        allKindsType.add(option.getDataType().getSimpleName());
      }
      structString.append("\n}\n");
      structFile.append(structString.toString());
      return true;
    } catch (RuntimeException e) {
      System.out.println(
          "Command " + commandTarget.getCommandName() + " not supported bcoz " + e.getMessage());
      e.printStackTrace();
      return false;
    }
  }
Exemplo n.º 4
0
 private static boolean argsStructRequired(CommandTarget tgr) {
   if ((tgr.getOptionParser().getArguments().size() + tgr.getOptionParser().getOptions().size())
       > 0) return true;
   else return false;
 }