| /* |
| * 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.kvm.resource.wrapper; |
| |
| import static com.cloud.hypervisor.kvm.storage.StorPoolStorageAdaptor.SP_LOG; |
| |
| import java.io.BufferedWriter; |
| import java.io.File; |
| import java.io.FileWriter; |
| import java.util.HashMap; |
| import java.util.Map; |
| |
| import org.apache.cloudstack.storage.command.CopyCmdAnswer; |
| import org.apache.cloudstack.storage.to.SnapshotObjectTO; |
| import org.apache.cloudstack.storage.to.TemplateObjectTO; |
| import org.apache.cloudstack.storage.to.VolumeObjectTO; |
| import org.apache.cloudstack.utils.qemu.QemuImg; |
| import org.apache.cloudstack.utils.qemu.QemuImg.PhysicalDiskFormat; |
| import org.apache.cloudstack.utils.qemu.QemuImgFile; |
| import org.apache.commons.io.FileUtils; |
| |
| import com.cloud.agent.api.storage.StorPoolBackupTemplateFromSnapshotCommand; |
| import com.cloud.agent.api.to.DataStoreTO; |
| import com.cloud.agent.api.to.DataTO; |
| import com.cloud.agent.api.to.NfsTO; |
| import com.cloud.hypervisor.kvm.resource.LibvirtComputingResource; |
| import com.cloud.hypervisor.kvm.storage.KVMStoragePool; |
| import com.cloud.hypervisor.kvm.storage.KVMStoragePoolManager; |
| import com.cloud.hypervisor.kvm.storage.StorPoolStorageAdaptor; |
| import com.cloud.resource.CommandWrapper; |
| import com.cloud.resource.ResourceWrapper; |
| import com.cloud.storage.Storage.ImageFormat; |
| import com.cloud.storage.StorageLayer; |
| import com.cloud.storage.template.Processor; |
| import com.cloud.storage.template.Processor.FormatInfo; |
| import com.cloud.storage.template.QCOW2Processor; |
| import com.cloud.storage.template.TemplateLocation; |
| import com.cloud.storage.template.TemplateProp; |
| |
| @ResourceWrapper(handles = StorPoolBackupTemplateFromSnapshotCommand.class) |
| public class StorPoolBackupTemplateFromSnapshotCommandWrapper extends CommandWrapper<StorPoolBackupTemplateFromSnapshotCommand, CopyCmdAnswer, LibvirtComputingResource> { |
| |
| |
| @Override |
| public CopyCmdAnswer execute(final StorPoolBackupTemplateFromSnapshotCommand cmd, final LibvirtComputingResource libvirtComputingResource) { |
| String srcPath = null; |
| KVMStoragePool secondaryPool = null; |
| String objectType = cmd.getSourceTO().getObjectType().toString().toLowerCase(); |
| |
| try { |
| final DataTO src = cmd.getSourceTO(); |
| final TemplateObjectTO dst = cmd.getDestinationTO(); |
| String name = null; |
| String volumeFormatExtension = null; |
| |
| if (src instanceof SnapshotObjectTO) { |
| name = ((SnapshotObjectTO) src).getName(); |
| volumeFormatExtension = ((SnapshotObjectTO) src).getVolume().getFormat().getFileExtension(); |
| } else if (src instanceof VolumeObjectTO) { |
| name = ((VolumeObjectTO) src).getName(); |
| volumeFormatExtension = ((VolumeObjectTO) src).getFormat().getFileExtension(); |
| } else { |
| return new CopyCmdAnswer("Backup of a template is not supported for data object: " + src.getObjectType() ); |
| } |
| final KVMStoragePoolManager storagePoolMgr = libvirtComputingResource.getStoragePoolMgr(); |
| StorageLayer storage = libvirtComputingResource.getStorage(); |
| Processor processor = new QCOW2Processor(); |
| String _tmpltpp = "template.properties"; |
| |
| SP_LOG("StorpoolBackupTemplateFromSnapshotCommandWrapper.execute: src=" + src.getPath() + "dst=" + dst.getPath()); |
| StorPoolStorageAdaptor.attachOrDetachVolume("attach", objectType, src.getPath()); |
| srcPath = src.getPath(); |
| |
| final QemuImgFile srcFile = new QemuImgFile(srcPath, PhysicalDiskFormat.RAW); |
| |
| final DataStoreTO dstDataStore = dst.getDataStore(); |
| if (!(dstDataStore instanceof NfsTO)) { |
| return new CopyCmdAnswer("Backup Storpool snapshot: Only NFS secondary supported at present!"); |
| } |
| |
| secondaryPool = storagePoolMgr.getStoragePoolByURI(dstDataStore.getUrl()); |
| |
| final String dstDir = secondaryPool.getLocalPath() + File.separator + dst.getPath(); |
| FileUtils.forceMkdir(new File(dstDir)); |
| |
| String nameWithExtension = name + "." + volumeFormatExtension; |
| |
| final String dstPath = dstDir + File.separator + nameWithExtension; |
| final QemuImgFile dstFile = new QemuImgFile(dstPath, PhysicalDiskFormat.QCOW2); |
| |
| final QemuImg qemu = new QemuImg(cmd.getWaitInMillSeconds()); |
| qemu.convert(srcFile, dstFile); |
| |
| storage.create(dstDir, _tmpltpp); |
| String metaFileName = dstDir + File.separator + _tmpltpp; |
| File metaFile = new File(metaFileName); |
| |
| try ( FileWriter writer = new FileWriter(metaFile); |
| BufferedWriter bufferWriter = new BufferedWriter(writer);) { |
| bufferWriter.write("uniquename=" + dst.getName()); |
| bufferWriter.write("\n"); |
| bufferWriter.write("filename=" + nameWithExtension); |
| } |
| Map<String, Object> params = new HashMap<String, Object>(); |
| params.put(StorageLayer.InstanceConfigKey, storage); |
| |
| processor.configure("template processor", params); |
| |
| FormatInfo info = processor.process(dstDir, null, name); |
| TemplateLocation loc = new TemplateLocation(storage, dstDir); |
| loc.create(1, true, dst.getName()); |
| loc.addFormat(info); |
| loc.save(); |
| |
| TemplateProp prop = loc.getTemplateInfo(); |
| final TemplateObjectTO template = new TemplateObjectTO(); |
| template.setPath(dst.getPath() + File.separator + nameWithExtension); |
| template.setFormat(ImageFormat.QCOW2); |
| template.setSize(prop.getSize()); |
| template.setPhysicalSize(prop.getPhysicalSize()); |
| |
| return new CopyCmdAnswer(template); |
| } catch (final Exception e) { |
| final String error = "failed to backup snapshot: " + e.getMessage(); |
| SP_LOG(error); |
| logger.debug(error); |
| return new CopyCmdAnswer(cmd, e); |
| } finally { |
| if (srcPath != null) { |
| StorPoolStorageAdaptor.attachOrDetachVolume("detach", objectType, srcPath); |
| } |
| |
| if (secondaryPool != null) { |
| try { |
| secondaryPool.delete(); |
| } catch (final Exception e) { |
| logger.debug("Failed to delete secondary storage", e); |
| } |
| } |
| } |
| } |
| } |