content: //
  // Licensed to the Apache Software Foundation (ASF) under one
  // or more contributor license agreements.  See the NOTICE file
  // distributed with this work for additional information
  // regarding copyright ownership.  The ASF licenses this file
  // to you under the Apache License, Version 2.0 (the
  // "License"); you may not use this file except in compliance
  // with the License.  You may obtain a copy of the License at
  //
  //   http://www.apache.org/licenses/LICENSE-2.0
  //
  // Unless required by applicable law or agreed to in writing,
  // software distributed under the License is distributed on an
  // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  // KIND, either express or implied.  See the License for the
  // specific language governing permissions and limitations
  // under the License.
  //
  
  package com.cloud.hypervisor.xenserver.resource.wrapper.xenbase;
  
  
  import com.cloud.agent.api.Answer;
  import com.cloud.agent.api.PrepareForMigrationAnswer;
  import com.cloud.agent.api.PrepareForMigrationCommand;
  import com.cloud.agent.api.to.NicTO;
  import com.cloud.agent.api.to.VirtualMachineTO;
  import com.cloud.hypervisor.xenserver.resource.CitrixResourceBase;
  import com.cloud.resource.CommandWrapper;
  import com.cloud.resource.ResourceWrapper;
  import com.xensource.xenapi.Connection;
  
  import java.util.List;
  
  @ResourceWrapper(handles =  PrepareForMigrationCommand.class)
  public final class CitrixPrepareForMigrationCommandWrapper extends CommandWrapper<PrepareForMigrationCommand, Answer, CitrixResourceBase> {
  
  
      @Override
      public Answer execute(final PrepareForMigrationCommand command, final CitrixResourceBase citrixResourceBase) {
          final Connection conn = citrixResourceBase.getConnection();
  
          final VirtualMachineTO vm = command.getVirtualMachine();
          List<String[]> vmDataList = vm.getVmData();
          String configDriveLabel = vm.getConfigDriveLabel();
  
          if (configDriveLabel == null)  {
              configDriveLabel = "config-2";
          }
  
          if (logger.isDebugEnabled()) {
              logger.debug("Preparing host for migrating " + vm);
          }
  
          final NicTO[] nics = vm.getNics();
          try {
              citrixResourceBase.prepareISO(conn, vm.getName(), vmDataList, configDriveLabel);
  
              for (final NicTO nic : nics) {
                  citrixResourceBase.getNetwork(conn, nic);
              }
              logger.debug("4. The VM " + vm.getName() + " is in Migrating state");
  
              return new PrepareForMigrationAnswer(command);
          } catch (final Exception e) {
              logger.warn("Catch Exception " + e.getClass().getName() + " prepare for migration failed due to " + e.toString(), e);
              return new PrepareForMigrationAnswer(command, e);
          }
      }
  }
  