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.OrgGroupPermissionTable;
027    import com.liferay.portal.upgrade.v4_3_0.util.OrgLaborTable;
028    import com.liferay.portal.upgrade.v4_3_0.util.OrganizationTable;
029    import com.liferay.portal.upgrade.v4_3_0.util.ValueMapperUtil;
030    
031    /**
032     * @author Brian Wing Shun Chan
033     */
034    public class UpgradeOrganization extends UpgradeProcess {
035    
036            protected void doUpgrade() throws Exception {
037    
038                    // Organization
039    
040                    PKUpgradeColumnImpl upgradePKColumn = new PKUpgradeColumnImpl(
041                            "organizationId", true);
042    
043                    UpgradeTable upgradeTable = UpgradeTableFactoryUtil.getUpgradeTable(
044                            OrganizationTable.TABLE_NAME, OrganizationTable.TABLE_COLUMNS,
045                            upgradePKColumn);
046    
047                    upgradeTable.setCreateSQL(OrganizationTable.TABLE_SQL_CREATE);
048    
049                    upgradeTable.updateTable();
050    
051                    ValueMapper organizationIdMapper = new DefaultPKMapper(
052                            upgradePKColumn.getValueMapper());
053    
054                    AvailableMappersUtil.setOrganizationIdMapper(organizationIdMapper);
055    
056                    UpgradeColumn upgradeParentOrganizationIdColumn =
057                            new SwapUpgradeColumnImpl(
058                                    "parentOrganizationId", organizationIdMapper);
059    
060                    upgradeTable = UpgradeTableFactoryUtil.getUpgradeTable(
061                            OrganizationTable.TABLE_NAME, OrganizationTable.TABLE_COLUMNS,
062                            upgradeParentOrganizationIdColumn);
063    
064                    upgradeTable.updateTable();
065    
066                    UpgradeColumn upgradeOrganizationIdColumn = new SwapUpgradeColumnImpl(
067                            "organizationId", organizationIdMapper);
068    
069                    // OrgGroupPermission
070    
071                    upgradeTable = UpgradeTableFactoryUtil.getUpgradeTable(
072                            OrgGroupPermissionTable.TABLE_NAME,
073                            OrgGroupPermissionTable.TABLE_COLUMNS, upgradeOrganizationIdColumn);
074    
075                    upgradeTable.setCreateSQL(OrgGroupPermissionTable.TABLE_SQL_CREATE);
076    
077                    upgradeTable.updateTable();
078    
079                    // OrgLabor
080    
081                    upgradePKColumn = new PKUpgradeColumnImpl("orgLaborId", true);
082    
083                    upgradeTable = UpgradeTableFactoryUtil.getUpgradeTable(
084                            OrgLaborTable.TABLE_NAME, OrgLaborTable.TABLE_COLUMNS,
085                            upgradePKColumn, upgradeOrganizationIdColumn);
086    
087                    upgradeTable.setCreateSQL(OrgLaborTable.TABLE_SQL_CREATE);
088    
089                    upgradeTable.updateTable();
090    
091                    ValueMapperUtil.persist(
092                            upgradePKColumn.getValueMapper(), "org-labor-id");
093            }
094    
095    }