From 310e99087ff8f46ac903f2a28169f3ffe2d13ebd Mon Sep 17 00:00:00 2001
From: Edison Su <sudison@gmail.com>
Date: Mon, 25 Feb 2013 10:45:04 -0800
Subject: [PATCH] CLOUDSTACK-1380: add db upgrade for vm_template, in 4.1 to
 4.2 upgrade procedure

---
 .../src/com/cloud/upgrade/dao/Upgrade410to420.java | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/server/src/com/cloud/upgrade/dao/Upgrade410to420.java b/server/src/com/cloud/upgrade/dao/Upgrade410to420.java
index a0592579a43..9000e15f7aa 100644
--- a/server/src/com/cloud/upgrade/dao/Upgrade410to420.java
+++ b/server/src/com/cloud/upgrade/dao/Upgrade410to420.java
@@ -59,6 +59,20 @@ public class Upgrade410to420 implements DbUpgrade {
 
 	@Override
 	public void performDataMigration(Connection conn) {
+	    PreparedStatement sql = null;
+	    try {
+	        sql = conn.prepareStatement("update vm_template set image_data_store_id = 1 where type = 'SYSTEM' or type = 'BUILTIN'");
+	        sql.executeUpdate();
+        } catch (SQLException e) {
+            throw new CloudRuntimeException("Failed to upgrade vm template data store uuid: " + e.toString());
+        } finally {
+            if (sql != null) {
+                try {
+                    sql.close();
+                } catch (SQLException e) {
+                }
+            }
+        }
 	}
 
 	@Override