001
014
015 package com.liferay.portlet.bookmarks.util;
016
017 import com.liferay.portal.kernel.dao.orm.ActionableDynamicQuery;
018 import com.liferay.portal.kernel.dao.orm.DynamicQuery;
019 import com.liferay.portal.kernel.dao.orm.Property;
020 import com.liferay.portal.kernel.dao.orm.PropertyFactoryUtil;
021 import com.liferay.portal.kernel.exception.PortalException;
022 import com.liferay.portal.kernel.exception.SystemException;
023 import com.liferay.portal.kernel.search.BaseIndexer;
024 import com.liferay.portal.kernel.search.BooleanQuery;
025 import com.liferay.portal.kernel.search.Document;
026 import com.liferay.portal.kernel.search.Field;
027 import com.liferay.portal.kernel.search.SearchContext;
028 import com.liferay.portal.kernel.search.SearchEngineUtil;
029 import com.liferay.portal.kernel.search.Summary;
030 import com.liferay.portal.kernel.util.CharPool;
031 import com.liferay.portal.kernel.util.GetterUtil;
032 import com.liferay.portal.kernel.util.StringUtil;
033 import com.liferay.portal.kernel.workflow.WorkflowConstants;
034 import com.liferay.portal.model.Group;
035 import com.liferay.portal.service.persistence.GroupActionableDynamicQuery;
036 import com.liferay.portal.util.PortletKeys;
037 import com.liferay.portlet.bookmarks.model.BookmarksEntry;
038 import com.liferay.portlet.bookmarks.model.BookmarksFolder;
039 import com.liferay.portlet.bookmarks.model.BookmarksFolderConstants;
040 import com.liferay.portlet.bookmarks.service.BookmarksEntryLocalServiceUtil;
041 import com.liferay.portlet.bookmarks.service.persistence.BookmarksEntryActionableDynamicQuery;
042 import com.liferay.portlet.bookmarks.service.persistence.BookmarksFolderActionableDynamicQuery;
043
044 import java.util.ArrayList;
045 import java.util.Collection;
046 import java.util.Locale;
047
048 import javax.portlet.PortletURL;
049
050
055 public class BookmarksEntryIndexer extends BaseIndexer {
056
057 public static final String[] CLASS_NAMES = {BookmarksEntry.class.getName()};
058
059 public static final String PORTLET_ID = PortletKeys.BOOKMARKS;
060
061 public BookmarksEntryIndexer() {
062 setPermissionAware(true);
063 }
064
065 @Override
066 public String[] getClassNames() {
067 return CLASS_NAMES;
068 }
069
070 @Override
071 public String getPortletId() {
072 return PORTLET_ID;
073 }
074
075 @Override
076 public void postProcessContextQuery(
077 BooleanQuery contextQuery, SearchContext searchContext)
078 throws Exception {
079
080 addStatus(contextQuery, searchContext);
081 }
082
083 @Override
084 protected void doDelete(Object obj) throws Exception {
085 BookmarksEntry entry = (BookmarksEntry)obj;
086
087 deleteDocument(entry.getCompanyId(), entry.getEntryId());
088 }
089
090 @Override
091 protected Document doGetDocument(Object obj) throws Exception {
092 BookmarksEntry entry = (BookmarksEntry)obj;
093
094 Document document = getBaseModelDocument(PORTLET_ID, entry);
095
096 document.addText(Field.DESCRIPTION, entry.getDescription());
097 document.addKeyword(Field.FOLDER_ID, entry.getFolderId());
098 document.addText(Field.TITLE, entry.getName());
099 document.addKeyword(
100 Field.TREE_PATH,
101 StringUtil.split(entry.getTreePath(), CharPool.SLASH));
102 document.addText(Field.URL, entry.getUrl());
103
104 return document;
105 }
106
107 @Override
108 protected Summary doGetSummary(
109 Document document, Locale locale, String snippet,
110 PortletURL portletURL) {
111
112 String entryId = document.get(Field.ENTRY_CLASS_PK);
113
114 portletURL.setParameter("struts_action", "/bookmarks/view_entry");
115 portletURL.setParameter("entryId", entryId);
116
117 Summary summary = createSummary(document, Field.TITLE, Field.URL);
118
119 summary.setPortletURL(portletURL);
120
121 return summary;
122 }
123
124 @Override
125 protected void doReindex(Object obj) throws Exception {
126 BookmarksEntry entry = (BookmarksEntry)obj;
127
128 Document document = getDocument(entry);
129
130 SearchEngineUtil.updateDocument(
131 getSearchEngineId(), entry.getCompanyId(), document);
132 }
133
134 @Override
135 protected void doReindex(String className, long classPK) throws Exception {
136 BookmarksEntry entry = BookmarksEntryLocalServiceUtil.getEntry(classPK);
137
138 doReindex(entry);
139 }
140
141 @Override
142 protected void doReindex(String[] ids) throws Exception {
143 long companyId = GetterUtil.getLong(ids[0]);
144
145 reindexFolders(companyId);
146 reindexRoot(companyId);
147 }
148
149 @Override
150 protected String getPortletId(SearchContext searchContext) {
151 return PORTLET_ID;
152 }
153
154 protected void reindexEntries(
155 long companyId, final long groupId, final long folderId)
156 throws PortalException, SystemException {
157
158 final Collection<Document> documents = new ArrayList<Document>();
159
160 ActionableDynamicQuery actionableDynamicQuery =
161 new BookmarksEntryActionableDynamicQuery() {
162
163 @Override
164 protected void addCriteria(DynamicQuery dynamicQuery) {
165 Property folderIdProperty = PropertyFactoryUtil.forName(
166 "folderId");
167
168 dynamicQuery.add(folderIdProperty.eq(folderId));
169
170 Property statusProperty = PropertyFactoryUtil.forName("status");
171
172 Integer[] statuses = {
173 WorkflowConstants.STATUS_APPROVED,
174 WorkflowConstants.STATUS_IN_TRASH
175 };
176
177 dynamicQuery.add(statusProperty.in(statuses));
178 }
179
180 @Override
181 protected void performAction(Object object) throws PortalException {
182 BookmarksEntry entry = (BookmarksEntry)object;
183
184 Document document = getDocument(entry);
185
186 documents.add(document);
187 }
188
189 };
190
191 actionableDynamicQuery.setGroupId(groupId);
192
193 actionableDynamicQuery.performActions();
194
195 SearchEngineUtil.updateDocuments(
196 getSearchEngineId(), companyId, documents);
197 }
198
199 protected void reindexFolders(final long companyId)
200 throws PortalException, SystemException {
201
202 ActionableDynamicQuery actionableDynamicQuery =
203 new BookmarksFolderActionableDynamicQuery() {
204
205 @Override
206 protected void performAction(Object object)
207 throws PortalException, SystemException {
208
209 BookmarksFolder folder = (BookmarksFolder)object;
210
211 long groupId = folder.getGroupId();
212 long folderId = folder.getFolderId();
213
214 reindexEntries(companyId, groupId, folderId);
215 }
216
217 };
218
219 actionableDynamicQuery.setCompanyId(companyId);
220
221 actionableDynamicQuery.performActions();
222 }
223
224 protected void reindexRoot(final long companyId)
225 throws PortalException, SystemException {
226
227 ActionableDynamicQuery actionableDynamicQuery =
228 new GroupActionableDynamicQuery() {
229
230 @Override
231 protected void performAction(Object object)
232 throws PortalException, SystemException {
233
234 Group group = (Group)object;
235
236 long groupId = group.getGroupId();
237 long folderId =
238 BookmarksFolderConstants.DEFAULT_PARENT_FOLDER_ID;
239
240 reindexEntries(companyId, groupId, folderId);
241 }
242
243 };
244
245 actionableDynamicQuery.setCompanyId(companyId);
246
247 actionableDynamicQuery.performActions();
248 }
249
250 }