diff --git a/engine/schema/src/main/java/com/cloud/upgrade/SystemVmTemplateRegistration.java b/engine/schema/src/main/java/com/cloud/upgrade/SystemVmTemplateRegistration.java index 0ddaf00643f..dc94dd708b1 100644 --- a/engine/schema/src/main/java/com/cloud/upgrade/SystemVmTemplateRegistration.java +++ b/engine/schema/src/main/java/com/cloud/upgrade/SystemVmTemplateRegistration.java @@ -509,13 +509,12 @@ public class SystemVmTemplateRegistration { } } - public void updateTemplateDetails(SystemVMTemplateDetails details, boolean updateTemplateDetails) { + public void updateTemplateDetails(SystemVMTemplateDetails details) { VMTemplateVO template = vmTemplateDao.findById(details.getId()); - if (updateTemplateDetails) { - template.setSize(details.getSize()); - template.setState(VirtualMachineTemplate.State.Active); - vmTemplateDao.update(template.getId(), template); - } + template.setSize(details.getSize()); + template.setState(VirtualMachineTemplate.State.Active); + vmTemplateDao.update(template.getId(), template); + TemplateDataStoreVO templateDataStoreVO = templateDataStoreDao.findByStoreTemplate(details.getStoreId(), template.getId()); templateDataStoreVO.setSize(details.getSize()); templateDataStoreVO.setPhysicalSize(details.getPhysicalSize()); @@ -613,7 +612,7 @@ public class SystemVmTemplateRegistration { private Long performTemplateRegistrationOperations(Pair hypervisorAndTemplateName, String url, String checksum, ImageFormat format, long guestOsId, - Long storeId, Long templateId, String filePath, boolean updateTmpltDetails) { + Long storeId, Long templateId, String filePath, TemplateDataStoreVO templateDataStoreVO) { Hypervisor.HypervisorType hypervisor = hypervisorAndTemplateName.first(); String templateName = UUID.randomUUID().toString(); Date created = new Date(DateUtil.currentGMTTime().getTime()); @@ -631,21 +630,24 @@ public class SystemVmTemplateRegistration { String destTempFolderName = String.valueOf(templateId); String destTempFolder = filePath + PARTIAL_TEMPLATE_FOLDER + destTempFolderName; details.setInstallPath(PARTIAL_TEMPLATE_FOLDER + destTempFolderName + File.separator + templateName + "." + hypervisorImageFormat.get(hypervisor).getFileExtension()); - createTemplateStoreRefEntry(details); + if (templateDataStoreVO == null) { + createTemplateStoreRefEntry(details); + } setupTemplate(templateName, hypervisorAndTemplateName, destTempFolder); readTemplateProperties(destTempFolder + "/template.properties", details); details.setUpdated(new Date(DateUtil.currentGMTTime().getTime())); - updateTemplateDetails(details, updateTmpltDetails); + updateTemplateDetails(details); return templateId; } public void registerTemplate(Pair hypervisorAndTemplateName, - Pair storeUrlAndId, VMTemplateVO templateVO, String filePath) { + Pair storeUrlAndId, VMTemplateVO templateVO, + TemplateDataStoreVO templateDataStoreVO, String filePath) { Long templateId = null; try { templateId = templateVO.getId(); performTemplateRegistrationOperations(hypervisorAndTemplateName, templateVO.getUrl(), templateVO.getChecksum(), - templateVO.getFormat(), templateVO.getGuestOSId(), storeUrlAndId.second(), templateId, filePath, false); + templateVO.getFormat(), templateVO.getGuestOSId(), storeUrlAndId.second(), templateId, filePath, templateDataStoreVO); } catch (Exception e) { String errMsg = String.format("Failed to register template for hypervisor: %s", hypervisorAndTemplateName.first()); LOGGER.error(errMsg, e); @@ -662,7 +664,7 @@ public class SystemVmTemplateRegistration { try { Hypervisor.HypervisorType hypervisor = hypervisorAndTemplateName.first(); templateId = performTemplateRegistrationOperations(hypervisorAndTemplateName, NewTemplateUrl.get(hypervisor), NewTemplateChecksum.get(hypervisor), - hypervisorImageFormat.get(hypervisor), hypervisorGuestOsMap.get(hypervisor), storeUrlAndId.second(), null, filePath, true); + hypervisorImageFormat.get(hypervisor), hypervisorGuestOsMap.get(hypervisor), storeUrlAndId.second(), null, filePath, null); Map configParams = new HashMap<>(); configParams.put(RouterTemplateConfigurationNames.get(hypervisorAndTemplateName.first()), hypervisorAndTemplateName.second()); configParams.put("minreq.sysvmtemplate.version", getSystemVmTemplateVersion()); @@ -783,7 +785,7 @@ public class SystemVmTemplateRegistration { if (validateIfSeeded(storeUrlAndId.first(), installPath)) { continue; } else if (templateVO != null) { - registerTemplate(hypervisorAndTemplateName, storeUrlAndId, templateVO, filePath); + registerTemplate(hypervisorAndTemplateName, storeUrlAndId, templateVO, templateDataStoreVO, filePath); continue; } } diff --git a/server/src/main/java/com/cloud/storage/StorageManagerImpl.java b/server/src/main/java/com/cloud/storage/StorageManagerImpl.java index 7bafbcc77b8..643f29287e2 100644 --- a/server/src/main/java/com/cloud/storage/StorageManagerImpl.java +++ b/server/src/main/java/com/cloud/storage/StorageManagerImpl.java @@ -2919,7 +2919,7 @@ public class StorageManagerImpl extends ManagerBase implements StorageManager, C } SystemVmTemplateRegistration.mountStore(storeUrlAndId.first(), filePath); if (templateVO != null && vmTemplateVO != null) { - systemVmTemplateRegistration.registerTemplate(hypervisorAndTemplateName, storeUrlAndId, vmTemplateVO, filePath); + systemVmTemplateRegistration.registerTemplate(hypervisorAndTemplateName, storeUrlAndId, vmTemplateVO, templateVO, filePath); } else { systemVmTemplateRegistration.registerTemplate(hypervisorAndTemplateName, storeUrlAndId, filePath); }