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.model.impl.OrgGroupPermissionImpl;
25  import com.liferay.portal.model.impl.OrgLaborImpl;
26  import com.liferay.portal.model.impl.OrganizationImpl;
27  import com.liferay.portal.upgrade.UpgradeException;
28  import com.liferay.portal.upgrade.UpgradeProcess;
29  import com.liferay.portal.upgrade.util.DefaultPKMapper;
30  import com.liferay.portal.upgrade.util.DefaultUpgradeTableImpl;
31  import com.liferay.portal.upgrade.util.PKUpgradeColumnImpl;
32  import com.liferay.portal.upgrade.util.SwapUpgradeColumnImpl;
33  import com.liferay.portal.upgrade.util.UpgradeColumn;
34  import com.liferay.portal.upgrade.util.UpgradeTable;
35  import com.liferay.portal.upgrade.util.ValueMapper;
36  import com.liferay.portal.upgrade.v4_3_0.util.AvailableMappersUtil;
37  import com.liferay.portal.upgrade.v4_3_0.util.ValueMapperUtil;
38  
39  /**
40   * <a href="UpgradeOrganization.java.html"><b><i>View Source</i></b></a>
41   *
42   * @author Brian Wing Shun Chan
43   *
44   */
45  public class UpgradeOrganization extends UpgradeProcess {
46  
47      public void upgrade() throws UpgradeException {
48          _log.info("Upgrading");
49  
50          try {
51              doUpgrade();
52          }
53          catch (Exception e) {
54              throw new UpgradeException(e);
55          }
56      }
57  
58      protected void doUpgrade() throws Exception {
59  
60          // Organization
61  
62          PKUpgradeColumnImpl upgradePKColumn = new PKUpgradeColumnImpl(
63              "organizationId", true);
64  
65          UpgradeTable upgradeTable = new DefaultUpgradeTableImpl(
66              OrganizationImpl.TABLE_NAME, OrganizationImpl.TABLE_COLUMNS,
67              upgradePKColumn);
68  
69          upgradeTable.setCreateSQL(OrganizationImpl.TABLE_SQL_CREATE);
70  
71          upgradeTable.updateTable();
72  
73          ValueMapper organizationIdMapper = new DefaultPKMapper(
74              upgradePKColumn.getValueMapper());
75  
76          AvailableMappersUtil.setOrganizationIdMapper(organizationIdMapper);
77  
78          UpgradeColumn upgradeParentOrganizationIdColumn =
79              new SwapUpgradeColumnImpl(
80                  "parentOrganizationId", organizationIdMapper);
81  
82          upgradeTable = new DefaultUpgradeTableImpl(
83              OrganizationImpl.TABLE_NAME, OrganizationImpl.TABLE_COLUMNS,
84              upgradeParentOrganizationIdColumn);
85  
86          upgradeTable.updateTable();
87  
88          UpgradeColumn upgradeOrganizationIdColumn = new SwapUpgradeColumnImpl(
89              "organizationId", organizationIdMapper);
90  
91          // OrgGroupPermission
92  
93          upgradeTable = new DefaultUpgradeTableImpl(
94              OrgGroupPermissionImpl.TABLE_NAME,
95              OrgGroupPermissionImpl.TABLE_COLUMNS, upgradeOrganizationIdColumn);
96  
97          upgradeTable.setCreateSQL(OrgGroupPermissionImpl.TABLE_SQL_CREATE);
98  
99          upgradeTable.updateTable();
100 
101         // OrgLabor
102 
103         upgradePKColumn = new PKUpgradeColumnImpl("orgLaborId", true);
104 
105         upgradeTable = new DefaultUpgradeTableImpl(
106             OrgLaborImpl.TABLE_NAME, OrgLaborImpl.TABLE_COLUMNS,
107             upgradePKColumn, upgradeOrganizationIdColumn);
108 
109         upgradeTable.setCreateSQL(OrgLaborImpl.TABLE_SQL_CREATE);
110 
111         upgradeTable.updateTable();
112 
113         ValueMapperUtil.persist(
114             upgradePKColumn.getValueMapper(), "org-labor-id");
115     }
116 
117     private static Log _log = LogFactoryUtil.getLog(UpgradeOrganization.class);
118 
119 }