001
014
015 package com.liferay.portlet.usersadmin.util;
016
017 import com.liferay.portal.kernel.dao.orm.ActionableDynamicQuery;
018 import com.liferay.portal.kernel.exception.PortalException;
019 import com.liferay.portal.kernel.log.Log;
020 import com.liferay.portal.kernel.log.LogFactoryUtil;
021 import com.liferay.portal.kernel.search.BaseIndexer;
022 import com.liferay.portal.kernel.search.BooleanQuery;
023 import com.liferay.portal.kernel.search.Document;
024 import com.liferay.portal.kernel.search.Field;
025 import com.liferay.portal.kernel.search.SearchContext;
026 import com.liferay.portal.kernel.search.SearchEngineUtil;
027 import com.liferay.portal.kernel.search.Summary;
028 import com.liferay.portal.kernel.search.filter.BooleanFilter;
029 import com.liferay.portal.kernel.spring.osgi.OSGiBeanProperties;
030 import com.liferay.portal.kernel.util.GetterUtil;
031 import com.liferay.portal.kernel.util.Validator;
032 import com.liferay.portal.kernel.workflow.WorkflowConstants;
033 import com.liferay.portal.model.Contact;
034 import com.liferay.portal.model.User;
035 import com.liferay.portal.service.ContactLocalServiceUtil;
036 import com.liferay.portal.service.UserLocalServiceUtil;
037
038 import java.util.LinkedHashMap;
039 import java.util.Locale;
040
041 import javax.portlet.PortletRequest;
042 import javax.portlet.PortletResponse;
043
044
049 @OSGiBeanProperties
050 public class ContactIndexer extends BaseIndexer {
051
052 public static final String CLASS_NAME = Contact.class.getName();
053
054 public ContactIndexer() {
055 setStagingAware(false);
056 }
057
058 @Override
059 public String getClassName() {
060 return CLASS_NAME;
061 }
062
063 @Override
064 public void postProcessSearchQuery(
065 BooleanQuery searchQuery, BooleanFilter fullQueryBooleanFilter,
066 SearchContext searchContext)
067 throws Exception {
068
069 addSearchTerm(searchQuery, searchContext, "city", false);
070 addSearchTerm(searchQuery, searchContext, "country", false);
071 addSearchTerm(searchQuery, searchContext, "emailAddress", false);
072 addSearchTerm(searchQuery, searchContext, "firstName", false);
073 addSearchTerm(searchQuery, searchContext, "fullName", false);
074 addSearchTerm(searchQuery, searchContext, "lastName", false);
075 addSearchTerm(searchQuery, searchContext, "middleName", false);
076 addSearchTerm(searchQuery, searchContext, "region", false);
077 addSearchTerm(searchQuery, searchContext, "screenName", false);
078 addSearchTerm(searchQuery, searchContext, "street", false);
079 addSearchTerm(searchQuery, searchContext, "zip", false);
080
081 LinkedHashMap<String, Object> params =
082 (LinkedHashMap<String, Object>)searchContext.getAttribute("params");
083
084 if (params != null) {
085 String expandoAttributes = (String)params.get("expandoAttributes");
086
087 if (Validator.isNotNull(expandoAttributes)) {
088 addSearchExpando(searchQuery, searchContext, expandoAttributes);
089 }
090 }
091 }
092
093 @Override
094 protected void doDelete(Object obj) throws Exception {
095 Contact contact = (Contact)obj;
096
097 deleteDocument(contact.getCompanyId(), contact.getContactId());
098 }
099
100 @Override
101 protected Document doGetDocument(Object obj) throws Exception {
102 Contact contact = (Contact)obj;
103
104 if (contact.isUser()) {
105 User user = UserLocalServiceUtil.fetchUserByContactId(
106 contact.getContactId());
107
108 if ((user == null) || user.isDefaultUser() ||
109 (user.getStatus() != WorkflowConstants.STATUS_APPROVED)) {
110
111 return null;
112 }
113 }
114
115 Document document = getBaseModelDocument(CLASS_NAME, contact);
116
117 document.addKeyword(Field.COMPANY_ID, contact.getCompanyId());
118 document.addDate(Field.MODIFIED_DATE, contact.getModifiedDate());
119 document.addKeyword(Field.USER_ID, contact.getUserId());
120 document.addKeyword(Field.USER_NAME, contact.getFullName());
121
122 document.addText("emailAddress", contact.getEmailAddress());
123 document.addText("firstName", contact.getFirstName());
124 document.addText("fullName", contact.getFullName());
125 document.addText("jobTitle", contact.getJobTitle());
126 document.addText("lastName", contact.getLastName());
127 document.addText("middleName", contact.getMiddleName());
128
129 return document;
130 }
131
132 @Override
133 protected String doGetSortField(String orderByCol) {
134 if (orderByCol.equals("email-address")) {
135 return "emailAddress";
136 }
137 else if (orderByCol.equals("first-name")) {
138 return "firstName";
139 }
140 else if (orderByCol.equals("job-title")) {
141 return "jobTitle";
142 }
143 else if (orderByCol.equals("last-name")) {
144 return "lastName";
145 }
146 else {
147 return orderByCol;
148 }
149 }
150
151 @Override
152 protected Summary doGetSummary(
153 Document document, Locale locale, String snippet,
154 PortletRequest portletRequest, PortletResponse portletResponse) {
155
156 return null;
157 }
158
159 @Override
160 protected void doReindex(Object obj) throws Exception {
161 Contact contact = (Contact)obj;
162
163 Document document = getDocument(contact);
164
165 if (document != null) {
166 SearchEngineUtil.updateDocument(
167 getSearchEngineId(), contact.getCompanyId(), document,
168 isCommitImmediately());
169 }
170 }
171
172 @Override
173 protected void doReindex(String className, long classPK) throws Exception {
174 Contact contact = ContactLocalServiceUtil.getContact(classPK);
175
176 doReindex(contact);
177 }
178
179 @Override
180 protected void doReindex(String[] ids) throws Exception {
181 long companyId = GetterUtil.getLong(ids[0]);
182
183 reindexContacts(companyId);
184 }
185
186 protected void reindexContacts(long companyId) throws PortalException {
187 final ActionableDynamicQuery actionableDynamicQuery =
188 ContactLocalServiceUtil.getActionableDynamicQuery();
189
190 actionableDynamicQuery.setCompanyId(companyId);
191 actionableDynamicQuery.setPerformActionMethod(
192 new ActionableDynamicQuery.PerformActionMethod() {
193
194 @Override
195 public void performAction(Object object) {
196 Contact contact = (Contact)object;
197
198 try {
199 Document document = getDocument(contact);
200
201 if (document != null) {
202 actionableDynamicQuery.addDocument(document);
203 }
204 }
205 catch (PortalException pe) {
206 if (_log.isWarnEnabled()) {
207 _log.warn(
208 "Unable to index contact " +
209 contact.getContactId(),
210 pe);
211 }
212 }
213 }
214
215 });
216 actionableDynamicQuery.setSearchEngineId(getSearchEngineId());
217
218 actionableDynamicQuery.performActions();
219 }
220
221 private static final Log _log = LogFactoryUtil.getLog(ContactIndexer.class);
222
223 }