BlockSnapshotRestoreCompleter.java example

Explorer
coprhd-controller-master
/*
 * Copyright (c) 2012 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.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;
import com.emc.storageos.services.OperationTypeEnum;
import com.emc.storageos.volumecontroller.impl.ControllerUtils;

public class BlockSnapshotRestoreCompleter extends BlockSnapshotTaskCompleter {
    private static final Logger _log = LoggerFactory.getLogger(BlockSnapshotRestoreCompleter.class);
    public static final String SNAPSHOT_RESTORED_MSG = "Restored volume %s from snapshot %s";
    public static final String SNAPSHOT_RESTORE_FAILED_MSG = "Failed to restore volume %s from snapshot %s";

    private Boolean _updateAndRecordOp = true;

    public BlockSnapshotRestoreCompleter(BlockSnapshot snapshot, String task) {
        super(BlockSnapshot.class, snapshot.getId(), task);
    }
    
    public BlockSnapshotRestoreCompleter(URI snapshotURI, String task) {
        super(BlockSnapshot.class, snapshotURI, task);
    }

    public BlockSnapshotRestoreCompleter(BlockSnapshot snapshot, String task, Boolean updateAndRecordOp) {
        super(BlockSnapshot.class, snapshot.getId(), task);
        _updateAndRecordOp = updateAndRecordOp;
    }

    @Override
    protected void complete(DbClient dbClient, Operation.Status status, ServiceCoded coded) throws DeviceControllerException {
        try {
            BlockSnapshot snapshot = dbClient.queryObject(BlockSnapshot.class, getId());
            Volume volume = dbClient.queryObject(Volume.class, snapshot.getParent());
            if (_updateAndRecordOp) {
                if (snapshot.getConsistencyGroup() != null) {
                    // For snapshot based on a consistency group, set status and send
                    // events for all related snaps
                    List<BlockSnapshot> snaps = ControllerUtils.getSnapshotsPartOfReplicationGroup(snapshot, dbClient);
                    for (BlockSnapshot snap : snaps) {
                        URI uri = snap.getId();
                        BlockSnapshot it = dbClient.queryObject(BlockSnapshot.class, uri);
                        switch (status) {
                            case error:
                                dbClient.error(BlockSnapshot.class, uri, getOpId(), coded);
                                break;
                            default:
                                dbClient.ready(BlockSnapshot.class, uri, getOpId());
                        }

                        recordBlockSnapshotOperation(dbClient, OperationTypeEnum.RESTORE_VOLUME_SNAPSHOT, status,
                                eventMessage(status, volume, it), it, volume);
                        _log.info(String.format("Completed CG snapshot restore of snap %s (%s), with status %s",
                                it.getLabel(), uri.toString(), status.name()));
                    }
                } else {
                    switch (status) {
                        case error:
                            dbClient.error(BlockSnapshot.class, getId(), getOpId(), coded);
                            break;
                        default:
                            dbClient.ready(BlockSnapshot.class, getId(), getOpId());
                    }

                    recordBlockSnapshotOperation(dbClient, OperationTypeEnum.RESTORE_VOLUME_SNAPSHOT, status,
                            eventMessage(status, volume, snapshot), snapshot, volume);
                }
            }
            _log.info("Done SnapshotRestore {}, with Status: {}", getOpId(), status.name());
            super.complete(dbClient, status, coded);
        } catch (Exception e) {
            _log.error("Failed updating status. SnapshotRestore {}, for task " + getOpId(), getId(), e);
        }
    }

    private String eventMessage(Operation.Status status, Volume volume, BlockSnapshot snapshot) {
        return (status == Operation.Status.ready) ?
                String.format(SNAPSHOT_RESTORED_MSG, volume.getLabel(), snapshot.getLabel()) :
                String.format(SNAPSHOT_RESTORE_FAILED_MSG, volume.getLabel(), snapshot.getLabel());
    }
}