public RestoreSnapshotStep( String daemon, CassandraState cassandraState, CassandraOfferRequirementProvider provider, BackupRestoreContext context) { super(daemon, RestoreSnapshotTask.nameForDaemon(daemon), cassandraState, provider); this.context = context; }
/** * Gets the name of a RestoreSnapshotTask for a CassandraDaemonTask. * * @param daemon The CassandraDaemonTask for which the snapshot will be * uploaded. * @return The name of the RestoreSnapshotTask for daemon. */ public static final String nameForDaemon(final CassandraDaemonTask daemon) { return nameForDaemon(daemon.getName()); }
public RestoreSnapshotTask getOrCreateRestoreSnapshot( CassandraDaemonTask daemon, BackupRestoreContext context) throws PersistenceException { String name = RestoreSnapshotTask.nameForDaemon(daemon); Map<String, RestoreSnapshotTask> snapshots = getRestoreSnapshotTasks(); if (snapshots.containsKey(name)) { return snapshots.get(name); } else { return createRestoreSnapshotTask(daemon, context); } }
public static RestoreSnapshotTask create( final Protos.TaskInfo template, final CassandraDaemonTask daemon, final BackupRestoreContext context) { CassandraData data = CassandraData.createRestoreSnapshotData( "", context .forNode(daemon.getName()) .withLocalLocation(daemon.getVolumePath() + "/data")); String name = nameForDaemon(daemon); Protos.TaskInfo completedTemplate = Protos.TaskInfo.newBuilder(template) .setName(name) .setTaskId(TaskUtils.toTaskId(name)) .setData(data.getBytes()) .build(); completedTemplate = org.apache.mesos.offer.TaskUtils.clearTransient(completedTemplate); return new RestoreSnapshotTask(completedTemplate); }