1   /**
2    * Copyright (c) 2000-2009 Liferay, Inc. All rights reserved.
3    *
4    *
5    *
6    *
7    * The contents of this file are subject to the terms of the Liferay Enterprise
8    * Subscription License ("License"). You may not use this file except in
9    * compliance with the License. You can obtain a copy of the License by
10   * contacting Liferay, Inc. See the License for the specific language governing
11   * permissions and limitations under the License, including but not limited to
12   * distribution rights 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.upgrade.v4_3_0;
24  
25  import com.liferay.portal.upgrade.UpgradeProcess;
26  import com.liferay.portal.upgrade.util.DefaultPKMapper;
27  import com.liferay.portal.upgrade.util.DefaultUpgradeTableImpl;
28  import com.liferay.portal.upgrade.util.PKUpgradeColumnImpl;
29  import com.liferay.portal.upgrade.util.SwapUpgradeColumnImpl;
30  import com.liferay.portal.upgrade.util.TempUpgradeColumnImpl;
31  import com.liferay.portal.upgrade.util.UpgradeColumn;
32  import com.liferay.portal.upgrade.util.UpgradeTable;
33  import com.liferay.portal.upgrade.util.ValueMapper;
34  import com.liferay.portal.upgrade.v4_3_0.util.AvailableMappersUtil;
35  import com.liferay.portal.upgrade.v4_3_0.util.DLFileEntryIdUpgradeColumnImpl;
36  import com.liferay.portal.upgrade.v4_3_0.util.DLFileEntryTable;
37  import com.liferay.portal.upgrade.v4_3_0.util.DLFileRankTable;
38  import com.liferay.portal.upgrade.v4_3_0.util.DLFileShortcutTable;
39  import com.liferay.portal.upgrade.v4_3_0.util.DLFileVersionTable;
40  import com.liferay.portal.upgrade.v4_3_0.util.DLFolderTable;
41  
42  import java.sql.Types;
43  
44  /**
45   * <a href="UpgradeDocumentLibrary.java.html"><b><i>View Source</i></b></a>
46   *
47   * @author Brian Wing Shun Chan
48   */
49  public class UpgradeDocumentLibrary extends UpgradeProcess {
50  
51      protected void doUpgrade() throws Exception {
52  
53          // DLFolder
54  
55          UpgradeColumn upgradeCompanyIdColumn = new SwapUpgradeColumnImpl(
56              "companyId", new Integer(Types.VARCHAR),
57              AvailableMappersUtil.getCompanyIdMapper());
58  
59          UpgradeColumn upgradeGroupIdColumn = new SwapUpgradeColumnImpl(
60              "groupId", AvailableMappersUtil.getGroupIdMapper());
61  
62          UpgradeColumn upgradeUserIdColumn = new SwapUpgradeColumnImpl(
63              "userId", new Integer(Types.VARCHAR),
64              AvailableMappersUtil.getUserIdMapper());
65  
66          PKUpgradeColumnImpl upgradePKColumn = new PKUpgradeColumnImpl(
67              "folderId", true);
68  
69          UpgradeTable upgradeTable = new DefaultUpgradeTableImpl(
70              DLFolderTable.TABLE_NAME, DLFolderTable.TABLE_COLUMNS,
71              upgradePKColumn, upgradeGroupIdColumn, upgradeUserIdColumn);
72  
73          upgradeTable.setCreateSQL(DLFolderTable.TABLE_SQL_CREATE);
74  
75          upgradeTable.updateTable();
76  
77          ValueMapper folderIdMapper = new DefaultPKMapper(
78              upgradePKColumn.getValueMapper());
79  
80          AvailableMappersUtil.setDLFolderIdMapper(folderIdMapper);
81  
82          UpgradeColumn upgradeParentFolderIdColumn = new SwapUpgradeColumnImpl(
83              "parentFolderId", folderIdMapper);
84  
85          upgradeTable = new DefaultUpgradeTableImpl(
86              DLFolderTable.TABLE_NAME, DLFolderTable.TABLE_COLUMNS,
87              upgradeParentFolderIdColumn);
88  
89          upgradeTable.updateTable();
90  
91          UpgradeColumn upgradeFolderIdColumn = new SwapUpgradeColumnImpl(
92              "folderId", folderIdMapper);
93  
94          UpgradeColumn upgradeToFolderIdColumn = new SwapUpgradeColumnImpl(
95              "toFolderId", folderIdMapper);
96  
97          // DLFileEntry
98  
99          UpgradeColumn upgradeNameColumn = new TempUpgradeColumnImpl("name");
100 
101         PKUpgradeColumnImpl fileEntryIdColumn =
102             new DLFileEntryIdUpgradeColumnImpl(
103                 upgradeCompanyIdColumn, upgradeFolderIdColumn,
104                 upgradeNameColumn);
105 
106         UpgradeColumn upgradeVersionUserIdColumn = new SwapUpgradeColumnImpl(
107             "versionUserId", new Integer(Types.VARCHAR),
108             AvailableMappersUtil.getUserIdMapper());
109 
110         upgradeTable = new DefaultUpgradeTableImpl(
111             DLFileEntryTable.TABLE_NAME, DLFileEntryTable.TABLE_COLUMNS,
112             upgradeCompanyIdColumn, upgradeFolderIdColumn, upgradeNameColumn,
113             fileEntryIdColumn, upgradeUserIdColumn, upgradeVersionUserIdColumn);
114 
115         upgradeTable.setCreateSQL(DLFileEntryTable.TABLE_SQL_CREATE);
116 
117         upgradeTable.updateTable();
118 
119         ValueMapper fileEntryIdMapper = fileEntryIdColumn.getValueMapper();
120 
121         AvailableMappersUtil.setDLFileEntryIdMapper(fileEntryIdMapper);
122 
123         // DLFileRank
124 
125         upgradeTable = new DefaultUpgradeTableImpl(
126             DLFileRankTable.TABLE_NAME, DLFileRankTable.TABLE_COLUMNS,
127             new PKUpgradeColumnImpl("fileRankId", false),
128             upgradeUserIdColumn, upgradeFolderIdColumn);
129 
130         upgradeTable.setCreateSQL(DLFileRankTable.TABLE_SQL_CREATE);
131 
132         upgradeTable.updateTable();
133 
134         // DLFileShortcut
135 
136         upgradePKColumn = new PKUpgradeColumnImpl("fileShortcutId", true);
137 
138         upgradeTable = new DefaultUpgradeTableImpl(
139             DLFileShortcutTable.TABLE_NAME, DLFileShortcutTable.TABLE_COLUMNS,
140             upgradePKColumn, upgradeUserIdColumn, upgradeFolderIdColumn,
141             upgradeToFolderIdColumn);
142 
143         upgradeTable.setCreateSQL(DLFileShortcutTable.TABLE_SQL_CREATE);
144 
145         upgradeTable.updateTable();
146 
147         ValueMapper fileShortcutIdMapper = upgradePKColumn.getValueMapper();
148 
149         AvailableMappersUtil.setDLFileShortcutIdMapper(fileShortcutIdMapper);
150 
151         // DLFileVersion
152 
153         upgradeTable = new DefaultUpgradeTableImpl(
154             DLFileVersionTable.TABLE_NAME, DLFileVersionTable.TABLE_COLUMNS,
155             new PKUpgradeColumnImpl("fileVersionId", false),
156             upgradeUserIdColumn, upgradeFolderIdColumn);
157 
158         upgradeTable.setCreateSQL(DLFileVersionTable.TABLE_SQL_CREATE);
159 
160         upgradeTable.updateTable();
161     }
162 
163 }