1   /**
2    * Copyright (c) 2000-2010 Liferay, Inc. All rights reserved.
3    *
4    * This library is free software; you can redistribute it and/or modify it under
5    * the terms of the GNU Lesser General Public License as published by the Free
6    * Software Foundation; either version 2.1 of the License, or (at your option)
7    * any later version.
8    *
9    * This library is distributed in the hope that it will be useful, but WITHOUT
10   * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
11   * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
12   * details.
13   */
14  
15  package com.liferay.portal.upgrade.v4_4_0;
16  
17  import com.liferay.portal.kernel.upgrade.UpgradeProcess;
18  import com.liferay.portal.kernel.upgrade.util.BaseUpgradeColumnImpl;
19  import com.liferay.portal.kernel.upgrade.util.TempUpgradeColumnImpl;
20  import com.liferay.portal.kernel.upgrade.util.UpgradeColumn;
21  import com.liferay.portal.kernel.upgrade.util.UpgradeTable;
22  import com.liferay.portal.kernel.upgrade.util.UpgradeTableFactoryUtil;
23  import com.liferay.portal.upgrade.v4_4_0.util.DLFileEntryTable;
24  import com.liferay.portal.upgrade.v4_4_0.util.DLFileEntryTitleColumnImpl;
25  import com.liferay.portal.upgrade.v4_4_0.util.DLFolderNameColumnImpl;
26  import com.liferay.portal.upgrade.v4_4_0.util.DLFolderTable;
27  
28  import java.util.Set;
29  
30  /**
31   * <a href="UpgradeDocumentLibrary.java.html"><b><i>View Source</i></b></a>
32   *
33   * @author Alexander Chow
34   */
35  public class UpgradeDocumentLibrary extends UpgradeProcess {
36  
37      protected void doUpgrade() throws Exception {
38  
39          // DLFolder
40  
41          UpgradeColumn groupIdColumn = new TempUpgradeColumnImpl("groupId");
42  
43          UpgradeColumn parentFolderIdColumn = new TempUpgradeColumnImpl(
44              "parentFolderId");
45  
46          DLFolderNameColumnImpl dlFolderNameColumn = new DLFolderNameColumnImpl(
47              groupIdColumn, parentFolderIdColumn);
48  
49          UpgradeTable upgradeTable = UpgradeTableFactoryUtil.getUpgradeTable(
50              DLFolderTable.TABLE_NAME, DLFolderTable.TABLE_COLUMNS,
51              groupIdColumn, parentFolderIdColumn, dlFolderNameColumn);
52  
53          upgradeTable.updateTable();
54  
55          Set<String> distinctNames = dlFolderNameColumn.getDistintNames();
56  
57          // DLFileEntry
58  
59          UpgradeColumn folderIdColumn = new TempUpgradeColumnImpl("folderId");
60  
61          UpgradeColumn nameColumn = new TempUpgradeColumnImpl("name");
62  
63          BaseUpgradeColumnImpl dlFileEntryTitleColumn =
64              new DLFileEntryTitleColumnImpl(
65                  groupIdColumn, folderIdColumn, nameColumn, distinctNames);
66  
67          upgradeTable = UpgradeTableFactoryUtil.getUpgradeTable(
68              DLFileEntryTable.TABLE_NAME, DLFileEntryTable.TABLE_COLUMNS,
69              folderIdColumn, nameColumn, dlFileEntryTitleColumn);
70  
71          upgradeTable.updateTable();
72      }
73  
74  }