// Copyright 2012 Citrix Systems, Inc. Licensed under the
// Apache License, Version 2.0 (the "License"); you may not use this
// file except in compliance with the License. Citrix Systems, Inc.
// reserves all rights not expressly granted by 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.
//
// Automatically generated by addcopyright.py at 04/03/2012
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 org.apache.log4j.Logger;
import com.cloud.utils.exception.CloudRuntimeException;
import com.cloud.utils.script.Script;
public class Upgrade227to228 implements DbUpgrade {
final static Logger s_logger = Logger.getLogger(Upgrade227to228.class);
@Override
public String[] getUpgradableVersionRange() {
return new String[] { "2.2.6", "2.2.7"};
}
@Override
public String getUpgradedVersion() {
return "2.2.8";
}
@Override
public boolean supportsRollingUpgrade() {
return true;
}
@Override
public File[] getPrepareScripts() {
String script = Script.findScript("", "db/schema-227to228.sql");
if (script == null) {
throw new CloudRuntimeException("Unable to find db/schema-227to228.sql");
}
return new File[] { new File(script) };
}
@Override
public void performDataMigration(Connection conn) {
try {
PreparedStatement pstmt = conn.prepareStatement("select id from data_center");
ResultSet rs = pstmt.executeQuery();
while (rs.next()) {
long dcId = rs.getLong(1);
pstmt = conn.prepareStatement("select id from host where data_center_id=? and type='SecondaryStorage'");
pstmt.setLong(1, dcId);
ResultSet rs1 = pstmt.executeQuery();
if (rs1.next()) {
long secHostId = rs1.getLong(1);
pstmt = conn.prepareStatement("update snapshots set sechost_id=? where data_center_id=?");
pstmt.setLong(1, secHostId);
pstmt.setLong(2, dcId);
pstmt.executeUpdate();
}
}
pstmt = conn.prepareStatement("update disk_offering set disk_size = disk_size * 1024 * 1024 where disk_size <= 2 * 1024 * 1024 and disk_size != 0");
pstmt.executeUpdate();
} catch (SQLException e) {
s_logger.error("Failed to DB migration for multiple secondary storages", e);
throw new CloudRuntimeException("Failed to DB migration for multiple secondary storages", e);
}
updateDomainLevelNetworks(conn);
updateVolumeUsageRecords(conn);
}
@Override
public File[] getCleanupScripts() {
return null;
}
private void updateDomainLevelNetworks(Connection conn) {
s_logger.debug("Updating domain level specific networks...");
try {
PreparedStatement pstmt = conn.prepareStatement("SELECT n.id FROM networks n, network_offerings o WHERE n.shared=1 AND o.system_only=0 AND o.id=n.network_offering_id");
ResultSet rs = pstmt.executeQuery();
ArrayList<Object[]> networks = new ArrayList<Object[]>();
while (rs.next()) {
Object[] network = new Object[10];
network[0] = rs.getLong(1); // networkId
networks.add(network);
}
rs.close();
pstmt.close();
for (Object[] network : networks) {
Long networkId = (Long) network[0];
pstmt = conn.prepareStatement("SELECT * from domain_network_ref where network_id=?");
pstmt.setLong(1, networkId);
rs = pstmt.executeQuery();
if (rs.next()) {
s_logger.debug("Setting network id=" + networkId + " as domain specific shared network");
pstmt = conn.prepareStatement("UPDATE networks set is_domain_specific=1 where id=?");
pstmt.setLong(1, networkId);
pstmt.executeUpdate();
}
rs.close();
pstmt.close();
}
s_logger.debug("Successfully updated domain level specific networks");
} catch (SQLException e) {
s_logger.error("Failed to set domain specific shared networks due to ", e);
throw new CloudRuntimeException("Failed to set domain specific shared networks due to ", e);
}
}
//this method inserts missing volume.delete events (events were missing when vm failed to create)
private void updateVolumeUsageRecords(Connection conn) {
try {
s_logger.debug("Inserting missing usage_event records for destroyed volumes...");
PreparedStatement pstmt = conn.prepareStatement("select id, account_id, data_center_id, name from volumes where state='Destroy' and id in (select resource_id from usage_event where type='volume.create') and id not in (select resource_id from usage_event where type='volume.delete')");
ResultSet rs = pstmt.executeQuery();
while (rs.next()) {
long volumeId = rs.getLong(1);
long accountId = rs.getLong(2);
long zoneId = rs.getLong(3);
String volumeName = rs.getString(4);
pstmt = conn.prepareStatement("insert into usage_event (type, account_id, created, zone_id, resource_name, resource_id) values ('VOLUME.DELETE', ?, now(), ?, ?, ?)");
pstmt.setLong(1, accountId);
pstmt.setLong(2, zoneId);
pstmt.setString(3, volumeName);
pstmt.setLong(4, volumeId);
pstmt.executeUpdate();
}
s_logger.debug("Successfully inserted missing usage_event records for destroyed volumes");
} catch (SQLException e) {
s_logger.error("Failed to insert missing delete usage records ", e);
throw new CloudRuntimeException("Failed to insert missing delete usage records ", e);
}
}
}