Java RequestSource.REQUEST_BY_USER_FORCED属性代码示例

/**
 * Ensure that we are allowed to manually manage the HA state of the target
 * service. If automatic failover is configured, then the automatic
 * failover controllers should be doing state management, and it is generally
 * an error to use the HAAdmin command line to do so.
 * 
 * @param target the target to check
 * @return true if manual state management is allowed
 */
private boolean checkManualStateManagementOK(HAServiceTarget target) {
  if (target.isAutoFailoverEnabled()) {
    if (requestSource != RequestSource.REQUEST_BY_USER_FORCED) {
      errOut.println(
          "Automatic failover is enabled for " + target + "\n" +
          "Refusing to manually manage HA state, since it may cause\n" +
          "a split-brain scenario or other incorrect state.\n" +
          "If you are very sure you know what you are doing, please \n" +
          "specify the --" + FORCEMANUAL + " flag.");
      return false;
    } else {
      LOG.warn("Proceeding with manual HA state management even though\n" +
          "automatic failover is enabled for " + target);
      return true;
    }
  }
  return true;
}
 

private StateChangeRequestInfo convert(HAStateChangeRequestInfoProto proto) {
  RequestSource src;
  switch (proto.getReqSource()) {
  case REQUEST_BY_USER:
    src = RequestSource.REQUEST_BY_USER;
    break;
  case REQUEST_BY_USER_FORCED:
    src = RequestSource.REQUEST_BY_USER_FORCED;
    break;
  case REQUEST_BY_ZKFC:
    src = RequestSource.REQUEST_BY_ZKFC;
    break;
  default:
    LOG.warn("Unknown request source: " + proto.getReqSource());
    src = null;
  }
  
  return new StateChangeRequestInfo(src);
}
 

/**
 * Ensure that we are allowed to manually manage the HA state of the target
 * service. If automatic failover is configured, then the automatic
 * failover controllers should be doing state management, and it is generally
 * an error to use the HAAdmin command line to do so.
 * 
 * @param target the target to check
 * @return true if manual state management is allowed
 */
private boolean checkManualStateManagementOK(HAServiceTarget target) {
  if (target.isAutoFailoverEnabled()) {
    if (requestSource != RequestSource.REQUEST_BY_USER_FORCED) {
      errOut.println(
          "Automatic failover is enabled for " + target + "\n" +
          "Refusing to manually manage HA state, since it may cause\n" +
          "a split-brain scenario or other incorrect state.\n" +
          "If you are very sure you know what you are doing, please \n" +
          "specify the " + FORCEMANUAL + " flag.");
      return false;
    } else {
      LOG.warn("Proceeding with manual HA state management even though\n" +
          "automatic failover is enabled for " + target);
      return true;
    }
  }
  return true;
}
 

protected int runCmd(String[] argv) throws Exception {
  if (argv.length < 1) {
    printUsage(errOut);
    return -1;
  }

  String cmd = argv[0];

  if (!cmd.startsWith("-")) {
    errOut.println("Bad command '" + cmd + "': expected command starting with '-'");
    printUsage(errOut);
    return -1;
  }
  
  if (!USAGE.containsKey(cmd)) {
    errOut.println(cmd.substring(1) + ": Unknown command");
    printUsage(errOut);
    return -1;
  }
  
  Options opts = new Options();

  // Add command-specific options
  if ("-failover".equals(cmd)) {
    addFailoverCliOpts(opts);
  }
  if("-transitionToActive".equals(cmd)) {
    addTransitionToActiveCliOpts(opts);
  }
  // Mutative commands take FORCEMANUAL option
  if ("-transitionToActive".equals(cmd) ||
      "-transitionToStandby".equals(cmd) ||
      "-failover".equals(cmd)) {
    opts.addOption(FORCEMANUAL, false,
        "force manual control even if auto-failover is enabled");
  }
       
  CommandLine cmdLine = parseOpts(cmd, opts, argv);
  if (cmdLine == null) {
    // error already printed
    return -1;
  }
  
  if (cmdLine.hasOption(FORCEMANUAL)) {
    if (!confirmForceManual()) {
      LOG.fatal("Aborted");
      return -1;
    }
    // Instruct the NNs to honor this request even if they're
    // configured for manual failover.
    requestSource = RequestSource.REQUEST_BY_USER_FORCED;
  }

  if ("-transitionToActive".equals(cmd)) {
    return transitionToActive(cmdLine);
  } else if ("-transitionToStandby".equals(cmd)) {
    return transitionToStandby(cmdLine);
  } else if ("-failover".equals(cmd)) {
    return failover(cmdLine);
  } else if ("-getServiceState".equals(cmd)) {
    return getServiceState(cmdLine);
  } else if ("-checkHealth".equals(cmd)) {
    return checkHealth(cmdLine);
  } else if ("-help".equals(cmd)) {
    return help(argv);
  } else {
    // we already checked command validity above, so getting here
    // would be a coding error
    throw new AssertionError("Should not get here, command: " + cmd);
  } 
}
 

protected int runCmd(String[] argv) throws Exception {
  if (argv.length < 1) {
    printUsage(errOut);
    return -1;
  }

  String cmd = argv[0];

  if (!cmd.startsWith("-")) {
    errOut.println("Bad command '" + cmd + "': expected command starting with '-'");
    printUsage(errOut);
    return -1;
  }
  
  if (!USAGE.containsKey(cmd)) {
    errOut.println(cmd.substring(1) + ": Unknown command");
    printUsage(errOut);
    return -1;
  }
  
  Options opts = new Options();

  // Add command-specific options
  if ("-failover".equals(cmd)) {
    addFailoverCliOpts(opts);
  }
  // Mutative commands take FORCEMANUAL option
  if ("-transitionToActive".equals(cmd) ||
      "-transitionToStandby".equals(cmd) ||
      "-failover".equals(cmd)) {
    opts.addOption(FORCEMANUAL, false,
        "force manual control even if auto-failover is enabled");
  }
       
  CommandLine cmdLine = parseOpts(cmd, opts, argv);
  if (cmdLine == null) {
    // error already printed
    return -1;
  }
  
  if (cmdLine.hasOption(FORCEMANUAL)) {
    if (!confirmForceManual()) {
      LOG.fatal("Aborted");
      return -1;
    }
    // Instruct the NNs to honor this request even if they're
    // configured for manual failover.
    requestSource = RequestSource.REQUEST_BY_USER_FORCED;
  }

  if ("-transitionToActive".equals(cmd)) {
    return transitionToActive(cmdLine);
  } else if ("-transitionToStandby".equals(cmd)) {
    return transitionToStandby(cmdLine);
  } else if ("-failover".equals(cmd)) {
    return failover(cmdLine);
  } else if ("-getServiceState".equals(cmd)) {
    return getServiceState(cmdLine);
  } else if ("-checkHealth".equals(cmd)) {
    return checkHealth(cmdLine);
  } else if ("-help".equals(cmd)) {
    return help(argv);
  } else {
    // we already checked command validity above, so getting here
    // would be a coding error
    throw new AssertionError("Should not get here, command: " + cmd);
  } 
}