@Override public DownloadSnapshotStatus createStatus( Protos.TaskState state, Optional<String> message) { Protos.TaskStatus.Builder builder = getStatusBuilder(); if (message.isPresent()) { builder.setMessage(message.get()); } return DownloadSnapshotStatus.create(builder .setData(CassandraData.createSnapshotDownloadStatusData().getBytes()) .setState(state) .build()); }
@Override public CassandraTaskStatus createStatus(Protos.TaskState state, Optional<String> message) { Protos.TaskStatus.Builder builder = getStatusBuilder(); if (message.isPresent()) { builder.setMessage(message.get()); } return TemplateTaskStatus.create(builder .setData(CassandraData.createTemplateData().getBytes()) .setState(state) .build()); }
@Override public RestoreSnapshotStatus createStatus( Protos.TaskState state, Optional<String> message) { Protos.TaskStatus.Builder builder = getStatusBuilder(); if (message.isPresent()) { builder.setMessage(message.get()); } return RestoreSnapshotStatus.create(builder .setData(CassandraData.createRestoreSnapshotStatusData().getBytes()) .setState(state) .build()); }
if (taskStatusOptional.get().hasData()) { getStateStore().storeStatus(Protos.TaskStatus.newBuilder(status) .setData(taskStatusOptional.get().getData()) .build()); } else {
private void onStop() { // Shutdown REEF Evaluator if (this.evaluatorProcess != null) { this.evaluatorProcess.destroy(); mesosExecutorDriver.sendStatusUpdate(TaskStatus.newBuilder() .setTaskId(TaskID.newBuilder() .setValue(mesosExecutorId) .build()) .setState(TaskState.TASK_FINISHED) .setMessage("Evaluator Process exited with status " + String.valueOf(evaluatorProcessExitValue)) .build()); } else { mesosExecutorDriver.sendStatusUpdate(TaskStatus.newBuilder() .setTaskId(TaskID.newBuilder() .setValue(mesosExecutorId) .build()) .setState(TaskState.TASK_FINISHED) .setData(ByteString.copyFromUtf8("eval_not_run")) // TODO[JIRA REEF-102]: a hack to pass closeEvaluator test, replace this with a better interface .setMessage("Evaluator Process exited with status " + String.valueOf(evaluatorProcessExitValue)) .build()); } // Shutdown Mesos Executor this.executorService.shutdown(); this.mesosExecutorDriver.stop(); }
/** * Gets the task's status. * * @return The status associated with the task. */ public Protos.TaskStatus.Builder getStatusBuilder( final Protos.TaskState state, final Optional<String> message ) { Protos.TaskStatus.Builder builder = Protos.TaskStatus.newBuilder() .setSlaveId(info.getSlaveId()) .setTaskId(info.getTaskId()) .setState(state) .setData(getData().withState(state).getBytes()) .setExecutorId(info.getExecutor().getExecutorId()) .setSource(Protos.TaskStatus.Source.SOURCE_EXECUTOR); message.map(builder::setMessage); return builder; }
@Override public RepairStatus createStatus( Protos.TaskState state, Optional<String> message) { Protos.TaskStatus.Builder builder = getStatusBuilder(); if (message.isPresent()) { builder.setMessage(message.get()); } return RepairStatus.create(builder .setData(CassandraData.createRepairStatusData().getBytes()) .setState(state) .build()); }
@Override public CleanupStatus createStatus( Protos.TaskState state, Optional<String> message) { Protos.TaskStatus.Builder builder = getStatusBuilder(); if (message.isPresent()) { builder.setMessage(message.get()); } return CleanupStatus.create(builder .setData(CassandraData.createCleanupStatusData().getBytes()) .setState(state) .build()); }
@Override public RestoreSchemaStatus createStatus( Protos.TaskState state, Optional<String> message) { Protos.TaskStatus.Builder builder = getStatusBuilder(); if (message.isPresent()) { builder.setMessage(message.get()); } return RestoreSchemaStatus.create(builder .setData(CassandraData.createRestoreSchemaStatusData().getBytes()) .setState(state) .build()); }
@Override public BackupSnapshotStatus createStatus( Protos.TaskState state, Optional<String> message) { Protos.TaskStatus.Builder builder = getStatusBuilder(); if (message.isPresent()) { builder.setMessage(message.get()); } return BackupSnapshotStatus.create(builder .setData(CassandraData.createBackupSnapshotStatusData().getBytes()) .setState(state) .build()); }
@Override public UpgradeSSTableStatus createStatus( Protos.TaskState state, Optional<String> message) { Protos.TaskStatus.Builder builder = getStatusBuilder(); if (message.isPresent()) { builder.setMessage(message.get()); } return UpgradeSSTableStatus.create(builder .setData(CassandraData.createUpgradeSSTableStatusData().getBytes()) .setState(state) .build()); }
@Override public BackupUploadStatus createStatus( Protos.TaskState state, Optional<String> message) { Protos.TaskStatus.Builder builder = getStatusBuilder(); if (message.isPresent()) { builder.setMessage(message.get()); } return BackupUploadStatus.create(builder .setData(CassandraData.createBackupUploadStatusData().getBytes()) .setState(state) .build()); }
@Override public BackupSchemaStatus createStatus( Protos.TaskState state, Optional<String> message) { Protos.TaskStatus.Builder builder = getStatusBuilder(); if (message.isPresent()) { builder.setMessage(message.get()); } return BackupSchemaStatus.create(builder .setData(CassandraData.createBackupSchemaStatusData().getBytes()) .setState(state) .build()); }
public Protos.TaskStatus getCurrentStatus() { return getStatusBuilder() .setTaskId(info.getTaskId()) .setData(getData().getBytes()) .setState(getData() .getState()).build(); }
public CassandraDaemonStatus createStatus(Protos.TaskState state, CassandraMode mode, Optional<String> message) { return CassandraDaemonStatus.create(getStatusBuilder(state, message) .setData(CassandraData.createDaemonStatusData(mode).getBytes()) .build()); }
public static Protos.TaskStatus generateStatus( Protos.TaskID taskID, Protos.TaskState taskState, CassandraMode cassandraMode) { return Protos.TaskStatus.newBuilder() .setTaskId(taskID) .setState(taskState) .setData(CassandraData.createDaemonStatusData(cassandraMode).getBytes()) .build(); } }