private Protos.TaskInfo buildCommandExecutorTaskInfo(final TaskContext taskContext, final CloudJobConfiguration jobConfig, final ShardingContexts shardingContexts, final Protos.Offer offer, final Protos.CommandInfo command) { Protos.TaskInfo.Builder result = Protos.TaskInfo.newBuilder().setTaskId(Protos.TaskID.newBuilder().setValue(taskContext.getId()).build()) .setName(taskContext.getTaskName()).setSlaveId(offer.getSlaveId()) .addResources(buildResource("cpus", jobConfig.getCpuCount(), offer.getResourcesList())) .addResources(buildResource("mem", jobConfig.getMemoryMB(), offer.getResourcesList())) .setData(ByteString.copyFrom(new TaskInfoData(shardingContexts, jobConfig).serialize())); return result.setCommand(command).build(); }
@Override public BackupSnapshotTask update(Protos.Offer offer) { return new BackupSnapshotTask(getBuilder() .setSlaveId(offer.getSlaveId()) .setData(getData().withHostname(offer.getHostname()).getBytes()) .build()); }
@Override public CassandraDaemonTask updateId() { return new CassandraDaemonTask(getBuilder() .setTaskId(createId(getName())) .setExecutor(getExecutor().clearId().getExecutorInfo()) .setData(getData() .withState(Protos.TaskState.TASK_STAGING).getBytes()) .build()); }
@Override public RepairTask update(Protos.Offer offer) { return new RepairTask(getBuilder() .setSlaveId(offer.getSlaveId()) .setData(getData().withHostname(offer.getHostname()).getBytes()) .build()); }
@Override public UpgradeSSTableTask update(Protos.Offer offer) { return new UpgradeSSTableTask(getBuilder() .setSlaveId(offer.getSlaveId()) .setData(getData().withHostname(offer.getHostname()).getBytes()) .build()); }
@Override public BackupUploadTask update(Protos.TaskState state) { return new BackupUploadTask(getBuilder().setData( getData().withState(state).getBytes()).build()); }
@Override public CassandraDaemonTask update(Protos.Offer offer) { return new CassandraDaemonTask( getBuilder() .setData(getData().withHostname(offer.getHostname()).getBytes()) .setSlaveId(offer.getSlaveId()) .build()); }
@Override public RestoreSnapshotTask update(Protos.TaskState state) { return new RestoreSnapshotTask(getBuilder().setData( getData().withState(state).getBytes()).build()); }
@Override public BackupSchemaTask update(Protos.TaskState state) { return new BackupSchemaTask(getBuilder().setData( getData().withState(state).getBytes()).build()); }
@Override public BackupSnapshotTask update(Protos.TaskState state) { return new BackupSnapshotTask(getBuilder().setData( getData().withState(state).getBytes()).build()); }
@Override public DownloadSnapshotTask update(Protos.Offer offer) { return new DownloadSnapshotTask(getBuilder() .setSlaveId(offer.getSlaveId()) .setData(getData().withHostname(offer.getHostname()).getBytes()) .build()); }
@Override public RepairTask update(Protos.TaskState state) { return new RepairTask(getBuilder().setData( getData().withState(state).getBytes()).build()); }
@Override public UpgradeSSTableTask update(Protos.TaskState state) { return new UpgradeSSTableTask(getBuilder().setData( getData().withState(state).getBytes()).build()); }
@Override public RestoreSnapshotTask update(Protos.Offer offer) { return new RestoreSnapshotTask(getBuilder() .setSlaveId(offer.getSlaveId()) .setData(getData().withHostname(offer.getHostname()).getBytes()) .build()); }
@Override public BackupSchemaTask update(Protos.Offer offer) { return new BackupSchemaTask(getBuilder() .setSlaveId(offer.getSlaveId()) .setData(getData().withHostname(offer.getHostname()).getBytes()) .build()); }
@Override public BackupUploadTask update(Protos.Offer offer) { return new BackupUploadTask(getBuilder() .setSlaveId(offer.getSlaveId()) .setData(getData().withHostname(offer.getHostname()).getBytes()) .build()); }
@Override public RestoreSchemaTask update(Protos.Offer offer) { return new RestoreSchemaTask(getBuilder() .setSlaveId(offer.getSlaveId()) .setData(getData().withHostname(offer.getHostname()).getBytes()) .build()); }
@Override public CleanupTask update(Protos.TaskState state) { return new CleanupTask(getBuilder().setData( getData().withState(state).getBytes()).build()); }
@Override public CassandraDaemonTask update(Protos.TaskState state) { return new CassandraDaemonTask( getBuilder() .setData(getData() .withState(state) .getBytes()).build()); }
@Override public DownloadSnapshotTask update(Protos.TaskState state) { return new DownloadSnapshotTask(getBuilder().setData( getData().withState(state).getBytes()).build()); }