1   /**
2    * Copyright (c) 2000-2009 Liferay, Inc. All rights reserved.
3    *
4    * The contents of this file are subject to the terms of the Liferay Enterprise
5    * Subscription License ("License"). You may not use this file except in
6    * compliance with the License. You can obtain a copy of the License by
7    * contacting Liferay, Inc. See the License for the specific language governing
8    * permissions and limitations under the License, including but not limited to
9    * distribution rights of the Software.
10   *
11   * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
12   * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
13   * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
14   * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
15   * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
16   * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
17   * SOFTWARE.
18   */
19  
20  package com.liferay.portal.upgrade.v4_3_0;
21  
22  import com.liferay.portal.kernel.log.Log;
23  import com.liferay.portal.kernel.log.LogFactoryUtil;
24  import com.liferay.portal.upgrade.UpgradeException;
25  import com.liferay.portal.upgrade.UpgradeProcess;
26  import com.liferay.portal.upgrade.util.DefaultUpgradeTableImpl;
27  import com.liferay.portal.upgrade.util.PKUpgradeColumnImpl;
28  import com.liferay.portal.upgrade.util.SwapUpgradeColumnImpl;
29  import com.liferay.portal.upgrade.util.UpgradeColumn;
30  import com.liferay.portal.upgrade.util.UpgradeTable;
31  import com.liferay.portal.upgrade.v4_3_0.util.AvailableMappersUtil;
32  import com.liferay.portal.upgrade.v4_3_0.util.ClassNameIdUpgradeColumnImpl;
33  import com.liferay.portal.upgrade.v4_3_0.util.ClassPKContainer;
34  import com.liferay.portal.upgrade.v4_3_0.util.ClassPKUpgradeColumnImpl;
35  import com.liferay.portal.util.PortalUtil;
36  import com.liferay.portlet.documentlibrary.model.DLFileEntry;
37  import com.liferay.portlet.ratings.model.impl.RatingsEntryImpl;
38  import com.liferay.portlet.ratings.model.impl.RatingsStatsImpl;
39  
40  import java.sql.Types;
41  
42  import java.util.HashMap;
43  import java.util.Map;
44  
45  /**
46   * <a href="UpgradeRatings.java.html"><b><i>View Source</i></b></a>
47   *
48   * @author Brian Wing Shun Chan
49   *
50   */
51  public class UpgradeRatings extends UpgradeProcess {
52  
53      public void upgrade() throws UpgradeException {
54          _log.info("Upgrading");
55  
56          try {
57              doUpgrade();
58          }
59          catch (Exception e) {
60              throw new UpgradeException(e);
61          }
62      }
63  
64      protected void doUpgrade() throws Exception {
65  
66          // RatingsEntry
67  
68          UpgradeColumn upgradeUserIdColumn = new SwapUpgradeColumnImpl(
69              "userId", new Integer(Types.VARCHAR),
70              AvailableMappersUtil.getUserIdMapper());
71  
72          ClassNameIdUpgradeColumnImpl classNameIdColumn =
73              new ClassNameIdUpgradeColumnImpl();
74  
75          Map<Long, ClassPKContainer> classPKContainers =
76              new HashMap<Long, ClassPKContainer>();
77  
78          classPKContainers.put(
79              new Long(PortalUtil.getClassNameId(DLFileEntry.class.getName())),
80              new ClassPKContainer(
81                  AvailableMappersUtil.getDLFileEntryIdMapper(), false));
82  
83          UpgradeColumn upgradeClassPKColumn = new ClassPKUpgradeColumnImpl(
84              classNameIdColumn, classPKContainers);
85  
86          UpgradeTable upgradeTable = new DefaultUpgradeTableImpl(
87              RatingsEntryImpl.TABLE_NAME, RatingsEntryImpl.TABLE_COLUMNS,
88              new PKUpgradeColumnImpl("entryId", false), upgradeUserIdColumn,
89              classNameIdColumn, upgradeClassPKColumn);
90  
91          upgradeTable.setCreateSQL(RatingsEntryImpl.TABLE_SQL_CREATE);
92  
93          upgradeTable.updateTable();
94  
95          // RatingsStats
96  
97          upgradeTable = new DefaultUpgradeTableImpl(
98              RatingsStatsImpl.TABLE_NAME, RatingsStatsImpl.TABLE_COLUMNS,
99              new PKUpgradeColumnImpl("statsId", false), classNameIdColumn,
100             upgradeClassPKColumn);
101 
102         upgradeTable.setCreateSQL(RatingsStatsImpl.TABLE_SQL_CREATE);
103 
104         upgradeTable.updateTable();
105     }
106 
107     private static Log _log = LogFactoryUtil.getLog(UpgradeRatings.class);
108 
109 }