/*
* Copyright (c) 2015 EMC Corporation
* All Rights Reserved
*/
package com.emc.storageos.volumecontroller.impl.block.taskcompleter;
import java.net.URI;
import java.util.List;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.emc.storageos.db.client.DbClient;
import com.emc.storageos.db.client.URIUtil;
import com.emc.storageos.db.client.model.BlockObject;
import com.emc.storageos.db.client.model.BlockSnapshot;
import com.emc.storageos.db.client.model.Operation;
import com.emc.storageos.db.client.model.Volume;
import com.emc.storageos.exceptions.DeviceControllerException;
import com.emc.storageos.svcs.errorhandling.model.ServiceCoded;
public class CloneTaskCompleter extends VolumeTaskCompleter {
private static final long serialVersionUID = -5808737150117378443L;
private static final Logger log = LoggerFactory.getLogger(CloneTaskCompleter.class);
public CloneTaskCompleter(List<URI> fullCopyVolumeURIs, String task) {
super(Volume.class, fullCopyVolumeURIs, task);
setNotifyWorkflow(true);
}
public CloneTaskCompleter(URI fullCopyVolumeURI, String task) {
super(Volume.class, fullCopyVolumeURI, task);
setNotifyWorkflow(true);
}
@Override
protected void complete(DbClient dbClient, Operation.Status status, ServiceCoded coded)
throws DeviceControllerException {
log.info("START FullCopyVolumeTaskCompleter complete");
super.complete(dbClient, status, coded);
try {
List<Volume> fullCopyvolumes = dbClient.queryObject(Volume.class, getIds());
if (fullCopyvolumes != null && !fullCopyvolumes.isEmpty()) {
Volume fullCopyVolume = fullCopyvolumes.get(0);
Class<? extends BlockObject> sourceClass = Volume.class;
if (URIUtil.isType(fullCopyVolume.getAssociatedSourceVolume(), BlockSnapshot.class)) {
sourceClass = BlockSnapshot.class;
}
switch (status) {
case error:
dbClient.error(sourceClass, fullCopyVolume.getId(), getOpId(), coded);
break;
default:
dbClient.ready(sourceClass, fullCopyVolume.getId(), getOpId());
}
}
} catch (Exception e) {
log.error("Failed to update status for volume clone", e);
}
}
}