Java Command类代码示例

import org.apache.twill.api.Command; //导入依赖的package包/类
@Override
public final ListenableFuture<Set<String>> restartInstances(
  Map<String, ? extends Set<Integer>> runnableToInstanceIds) {
  Map<String, String> runnableToStringInstanceIds =
    Maps.transformEntries(runnableToInstanceIds, new Maps.EntryTransformer<String, Set<Integer>, String>() {
      @Override
      public String transformEntry(String runnableName, Set<Integer> instanceIds) {
        validateInstanceIds(runnableName, instanceIds);
        return GSON.toJson(instanceIds, new TypeToken<Set<Integer>>() {}.getType());
      }
    });
  Command updateStateCommand = Command.Builder.of(Constants.RESTART_RUNNABLES_INSTANCES)
    .addOptions(runnableToStringInstanceIds)
    .build();
  Message message = SystemMessages.updateRunnablesInstances(updateStateCommand);

  return sendMessage(message, runnableToInstanceIds.keySet());
}
 

import org.apache.twill.api.Command; //导入依赖的package包/类
/**
 * Attempts to change the number of running instances.
 *
 * @return {@code true} if the message does requests for changes in number of running instances of a runnable,
 * {@code false} otherwise.
 */
private boolean handleSetInstances(Message message, Runnable completion) {
  if (message.getType() != Message.Type.SYSTEM || message.getScope() != Message.Scope.RUNNABLE) {
    return false;
  }

  Command command = message.getCommand();
  Map<String, String> options = command.getOptions();
  if (!"instances".equals(command.getCommand()) || !options.containsKey("count")) {
    return false;
  }

  final String runnableName = message.getRunnableName();
  if (runnableName == null || runnableName.isEmpty() || !twillSpec.getRunnables().containsKey(runnableName)) {
    LOG.info("Unknown runnable {}", runnableName);
    return false;
  }

  final int newCount = Integer.parseInt(options.get("count"));
  final int oldCount = expectedContainers.getExpected(runnableName);

  LOG.info("Received change instances request for {}, from {} to {}.", runnableName, oldCount, newCount);

  if (newCount == oldCount) {   // Nothing to do, simply complete the request.
    completion.run();
    return true;
  }

  instanceChangeExecutor.execute(createSetInstanceRunnable(message, completion, oldCount, newCount));
  return true;
}
 

import org.apache.twill.api.Command; //导入依赖的package包/类
@Override
public void handleCommand(Command command) throws Exception {
  super.handleCommand(command);
  if (HANGING_RUNNABLE_STOP_SECS.equals(command.getCommand())) {
    int time = Integer.parseInt(command.getOptions().get(HANGING_RUNNABLE_STOP_SECS));
    LOG.info("Got sleep time from message = {} secs", time);
    sleepTime.set(time);
  }
}
 

import org.apache.twill.api.Command; //导入依赖的package包/类
@Test
public void testFailureRestart() throws Exception {
  TwillRunner runner = getTwillRunner();

  ResourceSpecification resource = ResourceSpecification.Builder.with()
    .setVirtualCores(1)
    .setMemory(512, ResourceSpecification.SizeUnit.MEGA)
    .setInstances(2)
    .build();
  TwillController controller = runner.prepare(new FailureRunnable(), resource)
    .withApplicationArguments("failure")
    .withArguments(FailureRunnable.class.getSimpleName(), "failure2")
    .addLogHandler(new PrinterLogHandler(new PrintWriter(System.out, true)))
    .start();

  Iterable<Discoverable> discoverables = controller.discoverService("failure");
  Assert.assertTrue(waitForSize(discoverables, 2, 120));

  // Make sure we see the right instance IDs
  Assert.assertEquals(Sets.newHashSet(0, 1), getInstances(discoverables));

  // Kill server with instanceId = 0
  controller.sendCommand(FailureRunnable.class.getSimpleName(), Command.Builder.of("kill0").build());

  // Make sure the runnable is killed.
  Assert.assertTrue(waitForSize(discoverables, 1, 120));

  // Wait for the restart
  Assert.assertTrue(waitForSize(discoverables, 2, 120));

  // Make sure we see the right instance IDs
  Assert.assertEquals(Sets.newHashSet(0, 1), getInstances(discoverables));

  controller.terminate().get(120, TimeUnit.SECONDS);
}
 

import org.apache.twill.api.Command; //导入依赖的package包/类
@Override
public void handleCommand(Command command) throws Exception {
  if (command.getCommand().equals("kill" + getContext().getInstanceId())) {
    killed = true;
    running = false;
    serverSocket.close();
  }
}
 

import org.apache.twill.api.Command; //导入依赖的package包/类
@Override
public final ListenableFuture<String> restartAllInstances(String runnableName) {
  Command updateStateCommand = Command.Builder.of(Constants.RESTART_ALL_RUNNABLE_INSTANCES).
    build();
  Message message = SystemMessages.updateRunnableInstances(updateStateCommand, runnableName);
  return sendMessage(message, updateStateCommand.getCommand());
}
 

import org.apache.twill.api.Command; //导入依赖的package包/类
/**
 * Helper method to get System {@link Message} for updating the log levels for one or all runnables.
 *
 * @param runnableName The name of the runnable to update the log level, null if apply to all runnables.
 * @param logLevels The log levels to be updated.
 * @return An instance of System {@link Message} to update the log levels.
 */
public static Message updateLogLevels(@Nullable String runnableName, Map<String, LogEntry.Level> logLevels) {
  Preconditions.checkNotNull(logLevels);
  Preconditions.checkArgument(!(logLevels.containsKey(Logger.ROOT_LOGGER_NAME)
                                && logLevels.get(Logger.ROOT_LOGGER_NAME) == null));
  Map<String, String> options = convertLogEntryToString(logLevels);
  return new SimpleMessage(Message.Type.SYSTEM,
                           runnableName == null ? Message.Scope.ALL_RUNNABLE : Message.Scope.RUNNABLE,
                           runnableName, Command.Builder.of(SET_LOG_LEVEL).addOptions(options).build());
}
 

import org.apache.twill.api.Command; //导入依赖的package包/类
/**
* Helper method to get System {@link Message} for resetting log levels for one or all runnables.
*
* @param runnableName The name of the runnable to set the log level, null if apply to all runnables.
* @return An instance of System {@link Message} to reset the log levels.
*/
public static Message resetLogLevels(@Nullable String runnableName, Set<String> loggerNames) {
  return new SimpleMessage(Message.Type.SYSTEM,
                           runnableName == null ? Message.Scope.ALL_RUNNABLE : Message.Scope.RUNNABLE,
                           runnableName, Command.Builder.of(RESET_LOG_LEVEL)
                             .addOptions(Maps.uniqueIndex(loggerNames, Functions.toStringFunction())).build());
}
 

import org.apache.twill.api.Command; //导入依赖的package包/类
@Override
public Message deserialize(JsonElement json, Type typeOfT,
                           JsonDeserializationContext context) throws JsonParseException {
  JsonObject jsonObj = json.getAsJsonObject();

  Message.Type type = Message.Type.valueOf(jsonObj.get("type").getAsString());
  Message.Scope scope = Message.Scope.valueOf(jsonObj.get("scope").getAsString());
  JsonElement name = jsonObj.get("runnableName");
  String runnableName = (name == null || name.isJsonNull()) ? null : name.getAsString();
  Command command = context.deserialize(jsonObj.get("command"), Command.class);

  return new SimpleMessage(type, scope, runnableName, command);
}
 

import org.apache.twill.api.Command; //导入依赖的package包/类
@Override
public JsonElement serialize(Message message, Type typeOfSrc, JsonSerializationContext context) {
  JsonObject jsonObj = new JsonObject();
  jsonObj.addProperty("type", message.getType().name());
  jsonObj.addProperty("scope", message.getScope().name());
  jsonObj.addProperty("runnableName", message.getRunnableName());
  jsonObj.add("command", context.serialize(message.getCommand(), Command.class));

  return jsonObj;
}
 

import org.apache.twill.api.Command; //导入依赖的package包/类
@Test
public void testController() throws ExecutionException, InterruptedException, TimeoutException {
  InMemoryZKServer zkServer = InMemoryZKServer.builder().build();
  zkServer.startAndWait();

  LOG.info("ZKServer: " + zkServer.getConnectionStr());

  try {
    RunId runId = RunIds.generate();
    ZKClientService zkClientService = ZKClientService.Builder.of(zkServer.getConnectionStr()).build();
    zkClientService.startAndWait();

    Service service = createService(zkClientService, runId);
    service.startAndWait();

    TwillController controller = getController(zkClientService, "testController", runId);
    controller.sendCommand(Command.Builder.of("test").build()).get(2, TimeUnit.SECONDS);
    controller.terminate().get(2, TimeUnit.SECONDS);

    final CountDownLatch terminateLatch = new CountDownLatch(1);
    service.addListener(new ServiceListenerAdapter() {
      @Override
      public void terminated(Service.State from) {
        terminateLatch.countDown();
      }
    }, Threads.SAME_THREAD_EXECUTOR);

    Assert.assertTrue(service.state() == Service.State.TERMINATED || terminateLatch.await(2, TimeUnit.SECONDS));

    zkClientService.stopAndWait();

  } finally {
    zkServer.stopAndWait();
  }
}
 

import org.apache.twill.api.Command; //导入依赖的package包/类
/**
 * Attempts to change the number of running instances.
 * @return {@code true} if the message does requests for changes in number of running instances of a runnable,
 *         {@code false} otherwise.
 */
private boolean handleSetInstances(final Message message, final Runnable completion) {
  if (message.getType() != Message.Type.SYSTEM || message.getScope() != Message.Scope.RUNNABLE) {
    return false;
  }

  Command command = message.getCommand();
  Map<String, String> options = command.getOptions();
  if (!"instances".equals(command.getCommand()) || !options.containsKey("count")) {
    return false;
  }

  final String runnableName = message.getRunnableName();
  if (runnableName == null || runnableName.isEmpty() || !twillSpec.getRunnables().containsKey(runnableName)) {
    LOG.info("Unknown runnable {}", runnableName);
    return false;
  }

  final int newCount = Integer.parseInt(options.get("count"));
  final int oldCount = expectedContainers.getExpected(runnableName);

  LOG.info("Received change instances request for {}, from {} to {}.", runnableName, oldCount, newCount);

  if (newCount == oldCount) {   // Nothing to do, simply complete the request.
    completion.run();
    return true;
  }

  instanceChangeExecutor.execute(createSetInstanceRunnable(message, completion, oldCount, newCount));
  return true;
}
 

import org.apache.twill.api.Command; //导入依赖的package包/类
@Override
public ListenableFuture<String> onReceived(final String messageId, final Message message) {
  LOG.debug("Message received: {} {}.", messageId, message);

  if (handleSecureStoreUpdate(message)) {
    return Futures.immediateFuture(messageId);
  }

  final SettableFuture<String> result = SettableFuture.create();
  Command command = message.getCommand();
  if (message.getType() == Message.Type.SYSTEM
    && "instances".equals(command.getCommand()) && command.getOptions().containsKey("count")) {
    context.setInstanceCount(Integer.parseInt(command.getOptions().get("count")));
  }

  commandExecutor.execute(new Runnable() {

    @Override
    public void run() {
      try {
        runnable.handleCommand(message.getCommand());
        result.set(messageId);
      } catch (Exception e) {
        result.setException(e);
      }
    }
  });
  return result;
}
 

import org.apache.twill.api.Command; //导入依赖的package包/类
@Test
public void testFailureRestart() throws Exception {
  TwillRunner runner = YarnTestUtils.getTwillRunner();

  ResourceSpecification resource = ResourceSpecification.Builder.with()
    .setVirtualCores(1)
    .setMemory(512, ResourceSpecification.SizeUnit.MEGA)
    .setInstances(2)
    .build();
  TwillController controller = runner.prepare(new FailureRunnable(), resource)
    .withApplicationArguments("failure")
    .withArguments(FailureRunnable.class.getSimpleName(), "failure2")
    .addLogHandler(new PrinterLogHandler(new PrintWriter(System.out, true)))
    .start();

  Iterable<Discoverable> discoverables = controller.discoverService("failure");
  Assert.assertTrue(YarnTestUtils.waitForSize(discoverables, 2, 120));

  // Make sure we see the right instance IDs
  Assert.assertEquals(Sets.newHashSet(0, 1), getInstances(discoverables));

  // Kill server with instanceId = 0
  controller.sendCommand(FailureRunnable.class.getSimpleName(), Command.Builder.of("kill0").build());

  // Make sure the runnable is killed.
  Assert.assertTrue(YarnTestUtils.waitForSize(discoverables, 1, 120));

  // Wait for the restart
  Assert.assertTrue(YarnTestUtils.waitForSize(discoverables, 2, 120));

  // Make sure we see the right instance IDs
  Assert.assertEquals(Sets.newHashSet(0, 1), getInstances(discoverables));

  controller.stopAndWait();
}
 

import org.apache.twill.api.Command; //导入依赖的package包/类
@Test
public void testController() throws ExecutionException, InterruptedException, TimeoutException {
  InMemoryZKServer zkServer = InMemoryZKServer.builder().build();
  zkServer.startAndWait();

  LOG.info("ZKServer: " + zkServer.getConnectionStr());

  try {
    RunId runId = RunIds.generate();
    ZKClientService zkClientService = ZKClientService.Builder.of(zkServer.getConnectionStr()).build();
    zkClientService.startAndWait();

    Service service = createService(zkClientService, runId);
    service.startAndWait();

    TwillController controller = getController(zkClientService, runId);
    controller.sendCommand(Command.Builder.of("test").build()).get(2, TimeUnit.SECONDS);
    controller.stop().get(2, TimeUnit.SECONDS);

    Assert.assertEquals(ServiceController.State.TERMINATED, controller.state());

    final CountDownLatch terminateLatch = new CountDownLatch(1);
    service.addListener(new ServiceListenerAdapter() {
      @Override
      public void terminated(Service.State from) {
        terminateLatch.countDown();
      }
    }, Threads.SAME_THREAD_EXECUTOR);

    Assert.assertTrue(service.state() == Service.State.TERMINATED || terminateLatch.await(2, TimeUnit.SECONDS));

    zkClientService.stopAndWait();

  } finally {
    zkServer.stopAndWait();
  }
}