1   /**
2    * Copyright (c) 2000-2009 Liferay, Inc. All rights reserved.
3    *
4    * Permission is hereby granted, free of charge, to any person obtaining a copy
5    * of this software and associated documentation files (the "Software"), to deal
6    * in the Software without restriction, including without limitation the rights
7    * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8    * copies of the Software, and to permit persons to whom the Software is
9    * furnished to do so, subject to the following conditions:
10   *
11   * The above copyright notice and this permission notice shall be included in
12   * all copies or substantial portions of the Software.
13   *
14   * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15   * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16   * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
17   * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
18   * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
19   * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
20   * SOFTWARE.
21   */
22  
23  package com.liferay.portal.verify;
24  
25  import com.liferay.portal.kernel.dao.jdbc.DataAccess;
26  import com.liferay.portal.kernel.log.Log;
27  import com.liferay.portal.kernel.log.LogFactoryUtil;
28  import com.liferay.portal.tools.sql.DBUtil;
29  import com.liferay.portal.util.PropsValues;
30  
31  import java.sql.Connection;
32  import java.sql.PreparedStatement;
33  import java.sql.ResultSet;
34  
35  /**
36   * <a href="VerifyMySQL.java.html"><b><i>View Source</i></b></a>
37   *
38   * @author Brian Wing Shun Chan
39   *
40   */
41  public class VerifyMySQL extends VerifyProcess {
42  
43      public void verify() throws VerifyException {
44          _log.info("Verifying");
45  
46          try {
47              verifyMySQL();
48          }
49          catch (Exception e) {
50              throw new VerifyException(e);
51          }
52      }
53  
54      protected void alterTableEngine(String tableName) throws Exception {
55          _log.info(
56              "Updating table " + tableName + " to use engine " +
57                  PropsValues.DATABASE_MYSQL_ENGINE);
58  
59          Connection con = null;
60          PreparedStatement ps = null;
61  
62          try {
63              con = DataAccess.getConnection();
64  
65              ps = con.prepareStatement(
66                  "alter table " + tableName + " engine " +
67                      PropsValues.DATABASE_MYSQL_ENGINE);
68  
69              ps.executeUpdate();
70          }
71          finally {
72              DataAccess.cleanUp(con, ps);
73          }
74      }
75  
76      protected void verifyMySQL() throws Exception {
77          DBUtil dbUtil = DBUtil.getInstance();
78  
79          if (!dbUtil.getType().equals(DBUtil.TYPE_MYSQL)) {
80              return;
81          }
82  
83          Connection con = null;
84          PreparedStatement ps = null;
85          ResultSet rs = null;
86  
87          try {
88              con = DataAccess.getConnection();
89  
90              ps = con.prepareStatement("show table status");
91  
92              rs = ps.executeQuery();
93  
94              while (rs.next()) {
95                  String tableName = rs.getString("Name");
96                  String engine = rs.getString("Engine");
97  
98                  if (!engine.equalsIgnoreCase(
99                          PropsValues.DATABASE_MYSQL_ENGINE)) {
100 
101                     alterTableEngine(tableName);
102                 }
103             }
104         }
105         finally {
106             DataAccess.cleanUp(con, ps, rs);
107         }
108     }
109 
110     private static Log _log = LogFactoryUtil.getLog(VerifyMySQL.class);
111 
112 }