mirror of https://github.com/apache/cloudstack.git
136 lines
5.8 KiB
Java
136 lines
5.8 KiB
Java
/**
|
|
* Copyright (C) 2010 Cloud.com, Inc. All rights reserved.
|
|
*
|
|
* This software is licensed under the GNU General Public License v3 or later.
|
|
*
|
|
* It is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or any later version.
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*
|
|
*/
|
|
|
|
package com.cloud.async.executor;
|
|
|
|
import org.apache.log4j.Logger;
|
|
|
|
import com.cloud.agent.api.Answer;
|
|
import com.cloud.api.BaseCmd;
|
|
import com.cloud.async.AsyncJobManager;
|
|
import com.cloud.async.AsyncJobResult;
|
|
import com.cloud.async.AsyncJobVO;
|
|
import com.cloud.async.BaseAsyncJobExecutor;
|
|
import com.cloud.dc.DataCenterVO;
|
|
import com.cloud.domain.Domain;
|
|
import com.cloud.serializer.GsonHelper;
|
|
import com.cloud.server.ManagementServer;
|
|
import com.cloud.storage.VMTemplateHostVO;
|
|
import com.cloud.storage.VMTemplateStorageResourceAssoc.Status;
|
|
import com.cloud.storage.VMTemplateVO;
|
|
import com.cloud.user.Account;
|
|
import com.google.gson.Gson;
|
|
|
|
public class CopyTemplateExecutor extends BaseAsyncJobExecutor {
|
|
public static final Logger s_logger = Logger.getLogger(CopyTemplateExecutor.class.getName());
|
|
|
|
public boolean execute() {
|
|
Gson gson = GsonHelper.getBuilder().create();
|
|
AsyncJobManager asyncMgr = getAsyncJobMgr();
|
|
AsyncJobVO job = getJob();
|
|
|
|
CopyTemplateParam param = gson.fromJson(job.getCmdInfo(), CopyTemplateParam.class);
|
|
|
|
ManagementServer managementServer = asyncMgr.getExecutorContext().getManagementServer();
|
|
|
|
try {
|
|
boolean success = managementServer.copyTemplate(param.getUserId(), param.getTemplateId(), param.getSourceZoneId(), param.getDestZoneId(), param.getEventId());
|
|
|
|
if (success) {
|
|
VMTemplateVO template = managementServer.findTemplateById(param.getTemplateId());
|
|
DataCenterVO destZone = managementServer.findDataCenterById(param.getDestZoneId());
|
|
// VMTemplateHostVO templateHostRef = managementServer.findTemplateHostRef(param.getTemplateId(), destZone.getId());
|
|
long guestOsId = template.getGuestOSId();
|
|
Account owner = managementServer.findAccountById(template.getAccountId());
|
|
|
|
// DomainVO domain = managementServer.findDomainIdById(owner.getDomainId());
|
|
// String guestOSName = managementServer.findGuestOSById(guestOsId).getName();
|
|
// asyncMgr.completeAsyncJob(getJob().getId(), AsyncJobResult.STATUS_SUCCEEDED, 0, composeResultObject(template, templateHostRef, destZone,guestOSName, owner, domain));
|
|
|
|
} else {
|
|
asyncMgr.completeAsyncJob(getJob().getId(), AsyncJobResult.STATUS_FAILED, BaseCmd.INTERNAL_ERROR, "Failed to copy template.");
|
|
}
|
|
|
|
} catch (Exception e) {
|
|
s_logger.warn("Unable to copy template: " + e.getMessage(), e);
|
|
asyncMgr.completeAsyncJob(getJob().getId(), AsyncJobResult.STATUS_FAILED, BaseCmd.INTERNAL_ERROR, e.getMessage());
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
public void processAnswer(VMOperationListener listener, long agentId, long seq, Answer answer) {
|
|
}
|
|
|
|
public void processDisconnect(VMOperationListener listener, long agentId) {
|
|
}
|
|
|
|
public void processTimeout(VMOperationListener listener, long agentId, long seq) {
|
|
}
|
|
|
|
private CopyTemplateResultObject composeResultObject(VMTemplateVO template, VMTemplateHostVO templateHostRef, DataCenterVO destZone, String guestOSName, Account owner, Domain domain) {
|
|
CopyTemplateResultObject resultObject = new CopyTemplateResultObject();
|
|
|
|
|
|
// If the user is an admin, add the template download status
|
|
boolean isAdmin = false;
|
|
|
|
if(owner.getType() == Account.ACCOUNT_TYPE_ADMIN)
|
|
isAdmin = true;
|
|
|
|
if (isAdmin || owner.getId() == template.getAccountId()) {
|
|
// add download status
|
|
if (templateHostRef.getDownloadState()!=Status.DOWNLOADED) {
|
|
String templateStatus = "Processing";
|
|
if (templateHostRef.getDownloadState() == VMTemplateHostVO.Status.DOWNLOAD_IN_PROGRESS) {
|
|
if (templateHostRef.getDownloadPercent() == 100) {
|
|
templateStatus = "Installing Template";
|
|
} else {
|
|
templateStatus = templateHostRef.getDownloadPercent() + "% Downloaded";
|
|
}
|
|
} else {
|
|
templateStatus = templateHostRef.getErrorString();
|
|
}
|
|
resultObject.setTemplateStatus(templateStatus);
|
|
} else if (templateHostRef.getDownloadState() == VMTemplateHostVO.Status.DOWNLOADED) {
|
|
resultObject.setTemplateStatus("Download Complete");
|
|
} else {
|
|
resultObject.setTemplateStatus("Successfully Installed");
|
|
}
|
|
}
|
|
|
|
resultObject.setAccountName(owner.getAccountName());
|
|
resultObject.setDomainId(owner.getDomainId());
|
|
resultObject.setDomainName(domain.getName());
|
|
resultObject.setId(template.getId());
|
|
resultObject.setName(template.getName());
|
|
resultObject.setDisplayText(template.getDisplayText());
|
|
resultObject.setPublic(template.isPublicTemplate());
|
|
resultObject.setOsTypeName(guestOSName);
|
|
resultObject.setCreated(templateHostRef.getCreated());
|
|
resultObject.setReady(templateHostRef != null && templateHostRef.getDownloadState() == Status.DOWNLOADED);
|
|
resultObject.setFeatured(template.isFeatured());
|
|
resultObject.setPasswordEnabled(template.getEnablePassword());
|
|
resultObject.setFormat(template.getFormat());
|
|
resultObject.setGuestOsId(template.getGuestOSId());
|
|
resultObject.setZoneId(destZone.getId());
|
|
resultObject.setZoneName(destZone.getName());
|
|
return resultObject;
|
|
}
|
|
}
|