mirror of https://github.com/apache/cloudstack.git
235 lines
11 KiB
Java
235 lines
11 KiB
Java
// 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.upgrade.dao;
|
|
|
|
import java.io.File;
|
|
import java.sql.Connection;
|
|
import java.sql.PreparedStatement;
|
|
import java.sql.ResultSet;
|
|
import java.sql.SQLException;
|
|
import java.util.ArrayList;
|
|
import java.util.HashMap;
|
|
import java.util.HashSet;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
|
|
import com.cloud.hypervisor.Hypervisor;
|
|
import org.apache.log4j.Logger;
|
|
|
|
import com.cloud.utils.exception.CloudRuntimeException;
|
|
import com.cloud.utils.script.Script;
|
|
|
|
public class Upgrade441to450 implements DbUpgrade {
|
|
final static Logger s_logger = Logger.getLogger(Upgrade441to450.class);
|
|
|
|
@Override
|
|
public String[] getUpgradableVersionRange() {
|
|
return new String[] {"4.4.1", "4.5.0"};
|
|
}
|
|
|
|
@Override
|
|
public String getUpgradedVersion() {
|
|
return "4.5.0";
|
|
}
|
|
|
|
@Override
|
|
public boolean supportsRollingUpgrade() {
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
public File[] getPrepareScripts() {
|
|
String script = Script.findScript("", "db/schema-441to450.sql");
|
|
if (script == null) {
|
|
throw new CloudRuntimeException("Unable to find db/schema-441to450.sql");
|
|
}
|
|
|
|
return new File[] {new File(script)};
|
|
}
|
|
|
|
@Override
|
|
public void performDataMigration(Connection conn) {
|
|
updateSystemVmTemplates(conn);
|
|
dropInvalidKeyFromStoragePoolTable(conn);
|
|
dropDuplicatedForeignKeyFromAsyncJobTable(conn);
|
|
}
|
|
|
|
|
|
@Override
|
|
public File[] getCleanupScripts() {
|
|
String script = Script.findScript("", "db/schema-441to450-cleanup.sql");
|
|
if (script == null) {
|
|
throw new CloudRuntimeException("Unable to find db/schema-441to450-cleanup.sql");
|
|
}
|
|
|
|
return new File[] {new File(script)};
|
|
}
|
|
|
|
private void updateSystemVmTemplates(Connection conn) {
|
|
PreparedStatement pstmt = null;
|
|
ResultSet rs = null;
|
|
s_logger.debug("Updating System Vm template IDs");
|
|
try{
|
|
//Get all hypervisors in use
|
|
Set<Hypervisor.HypervisorType> hypervisorsListInUse = new HashSet<Hypervisor.HypervisorType>();
|
|
try {
|
|
pstmt = conn.prepareStatement("select distinct(hypervisor_type) from `cloud`.`cluster` where removed is null");
|
|
rs = pstmt.executeQuery();
|
|
while(rs.next()){
|
|
switch (Hypervisor.HypervisorType.getType(rs.getString(1))) {
|
|
case XenServer: hypervisorsListInUse.add(Hypervisor.HypervisorType.XenServer);
|
|
break;
|
|
case KVM: hypervisorsListInUse.add(Hypervisor.HypervisorType.KVM);
|
|
break;
|
|
case VMware: hypervisorsListInUse.add(Hypervisor.HypervisorType.VMware);
|
|
break;
|
|
case Hyperv: hypervisorsListInUse.add(Hypervisor.HypervisorType.Hyperv);
|
|
break;
|
|
case LXC: hypervisorsListInUse.add(Hypervisor.HypervisorType.LXC);
|
|
break;
|
|
}
|
|
}
|
|
} catch (SQLException e) {
|
|
throw new CloudRuntimeException("Error while listing hypervisors in use", e);
|
|
}
|
|
|
|
Map<Hypervisor.HypervisorType, String> NewTemplateNameList = new HashMap<Hypervisor.HypervisorType, String>(){
|
|
{ put(Hypervisor.HypervisorType.XenServer, "systemvm-xenserver-4.5");
|
|
put(Hypervisor.HypervisorType.VMware, "systemvm-vmware-4.5");
|
|
put(Hypervisor.HypervisorType.KVM, "systemvm-kvm-4.5");
|
|
put(Hypervisor.HypervisorType.LXC, "systemvm-lxc-4.5");
|
|
put(Hypervisor.HypervisorType.Hyperv, "systemvm-hyperv-4.5");
|
|
}
|
|
};
|
|
|
|
Map<Hypervisor.HypervisorType, String> routerTemplateConfigurationNames = new HashMap<Hypervisor.HypervisorType, String>(){
|
|
{ put(Hypervisor.HypervisorType.XenServer, "router.template.xen");
|
|
put(Hypervisor.HypervisorType.VMware, "router.template.vmware");
|
|
put(Hypervisor.HypervisorType.KVM, "router.template.kvm");
|
|
put(Hypervisor.HypervisorType.LXC, "router.template.lxc");
|
|
put(Hypervisor.HypervisorType.Hyperv, "router.template.hyperv");
|
|
}
|
|
};
|
|
|
|
Map<Hypervisor.HypervisorType, String> newTemplateUrl = new HashMap<Hypervisor.HypervisorType, String>(){
|
|
{ put(Hypervisor.HypervisorType.XenServer, "http://download.cloud.com/templates/4.5/systemvm64template-4.5-xen.vhd.bz2");
|
|
put(Hypervisor.HypervisorType.VMware, "http://download.cloud.com/templates/4.5/systemvm64template-4.5-vmware.ova");
|
|
put(Hypervisor.HypervisorType.KVM, "http://download.cloud.com/templates/4.5/systemvm64template-4.5-kvm.qcow2.bz2");
|
|
put(Hypervisor.HypervisorType.LXC, "http://download.cloud.com/templates/4.5/systemvm64template-4.5-kvm.qcow2.bz2");
|
|
put(Hypervisor.HypervisorType.Hyperv, "http://download.cloud.com/templates/4.5/systemvm64template-4.5-hyperv.vhd.zip");
|
|
}
|
|
};
|
|
|
|
Map<Hypervisor.HypervisorType, String> newTemplateChecksum = new HashMap<Hypervisor.HypervisorType, String>(){
|
|
{ put(Hypervisor.HypervisorType.XenServer, "2b15ab4401c2d655264732d3fc600241");
|
|
put(Hypervisor.HypervisorType.VMware, "3106a79a4ce66cd7f6a7c50e93f2db57");
|
|
put(Hypervisor.HypervisorType.KVM, "aa9f501fecd3de1daeb9e2f357f6f002");
|
|
put(Hypervisor.HypervisorType.LXC, "aa9f501fecd3de1daeb9e2f357f6f002");
|
|
put(Hypervisor.HypervisorType.Hyperv, "70bd30ea02ee9ed67d2c6b85c179cee9");
|
|
}
|
|
};
|
|
|
|
for (Map.Entry<Hypervisor.HypervisorType, String> hypervisorAndTemplateName : NewTemplateNameList.entrySet()){
|
|
s_logger.debug("Updating " + hypervisorAndTemplateName.getKey() + " System Vms");
|
|
try {
|
|
//Get 4.5.0 system Vm template Id for corresponding hypervisor
|
|
pstmt = conn.prepareStatement("select id from `cloud`.`vm_template` where name = ? and removed is null order by id desc limit 1");
|
|
pstmt.setString(1, hypervisorAndTemplateName.getValue());
|
|
rs = pstmt.executeQuery();
|
|
if(rs.next()){
|
|
long templateId = rs.getLong(1);
|
|
rs.close();
|
|
pstmt.close();
|
|
pstmt = conn.prepareStatement("update `cloud`.`vm_template` set type='SYSTEM' where id = ?");
|
|
pstmt.setLong(1, templateId);
|
|
pstmt.executeUpdate();
|
|
pstmt.close();
|
|
// update templete ID of system Vms
|
|
pstmt = conn.prepareStatement("update `cloud`.`vm_instance` set vm_template_id = ? where type <> 'User' and hypervisor_type = ?");
|
|
pstmt.setLong(1, templateId);
|
|
pstmt.setString(2, hypervisorAndTemplateName.getKey().toString());
|
|
pstmt.executeUpdate();
|
|
pstmt.close();
|
|
// Change value of global configuration parameter router.template.* for the corresponding hypervisor
|
|
pstmt = conn.prepareStatement("UPDATE `cloud`.`configuration` SET value = ? WHERE name = ?");
|
|
pstmt.setString(1, hypervisorAndTemplateName.getValue());
|
|
pstmt.setString(2, routerTemplateConfigurationNames.get(hypervisorAndTemplateName.getKey()));
|
|
pstmt.executeUpdate();
|
|
pstmt.close();
|
|
} else {
|
|
if (hypervisorsListInUse.contains(hypervisorAndTemplateName.getKey())){
|
|
throw new CloudRuntimeException("4.5.0 " + hypervisorAndTemplateName.getKey() + " SystemVm template not found. Cannot upgrade system Vms");
|
|
} else {
|
|
s_logger.warn("4.5.0 " + hypervisorAndTemplateName.getKey() + " SystemVm template not found. " + hypervisorAndTemplateName.getKey() + " hypervisor is not used, so not failing upgrade");
|
|
// Update the latest template URLs for corresponding hypervisor
|
|
pstmt = conn.prepareStatement("UPDATE `cloud`.`vm_template` SET url = ? , checksum = ? WHERE hypervisor_type = ? AND type = 'SYSTEM' AND removed is null order by id desc limit 1");
|
|
pstmt.setString(1, newTemplateUrl.get(hypervisorAndTemplateName.getKey()));
|
|
pstmt.setString(2, newTemplateChecksum.get(hypervisorAndTemplateName.getKey()));
|
|
pstmt.setString(3, hypervisorAndTemplateName.getKey().toString());
|
|
pstmt.executeUpdate();
|
|
pstmt.close();
|
|
}
|
|
}
|
|
} catch (SQLException e) {
|
|
throw new CloudRuntimeException("Error while updating "+ hypervisorAndTemplateName.getKey() +" systemVm template", e);
|
|
}
|
|
}
|
|
s_logger.debug("Updating System Vm Template IDs Complete");
|
|
} finally {
|
|
try {
|
|
if (rs != null) {
|
|
rs.close();
|
|
}
|
|
|
|
if (pstmt != null) {
|
|
pstmt.close();
|
|
}
|
|
} catch (SQLException e) {
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
private void dropInvalidKeyFromStoragePoolTable(Connection conn) {
|
|
HashMap<String, List<String>> uniqueKeys = new HashMap<String, List<String>>();
|
|
List<String> keys = new ArrayList<String>();
|
|
|
|
keys.add("id_2");
|
|
uniqueKeys.put("storage_pool", keys);
|
|
|
|
s_logger.debug("Dropping id_2 key from storage_pool table");
|
|
for (Map.Entry<String, List<String>> entry: uniqueKeys.entrySet()) {
|
|
DbUpgradeUtils.dropKeysIfExist(conn,entry.getKey(), entry.getValue(), false);
|
|
}
|
|
}
|
|
|
|
private void dropDuplicatedForeignKeyFromAsyncJobTable(Connection conn) {
|
|
HashMap<String, List<String>> foreignKeys = new HashMap<String, List<String>>();
|
|
List<String> keys = new ArrayList<String>();
|
|
|
|
keys.add("fk_async_job_join_map__join_job_id");
|
|
foreignKeys.put("async_job_join_map", keys);
|
|
|
|
s_logger.debug("Dropping fk_async_job_join_map__join_job_id key from async_job_join_map table");
|
|
for (Map.Entry<String, List<String>> entry: foreignKeys.entrySet()) {
|
|
DbUpgradeUtils.dropKeysIfExist(conn,entry.getKey(), entry.getValue(), true);
|
|
}
|
|
}
|
|
}
|