001
014
015 package com.liferay.portlet.bookmarks.lar;
016
017 import com.liferay.portal.kernel.lar.BaseStagedModelDataHandler;
018 import com.liferay.portal.kernel.lar.ExportImportPathUtil;
019 import com.liferay.portal.kernel.lar.PortletDataContext;
020 import com.liferay.portal.kernel.util.MapUtil;
021 import com.liferay.portal.kernel.xml.Element;
022 import com.liferay.portal.service.ServiceContext;
023 import com.liferay.portlet.bookmarks.model.BookmarksFolder;
024 import com.liferay.portlet.bookmarks.model.BookmarksFolderConstants;
025 import com.liferay.portlet.bookmarks.service.BookmarksFolderLocalServiceUtil;
026 import com.liferay.portlet.bookmarks.service.persistence.BookmarksFolderUtil;
027
028 import java.util.Map;
029
030
034 public class BookmarksFolderStagedModelDataHandler
035 extends BaseStagedModelDataHandler<BookmarksFolder> {
036
037 public static final String[] CLASS_NAMES =
038 {BookmarksFolder.class.getName()};
039
040 @Override
041 public String[] getClassNames() {
042 return CLASS_NAMES;
043 }
044
045 @Override
046 protected void doExportStagedModel(
047 PortletDataContext portletDataContext, BookmarksFolder folder)
048 throws Exception {
049
050 if (folder.getParentFolderId() !=
051 BookmarksFolderConstants.DEFAULT_PARENT_FOLDER_ID) {
052
053 exportStagedModel(portletDataContext, folder.getParentFolder());
054 }
055
056 Element folderElement =
057 portletDataContext.getExportDataStagedModelElement(folder);
058
059 portletDataContext.addClassedModel(
060 folderElement, ExportImportPathUtil.getModelPath(folder), folder,
061 BookmarksPortletDataHandler.NAMESPACE);
062 }
063
064 @Override
065 protected void doImportStagedModel(
066 PortletDataContext portletDataContext, BookmarksFolder folder)
067 throws Exception {
068
069 long userId = portletDataContext.getUserId(folder.getUserUuid());
070
071 Map<Long, Long> folderIds =
072 (Map<Long, Long>)portletDataContext.getNewPrimaryKeysMap(
073 BookmarksFolder.class);
074
075 long parentFolderId = MapUtil.getLong(
076 folderIds, folder.getParentFolderId(), folder.getParentFolderId());
077
078 if ((parentFolderId !=
079 BookmarksFolderConstants.DEFAULT_PARENT_FOLDER_ID) &&
080 (parentFolderId == folder.getParentFolderId())) {
081
082 String parentFolderPath = ExportImportPathUtil.getModelPath(
083 portletDataContext, BookmarksFolder.class.getName(),
084 parentFolderId);
085
086 BookmarksFolder parentFolder =
087 (BookmarksFolder)portletDataContext.getZipEntryAsObject(
088 parentFolderPath);
089
090 importStagedModel(portletDataContext, parentFolder);
091
092 parentFolderId = MapUtil.getLong(
093 folderIds, folder.getParentFolderId(),
094 folder.getParentFolderId());
095 }
096
097 ServiceContext serviceContext = portletDataContext.createServiceContext(
098 folder, BookmarksPortletDataHandler.NAMESPACE);
099
100 BookmarksFolder importedFolder = null;
101
102 if (portletDataContext.isDataStrategyMirror()) {
103 BookmarksFolder existingFolder = BookmarksFolderUtil.fetchByUUID_G(
104 folder.getUuid(), portletDataContext.getScopeGroupId());
105
106 if (existingFolder == null) {
107 serviceContext.setUuid(folder.getUuid());
108
109 importedFolder = BookmarksFolderLocalServiceUtil.addFolder(
110 userId, parentFolderId, folder.getName(),
111 folder.getDescription(), serviceContext);
112 }
113 else {
114 importedFolder = BookmarksFolderLocalServiceUtil.updateFolder(
115 userId, existingFolder.getFolderId(), parentFolderId,
116 folder.getName(), folder.getDescription(), false,
117 serviceContext);
118 }
119 }
120 else {
121 importedFolder = BookmarksFolderLocalServiceUtil.addFolder(
122 userId, parentFolderId, folder.getName(),
123 folder.getDescription(), serviceContext);
124 }
125
126 portletDataContext.importClassedModel(
127 folder, importedFolder, BookmarksPortletDataHandler.NAMESPACE);
128 }
129
130 }