| // |
| // 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.storage.resource; |
| |
| import com.cloud.serializer.GsonHelper; |
| import org.apache.cloudstack.agent.directdownload.DirectDownloadCommand; |
| import org.apache.cloudstack.storage.to.VolumeObjectTO; |
| import org.apache.cloudstack.storage.command.CheckDataStoreStoragePolicyComplainceCommand; |
| import org.apache.log4j.Logger; |
| |
| import org.apache.cloudstack.storage.command.AttachCommand; |
| import org.apache.cloudstack.storage.command.CopyCommand; |
| import org.apache.cloudstack.storage.command.CreateObjectAnswer; |
| import org.apache.cloudstack.storage.command.CreateObjectCommand; |
| import org.apache.cloudstack.storage.command.DeleteCommand; |
| import org.apache.cloudstack.storage.command.DettachCommand; |
| import org.apache.cloudstack.storage.command.IntroduceObjectCmd; |
| import org.apache.cloudstack.storage.command.ResignatureCommand; |
| import org.apache.cloudstack.storage.command.SnapshotAndCopyCommand; |
| import org.apache.cloudstack.storage.command.StorageSubSystemCommand; |
| import org.apache.cloudstack.storage.command.SyncVolumePathCommand; |
| |
| import com.cloud.agent.api.Answer; |
| import com.cloud.agent.api.Command; |
| import com.cloud.agent.api.to.DataObjectType; |
| import com.cloud.agent.api.to.DataStoreTO; |
| import com.cloud.agent.api.to.DataTO; |
| import com.cloud.agent.api.to.DiskTO; |
| import com.cloud.storage.DataStoreRole; |
| import com.cloud.storage.Volume; |
| import com.google.gson.Gson; |
| |
| public class StorageSubsystemCommandHandlerBase implements StorageSubsystemCommandHandler { |
| private static final Logger s_logger = Logger.getLogger(StorageSubsystemCommandHandlerBase.class); |
| protected static final Gson s_gogger = GsonHelper.getGsonLogger(); |
| protected StorageProcessor processor; |
| |
| public StorageSubsystemCommandHandlerBase(StorageProcessor processor) { |
| this.processor = processor; |
| } |
| |
| @Override |
| public Answer handleStorageCommands(StorageSubSystemCommand command) { |
| logCommand(command); |
| if (command instanceof CopyCommand) { |
| return this.execute((CopyCommand)command); |
| } else if (command instanceof CreateObjectCommand) { |
| return execute((CreateObjectCommand)command); |
| } else if (command instanceof DeleteCommand) { |
| return execute((DeleteCommand)command); |
| } else if (command instanceof AttachCommand) { |
| return execute((AttachCommand)command); |
| } else if (command instanceof DettachCommand) { |
| return execute((DettachCommand)command); |
| } else if (command instanceof IntroduceObjectCmd) { |
| return processor.introduceObject((IntroduceObjectCmd)command); |
| } else if (command instanceof SnapshotAndCopyCommand) { |
| return processor.snapshotAndCopy((SnapshotAndCopyCommand)command); |
| } else if (command instanceof ResignatureCommand) { |
| return processor.resignature((ResignatureCommand) command); |
| } else if (command instanceof DirectDownloadCommand) { |
| return processor.handleDownloadTemplateToPrimaryStorage((DirectDownloadCommand) command); |
| } else if (command instanceof CheckDataStoreStoragePolicyComplainceCommand) { |
| return processor.checkDataStoreStoragePolicyCompliance((CheckDataStoreStoragePolicyComplainceCommand) command); |
| } else if (command instanceof SyncVolumePathCommand) { |
| return processor.syncVolumePath((SyncVolumePathCommand) command); |
| } |
| |
| return new Answer((Command)command, false, "not implemented yet"); |
| } |
| |
| protected Answer execute(CopyCommand cmd) { |
| DataTO srcData = cmd.getSrcTO(); |
| DataTO destData = cmd.getDestTO(); |
| DataStoreTO srcDataStore = srcData.getDataStore(); |
| DataStoreTO destDataStore = destData.getDataStore(); |
| |
| if (srcData.getObjectType() == DataObjectType.TEMPLATE && |
| (srcData.getDataStore().getRole() == DataStoreRole.Image || srcData.getDataStore().getRole() == DataStoreRole.ImageCache) && |
| destData.getDataStore().getRole() == DataStoreRole.Primary) { |
| //copy template to primary storage |
| return processor.copyTemplateToPrimaryStorage(cmd); |
| } else if (srcData.getObjectType() == DataObjectType.TEMPLATE && srcDataStore.getRole() == DataStoreRole.Primary && |
| destDataStore.getRole() == DataStoreRole.Primary) { |
| //clone template to a volume |
| return processor.cloneVolumeFromBaseTemplate(cmd); |
| } else if (srcData.getObjectType() == DataObjectType.VOLUME && |
| (srcData.getDataStore().getRole() == DataStoreRole.ImageCache || srcDataStore.getRole() == DataStoreRole.Image)) { |
| //copy volume from image cache to primary |
| return processor.copyVolumeFromImageCacheToPrimary(cmd); |
| } else if (srcData.getObjectType() == DataObjectType.VOLUME && srcData.getDataStore().getRole() == DataStoreRole.Primary) { |
| if (destData.getObjectType() == DataObjectType.VOLUME) { |
| if ((srcData instanceof VolumeObjectTO && ((VolumeObjectTO)srcData).isDirectDownload()) || |
| destData.getDataStore().getRole() == DataStoreRole.Primary) { |
| return processor.copyVolumeFromPrimaryToPrimary(cmd); |
| } else { |
| return processor.copyVolumeFromPrimaryToSecondary(cmd); |
| } |
| } else if (destData.getObjectType() == DataObjectType.TEMPLATE) { |
| return processor.createTemplateFromVolume(cmd); |
| } |
| } else if (srcData.getObjectType() == DataObjectType.SNAPSHOT && destData.getObjectType() == DataObjectType.SNAPSHOT && |
| srcData.getDataStore().getRole() == DataStoreRole.Primary) { |
| return processor.backupSnapshot(cmd); |
| } else if (srcData.getObjectType() == DataObjectType.SNAPSHOT && destData.getObjectType() == DataObjectType.VOLUME) { |
| return processor.createVolumeFromSnapshot(cmd); |
| } else if (srcData.getObjectType() == DataObjectType.SNAPSHOT && destData.getObjectType() == DataObjectType.TEMPLATE) { |
| return processor.createTemplateFromSnapshot(cmd); |
| } |
| |
| return new Answer(cmd, false, "not implemented yet"); |
| } |
| |
| protected Answer execute(CreateObjectCommand cmd) { |
| DataTO data = cmd.getData(); |
| try { |
| if (data.getObjectType() == DataObjectType.VOLUME) { |
| return processor.createVolume(cmd); |
| } else if (data.getObjectType() == DataObjectType.SNAPSHOT) { |
| return processor.createSnapshot(cmd); |
| } |
| return new CreateObjectAnswer("not supported type"); |
| } catch (Exception e) { |
| s_logger.debug("Failed to create object: " + data.getObjectType() + ": " + e.toString()); |
| return new CreateObjectAnswer(e.toString()); |
| } |
| } |
| |
| protected Answer execute(DeleteCommand cmd) { |
| DataTO data = cmd.getData(); |
| Answer answer = null; |
| if (data.getObjectType() == DataObjectType.VOLUME) { |
| answer = processor.deleteVolume(cmd); |
| } else if (data.getObjectType() == DataObjectType.SNAPSHOT) { |
| answer = processor.deleteSnapshot(cmd); |
| } else { |
| answer = new Answer(cmd, false, "unsupported type"); |
| } |
| |
| return answer; |
| } |
| |
| protected Answer execute(AttachCommand cmd) { |
| DiskTO disk = cmd.getDisk(); |
| if (disk.getType() == Volume.Type.ISO) { |
| return processor.attachIso(cmd); |
| } else { |
| return processor.attachVolume(cmd); |
| } |
| } |
| |
| protected Answer execute(DettachCommand cmd) { |
| DiskTO disk = cmd.getDisk(); |
| if (disk.getType() == Volume.Type.ISO) { |
| return processor.dettachIso(cmd); |
| } else { |
| return processor.dettachVolume(cmd); |
| } |
| } |
| |
| private void logCommand(Command cmd) { |
| try { |
| s_logger.debug(String.format("Executing command %s: [%s].", cmd.getClass().getSimpleName(), s_gogger.toJson(cmd))); |
| } catch (Exception e) { |
| s_logger.debug(String.format("Executing command %s.", cmd.getClass().getSimpleName())); |
| } |
| } |
| } |
