コード例 #1
0
  @Before
  public void setup() throws Exception {
    MockitoAnnotations.initMocks(this);
    storageCmd = PowerMockito.mock(CopyCommand.class);
    doReturn(context).when(_resource).getServiceContext(null);
    when(cmd.getVirtualMachine()).thenReturn(vmSpec);

    when(storageCmd.getSrcTO()).thenReturn(srcDataTO);
    when(srcDataTO.getDataStore()).thenReturn(srcDataNfsTO);
    when(srcDataNfsTO.getNfsVersion()).thenReturn(NFS_VERSION);
    when(videoCard.getVideoRamSizeInKB()).thenReturn(VIDEO_CARD_MEMORY_SIZE);
    when(volume.getPath()).thenReturn(VOLUME_PATH);

    when(storageCmd.getDestTO()).thenReturn(destDataTO);
    when(destDataTO.getHypervisorType()).thenReturn(HypervisorType.VMware);
    when(destDataTO.getDataStore()).thenReturn(destDataStoreTO);
    when(destDataStoreTO.isFullCloneFlag()).thenReturn(FULL_CLONE_FLAG);
    when(volume.getPath()).thenReturn(VOLUME_PATH);
    when(vmSpec.getDetails()).thenReturn(specsArray);

    when(vmMo.getContext()).thenReturn(context);
    when(vmMo.getRunningHost()).thenReturn(host);
    when(host.getMor()).thenReturn(hostRef);
    when(context.getVimClient()).thenReturn(client);
    when(client.getMoRefProp(hostRef, "parent")).thenReturn(computeRef);
    when(client.getMoRefProp(computeRef, "environmentBrowser")).thenReturn(envRef);
    when(context.getService()).thenReturn(vimService);
    when(vimService.queryTargetCapabilities(envRef, hostRef)).thenReturn(hostCapability);
    when(hostCapability.isNestedHVSupported()).thenReturn(true);
  }
コード例 #2
0
  @Override
  protected Answer execute(CopyCommand cmd) {
    DataTO srcData = cmd.getSrcTO();
    DataTO destData = cmd.getDestTO();
    DataStoreTO srcDataStore = srcData.getDataStore();
    DataStoreTO destDataStore = destData.getDataStore();
    // if copied between s3 and nfs cache, go to resource
    boolean needDelegation = false;
    if (destDataStore instanceof NfsTO && destDataStore.getRole() == DataStoreRole.ImageCache) {
      if (srcDataStore instanceof S3TO || srcDataStore instanceof SwiftTO) {
        needDelegation = true;
      }
    }

    if (srcDataStore.getRole() == DataStoreRole.ImageCache
        && destDataStore.getRole() == DataStoreRole.Image) {
      // need to take extra processing for vmware, such as packing to ova, before sending to S3
      if (srcData.getObjectType() == DataObjectType.VOLUME) {
        NfsTO cacheStore = (NfsTO) srcDataStore;
        String parentPath = storageResource.getRootDir(cacheStore.getUrl(), _nfsVersion);
        VolumeObjectTO vol = (VolumeObjectTO) srcData;
        String path = vol.getPath();
        int index = path.lastIndexOf(File.separator);
        String name = path.substring(index + 1);
        storageManager.createOva(parentPath + File.separator + path, name);
        vol.setPath(path + File.separator + name + ".ova");
      } else if (srcData.getObjectType() == DataObjectType.TEMPLATE) {
        // sync template from NFS cache to S3 in NFS migration to S3 case
        storageManager.createOvaForTemplate((TemplateObjectTO) srcData);
      } else if (srcData.getObjectType() == DataObjectType.SNAPSHOT) {
        // pack ova first
        // sync snapshot from NFS cache to S3 in NFS migration to S3 case
        String parentPath = storageResource.getRootDir(srcDataStore.getUrl(), _nfsVersion);
        SnapshotObjectTO snap = (SnapshotObjectTO) srcData;
        String path = snap.getPath();
        int index = path.lastIndexOf(File.separator);
        String name = path.substring(index + 1);
        String snapDir = path.substring(0, index);
        storageManager.createOva(parentPath + File.separator + snapDir, name);
        if (destData.getObjectType() == DataObjectType.TEMPLATE) {
          // create template from snapshot on src at first, then copy it to s3
          TemplateObjectTO cacheTemplate = (TemplateObjectTO) destData;
          cacheTemplate.setDataStore(srcDataStore);
          CopyCmdAnswer answer = (CopyCmdAnswer) processor.createTemplateFromSnapshot(cmd);
          if (!answer.getResult()) {
            return answer;
          }
          cacheTemplate.setDataStore(destDataStore);
          TemplateObjectTO template = (TemplateObjectTO) answer.getNewData();
          template.setDataStore(srcDataStore);
          CopyCommand newCmd =
              new CopyCommand(template, destData, cmd.getWait(), cmd.executeInSequence());
          Answer result = storageResource.defaultAction(newCmd);
          // clean up template data on staging area
          try {
            DeleteCommand deleteCommand = new DeleteCommand(template);
            storageResource.defaultAction(deleteCommand);
          } catch (Exception e) {
            s_logger.debug("Failed to clean up staging area:", e);
          }
          return result;
        }
      }
      needDelegation = true;
    }

    if (srcData.getObjectType() == DataObjectType.SNAPSHOT
        && srcData.getDataStore().getRole() == DataStoreRole.Primary) {
      // for back up snapshot, we need to do backup to cache, then to object store if object store
      // is used.
      if (cmd.getCacheTO() != null) {
        cmd.setDestTO(cmd.getCacheTO());

        CopyCmdAnswer answer = (CopyCmdAnswer) processor.backupSnapshot(cmd);
        if (!answer.getResult()) {
          return answer;
        }
        NfsTO cacheStore = (NfsTO) cmd.getCacheTO().getDataStore();
        String parentPath = storageResource.getRootDir(cacheStore.getUrl(), _nfsVersion);
        SnapshotObjectTO newSnapshot = (SnapshotObjectTO) answer.getNewData();
        String path = newSnapshot.getPath();
        int index = path.lastIndexOf(File.separator);
        String name = path.substring(index + 1);
        String dir = path.substring(0, index);
        storageManager.createOva(parentPath + File.separator + dir, name);
        newSnapshot.setPath(newSnapshot.getPath() + ".ova");
        newSnapshot.setDataStore(cmd.getCacheTO().getDataStore());
        CopyCommand newCmd =
            new CopyCommand(newSnapshot, destData, cmd.getWait(), cmd.executeInSequence());
        Answer result = storageResource.defaultAction(newCmd);

        // clean up data on staging area
        try {
          newSnapshot.setPath(path);
          DeleteCommand deleteCommand = new DeleteCommand(newSnapshot);
          storageResource.defaultAction(deleteCommand);
        } catch (Exception e) {
          s_logger.debug("Failed to clean up staging area:", e);
        }
        return result;
      }
    }

    if (needDelegation) {
      return storageResource.defaultAction(cmd);
    } else {
      return super.execute(cmd);
    }
  }