001
014
015 package com.liferay.portlet.messageboards.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.DocumentImpl;
027 import com.liferay.portal.kernel.search.Field;
028 import com.liferay.portal.kernel.search.SearchContext;
029 import com.liferay.portal.kernel.search.SearchEngineUtil;
030 import com.liferay.portal.kernel.search.Summary;
031 import com.liferay.portal.kernel.util.GetterUtil;
032 import com.liferay.portal.kernel.workflow.WorkflowConstants;
033 import com.liferay.portal.model.Group;
034 import com.liferay.portal.security.permission.PermissionChecker;
035 import com.liferay.portal.service.persistence.GroupActionableDynamicQuery;
036 import com.liferay.portal.util.PortletKeys;
037 import com.liferay.portlet.messageboards.NoSuchDiscussionException;
038 import com.liferay.portlet.messageboards.model.MBCategory;
039 import com.liferay.portlet.messageboards.model.MBCategoryConstants;
040 import com.liferay.portlet.messageboards.model.MBThread;
041 import com.liferay.portlet.messageboards.service.MBDiscussionLocalServiceUtil;
042 import com.liferay.portlet.messageboards.service.MBThreadLocalServiceUtil;
043 import com.liferay.portlet.messageboards.service.persistence.MBCategoryActionableDynamicQuery;
044 import com.liferay.portlet.messageboards.service.persistence.MBThreadActionableDynamicQuery;
045
046 import java.util.Locale;
047
048 import javax.portlet.PortletURL;
049
050
053 public class MBThreadIndexer extends BaseIndexer {
054
055 public static final String[] CLASS_NAMES = {MBThread.class.getName()};
056
057 public static final String PORTLET_ID = PortletKeys.MESSAGE_BOARDS;
058
059 public MBThreadIndexer() {
060 setFilterSearch(true);
061 setPermissionAware(true);
062 }
063
064 @Override
065 public String[] getClassNames() {
066 return CLASS_NAMES;
067 }
068
069 @Override
070 public String getPortletId() {
071 return PORTLET_ID;
072 }
073
074 @Override
075 public boolean hasPermission(
076 PermissionChecker permissionChecker, String entryClassName,
077 long entryClassPK, String actionId)
078 throws Exception {
079
080 return true;
081 }
082
083 @Override
084 public void postProcessContextQuery(
085 BooleanQuery contextQuery, SearchContext searchContext)
086 throws Exception {
087
088 addStatus(contextQuery, searchContext);
089
090 boolean discussion = GetterUtil.getBoolean(
091 searchContext.getAttribute("discussion"), false);
092
093 contextQuery.addRequiredTerm("discussion", discussion);
094
095 long endDate = GetterUtil.getLong(
096 searchContext.getAttribute("endDate"));
097 long startDate = GetterUtil.getLong(
098 searchContext.getAttribute("startDate"));
099
100 if ((endDate > 0) && (startDate > 0)) {
101 contextQuery.addRangeTerm("lastPostDate", startDate, endDate);
102 }
103
104 long participantUserId = GetterUtil.getLong(
105 searchContext.getAttribute("participantUserId"));
106
107 if (participantUserId > 0) {
108 contextQuery.addRequiredTerm(
109 "participantUserIds", participantUserId);
110 }
111 }
112
113 @Override
114 protected void doDelete(Object obj) throws Exception {
115 SearchContext searchContext = new SearchContext();
116
117 searchContext.setSearchEngineId(getSearchEngineId());
118
119 MBThread thread = (MBThread)obj;
120
121 Document document = new DocumentImpl();
122
123 document.addUID(PORTLET_ID, thread.getThreadId());
124
125 SearchEngineUtil.deleteDocument(
126 getSearchEngineId(), thread.getCompanyId(),
127 document.get(Field.UID));
128 }
129
130 @Override
131 protected Document doGetDocument(Object obj) throws Exception {
132 MBThread thread = (MBThread)obj;
133
134 Document document = getBaseModelDocument(PORTLET_ID, thread);
135
136 try {
137 MBDiscussionLocalServiceUtil.getThreadDiscussion(
138 thread.getThreadId());
139
140 document.addKeyword("discussion", true);
141 }
142 catch (NoSuchDiscussionException nsde) {
143 document.addKeyword("discussion", false);
144 }
145
146 document.addKeyword("lastPostDate", thread.getLastPostDate().getTime());
147 document.addKeyword(
148 "participantUserIds", thread.getParticipantUserIds());
149
150 return document;
151 }
152
153 @Override
154 protected Summary doGetSummary(
155 Document document, Locale locale, String snippet,
156 PortletURL portletURL) {
157
158 return null;
159 }
160
161 @Override
162 protected void doReindex(Object obj) throws Exception {
163 MBThread thread = (MBThread)obj;
164
165 Document document = getDocument(thread);
166
167 SearchEngineUtil.updateDocument(
168 getSearchEngineId(), thread.getCompanyId(), document);
169 }
170
171 @Override
172 protected void doReindex(String className, long classPK) throws Exception {
173 MBThread thread = MBThreadLocalServiceUtil.getThread(classPK);
174
175 doReindex(thread);
176 }
177
178 @Override
179 protected void doReindex(String[] ids) throws Exception {
180 long companyId = GetterUtil.getLong(ids[0]);
181
182 reindexCategories(companyId);
183 reindexDiscussions(companyId);
184 reindexRoot(companyId);
185 }
186
187 @Override
188 protected String getPortletId(SearchContext searchContext) {
189 return PORTLET_ID;
190 }
191
192 protected void reindexCategories(final long companyId)
193 throws PortalException, SystemException {
194
195 ActionableDynamicQuery actionableDynamicQuery =
196 new MBCategoryActionableDynamicQuery() {
197
198 @Override
199 protected void performAction(Object object)
200 throws PortalException, SystemException {
201
202 MBCategory category = (MBCategory)object;
203
204 reindexThreads(
205 companyId, category.getGroupId(), category.getCategoryId());
206 }
207
208 };
209
210 actionableDynamicQuery.setCompanyId(companyId);
211
212 actionableDynamicQuery.performActions();
213 }
214
215 protected void reindexDiscussions(final long companyId)
216 throws PortalException, SystemException {
217
218 ActionableDynamicQuery actionableDynamicQuery =
219 new GroupActionableDynamicQuery() {
220
221 @Override
222 protected void performAction(Object object)
223 throws PortalException, SystemException {
224
225 Group group = (Group)object;
226
227 reindexThreads(
228 companyId, group.getGroupId(),
229 MBCategoryConstants.DISCUSSION_CATEGORY_ID);
230 }
231
232 };
233
234 actionableDynamicQuery.setCompanyId(companyId);
235
236 actionableDynamicQuery.performActions();
237 }
238
239 protected void reindexRoot(final long companyId)
240 throws PortalException, SystemException {
241
242 ActionableDynamicQuery actionableDynamicQuery =
243 new GroupActionableDynamicQuery() {
244
245 @Override
246 protected void performAction(Object object)
247 throws PortalException, SystemException {
248
249 Group group = (Group)object;
250
251 reindexThreads(
252 companyId, group.getGroupId(),
253 MBCategoryConstants.DEFAULT_PARENT_CATEGORY_ID);
254 }
255
256 };
257
258 actionableDynamicQuery.setCompanyId(companyId);
259
260 actionableDynamicQuery.performActions();
261 }
262
263 protected void reindexThreads(
264 long companyId, long groupId, final long categoryId)
265 throws PortalException, SystemException {
266
267 ActionableDynamicQuery actionableDynamicQuery =
268 new MBThreadActionableDynamicQuery() {
269
270 @Override
271 protected void addCriteria(DynamicQuery dynamicQuery) {
272 Property categoryIdProperty = PropertyFactoryUtil.forName(
273 "categoryId");
274
275 dynamicQuery.add(categoryIdProperty.eq(categoryId));
276
277 Property statusProperty = PropertyFactoryUtil.forName("status");
278
279 dynamicQuery.add(
280 statusProperty.eq(WorkflowConstants.STATUS_APPROVED));
281 }
282
283 @Override
284 protected void performAction(Object object) throws PortalException {
285 MBThread thread = (MBThread)object;
286
287 Document document = getDocument(thread);
288
289 addDocument(document);
290 }
291
292 };
293
294 actionableDynamicQuery.setCompanyId(companyId);
295 actionableDynamicQuery.setGroupId(groupId);
296 actionableDynamicQuery.setSearchEngineId(getSearchEngineId());
297
298 actionableDynamicQuery.performActions();
299 }
300
301 }