001    /**
002     * Copyright (c) 2000-2010 Liferay, Inc. All rights reserved.
003     *
004     * This library is free software; you can redistribute it and/or modify it under
005     * the terms of the GNU Lesser General Public License as published by the Free
006     * Software Foundation; either version 2.1 of the License, or (at your option)
007     * any later version.
008     *
009     * This library is distributed in the hope that it will be useful, but WITHOUT
010     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
011     * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
012     * details.
013     */
014    
015    package com.liferay.portal.upgrade.v4_3_0;
016    
017    import com.liferay.portal.kernel.upgrade.UpgradeProcess;
018    import com.liferay.portal.kernel.upgrade.util.DefaultPKMapper;
019    import com.liferay.portal.kernel.upgrade.util.SwapUpgradeColumnImpl;
020    import com.liferay.portal.kernel.upgrade.util.UpgradeColumn;
021    import com.liferay.portal.kernel.upgrade.util.UpgradeTable;
022    import com.liferay.portal.kernel.upgrade.util.UpgradeTableFactoryUtil;
023    import com.liferay.portal.kernel.upgrade.util.ValueMapper;
024    import com.liferay.portal.upgrade.util.PKUpgradeColumnImpl;
025    import com.liferay.portal.upgrade.v4_3_0.util.AvailableMappersUtil;
026    import com.liferay.portal.upgrade.v4_3_0.util.IGFolderTable;
027    import com.liferay.portal.upgrade.v4_3_0.util.IGImageIdUpgradeColumnImpl;
028    import com.liferay.portal.upgrade.v4_3_0.util.IGImageTable;
029    import com.liferay.portal.upgrade.v4_3_0.util.IGLargeImageIdUpgradeColumnImpl;
030    import com.liferay.portal.upgrade.v4_3_0.util.IGSmallImageIdUpgradeColumnImpl;
031    
032    import java.sql.Types;
033    
034    /**
035     * @author Brian Wing Shun Chan
036     */
037    public class UpgradeImageGallery extends UpgradeProcess {
038    
039            protected void doUpgrade() throws Exception {
040    
041                    // IGFolder
042    
043                    UpgradeColumn upgradeCompanyIdColumn = new SwapUpgradeColumnImpl(
044                            "companyId", new Integer(Types.VARCHAR),
045                            AvailableMappersUtil.getCompanyIdMapper());
046    
047                    UpgradeColumn upgradeGroupIdColumn = new SwapUpgradeColumnImpl(
048                            "groupId", AvailableMappersUtil.getGroupIdMapper());
049    
050                    UpgradeColumn upgradeUserIdColumn = new SwapUpgradeColumnImpl(
051                            "userId", new Integer(Types.VARCHAR),
052                            AvailableMappersUtil.getUserIdMapper());
053    
054                    PKUpgradeColumnImpl upgradePKColumn = new PKUpgradeColumnImpl(
055                            "folderId", true);
056    
057                    UpgradeTable upgradeTable = UpgradeTableFactoryUtil.getUpgradeTable(
058                            IGFolderTable.TABLE_NAME, IGFolderTable.TABLE_COLUMNS,
059                            upgradePKColumn, upgradeGroupIdColumn, upgradeUserIdColumn);
060    
061                    upgradeTable.setCreateSQL(IGFolderTable.TABLE_SQL_CREATE);
062    
063                    upgradeTable.updateTable();
064    
065                    ValueMapper folderIdMapper = new DefaultPKMapper(
066                            upgradePKColumn.getValueMapper());
067    
068                    AvailableMappersUtil.setIGFolderIdMapper(folderIdMapper);
069    
070                    UpgradeColumn upgradeParentFolderIdColumn = new SwapUpgradeColumnImpl(
071                            "parentFolderId", folderIdMapper);
072    
073                    upgradeTable = UpgradeTableFactoryUtil.getUpgradeTable(
074                            IGFolderTable.TABLE_NAME, IGFolderTable.TABLE_COLUMNS,
075                            upgradeParentFolderIdColumn);
076    
077                    upgradeTable.updateTable();
078    
079                    UpgradeColumn upgradeFolderIdColumn = new SwapUpgradeColumnImpl(
080                            "folderId", folderIdMapper);
081    
082                    // IGImage
083    
084                    PKUpgradeColumnImpl upgradeImageIdColumn =
085                            new IGImageIdUpgradeColumnImpl(upgradeCompanyIdColumn);
086    
087                    UpgradeColumn upgradeSmallImageIdColumn =
088                            new IGSmallImageIdUpgradeColumnImpl(
089                                    upgradeCompanyIdColumn, upgradeImageIdColumn,
090                                    AvailableMappersUtil.getImageIdMapper());
091    
092                    UpgradeColumn upgradeLargeImageIdColumn =
093                            new IGLargeImageIdUpgradeColumnImpl(
094                                    upgradeCompanyIdColumn, upgradeImageIdColumn,
095                                    AvailableMappersUtil.getImageIdMapper());
096    
097                    upgradeTable = UpgradeTableFactoryUtil.getUpgradeTable(
098                            IGImageTable.TABLE_NAME, IGImageTable.TABLE_COLUMNS,
099                            upgradeCompanyIdColumn, upgradeImageIdColumn, upgradeUserIdColumn,
100                            upgradeFolderIdColumn, upgradeSmallImageIdColumn,
101                            upgradeLargeImageIdColumn);
102    
103                    upgradeTable.setCreateSQL(IGImageTable.TABLE_SQL_CREATE);
104    
105                    upgradeTable.updateTable();
106    
107                    ValueMapper imageIdMapper = upgradeImageIdColumn.getValueMapper();
108    
109                    AvailableMappersUtil.setIGImageIdMapper(imageIdMapper);
110            }
111    
112    }