001    /**
002     * Copyright (c) 2000-2013 Liferay, Inc. All rights reserved.
003     *
004     * This library is free software; you can redistribute it and/or modify it under
005     * the terms of the GNU Lesser General Public License as published by the Free
006     * Software Foundation; either version 2.1 of the License, or (at your option)
007     * any later version.
008     *
009     * This library is distributed in the hope that it will be useful, but WITHOUT
010     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
011     * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
012     * details.
013     */
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.search.BaseIndexer;
020    import com.liferay.portal.kernel.search.BooleanClauseOccur;
021    import com.liferay.portal.kernel.search.BooleanQuery;
022    import com.liferay.portal.kernel.search.BooleanQueryFactoryUtil;
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.util.GetterUtil;
029    import com.liferay.portal.kernel.util.Validator;
030    import com.liferay.portal.model.Organization;
031    import com.liferay.portal.model.OrganizationConstants;
032    import com.liferay.portal.service.OrganizationLocalServiceUtil;
033    import com.liferay.portal.service.persistence.OrganizationActionableDynamicQuery;
034    import com.liferay.portal.util.PortletKeys;
035    import com.liferay.portal.util.PropsValues;
036    
037    import java.util.ArrayList;
038    import java.util.Collection;
039    import java.util.HashMap;
040    import java.util.LinkedHashMap;
041    import java.util.List;
042    import java.util.Locale;
043    import java.util.Map;
044    
045    import javax.portlet.PortletURL;
046    
047    /**
048     * @author Raymond Aug??
049     * @author Zsigmond Rab
050     * @author Hugo Huijser
051     */
052    public class OrganizationIndexer extends BaseIndexer {
053    
054            public static final String[] CLASS_NAMES = {Organization.class.getName()};
055    
056            public static final String PORTLET_ID = PortletKeys.USERS_ADMIN;
057    
058            public OrganizationIndexer() {
059                    setIndexerEnabled(PropsValues.ORGANIZATIONS_INDEXER_ENABLED);
060                    setPermissionAware(true);
061                    setStagingAware(false);
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 void postProcessContextQuery(
076                            BooleanQuery contextQuery, SearchContext searchContext)
077                    throws Exception {
078    
079                    LinkedHashMap<String, Object> params =
080                            (LinkedHashMap<String, Object>)searchContext.getAttribute("params");
081    
082                    if (params == null) {
083                            return;
084                    }
085    
086                    List<Long> excludedOrganizationIds = (List<Long>)params.get(
087                            "excludedOrganizationIds");
088    
089                    if ((excludedOrganizationIds != null) &&
090                            !excludedOrganizationIds.isEmpty()) {
091    
092                            BooleanQuery booleanQuery = BooleanQueryFactoryUtil.create(
093                                    searchContext);
094    
095                            for (long excludedOrganizationId : excludedOrganizationIds) {
096                                    booleanQuery.addTerm(
097                                            "organizationId", String.valueOf(excludedOrganizationId));
098                            }
099    
100                            contextQuery.add(booleanQuery, BooleanClauseOccur.MUST_NOT);
101                    }
102    
103                    List<Organization> organizationsTree = (List<Organization>)params.get(
104                            "organizationsTree");
105    
106                    if ((organizationsTree != null) && !organizationsTree.isEmpty()) {
107                            BooleanQuery booleanQuery = BooleanQueryFactoryUtil.create(
108                                    searchContext);
109    
110                            for (Organization organization : organizationsTree) {
111                                    String treePath = organization.buildTreePath();
112    
113                                    booleanQuery.addTerm(Field.TREE_PATH, treePath, true);
114                            }
115    
116                            contextQuery.add(booleanQuery, BooleanClauseOccur.MUST);
117                    }
118                    else {
119                            long parentOrganizationId = GetterUtil.getLong(
120                                    searchContext.getAttribute("parentOrganizationId"));
121    
122                            if (parentOrganizationId !=
123                                            OrganizationConstants.ANY_PARENT_ORGANIZATION_ID) {
124    
125                                    contextQuery.addRequiredTerm(
126                                            "parentOrganizationId", parentOrganizationId);
127                            }
128                    }
129            }
130    
131            @Override
132            public void postProcessSearchQuery(
133                            BooleanQuery searchQuery, SearchContext searchContext)
134                    throws Exception {
135    
136                    addSearchTerm(searchQuery, searchContext, "city", false);
137                    addSearchTerm(searchQuery, searchContext, "country", false);
138                    addSearchTerm(searchQuery, searchContext, "name", false);
139                    addSearchTerm(searchQuery, searchContext, "region", false);
140                    addSearchTerm(searchQuery, searchContext, "street", false);
141                    addSearchTerm(searchQuery, searchContext, "type", false);
142                    addSearchTerm(searchQuery, searchContext, "zip", false);
143    
144                    LinkedHashMap<String, Object> params =
145                            (LinkedHashMap<String, Object>)searchContext.getAttribute("params");
146    
147                    if (params != null) {
148                            String expandoAttributes = (String)params.get("expandoAttributes");
149    
150                            if (Validator.isNotNull(expandoAttributes)) {
151                                    addSearchExpando(searchQuery, searchContext, expandoAttributes);
152                            }
153                    }
154            }
155    
156            @Override
157            protected void doDelete(Object obj) throws Exception {
158                    Organization organization = (Organization)obj;
159    
160                    deleteDocument(
161                            organization.getCompanyId(), organization.getOrganizationId());
162            }
163    
164            @Override
165            protected Document doGetDocument(Object obj) throws Exception {
166                    Organization organization = (Organization)obj;
167    
168                    Document document = getBaseModelDocument(PORTLET_ID, organization);
169    
170                    document.addKeyword(Field.COMPANY_ID, organization.getCompanyId());
171                    document.addText(Field.NAME, organization.getName());
172                    document.addKeyword(
173                            Field.ORGANIZATION_ID, organization.getOrganizationId());
174                    document.addKeyword(Field.TREE_PATH, organization.buildTreePath());
175                    document.addKeyword(Field.TYPE, organization.getType());
176    
177                    document.addKeyword(
178                            "parentOrganizationId", organization.getParentOrganizationId());
179    
180                    populateAddresses(
181                            document, organization.getAddresses(), organization.getRegionId(),
182                            organization.getCountryId());
183    
184                    return document;
185            }
186    
187            @Override
188            protected String doGetSortField(String orderByCol) {
189                    if (orderByCol.equals("name")) {
190                            return "name";
191                    }
192                    else if (orderByCol.equals("type")) {
193                            return "type";
194                    }
195                    else {
196                            return orderByCol;
197                    }
198            }
199    
200            @Override
201            protected Summary doGetSummary(
202                    Document document, Locale locale, String snippet,
203                    PortletURL portletURL) {
204    
205                    String title = document.get("name");
206    
207                    String content = null;
208    
209                    String organizationId = document.get(Field.ORGANIZATION_ID);
210    
211                    portletURL.setParameter(
212                            "struts_action", "/users_admin/edit_organization");
213                    portletURL.setParameter("organizationId", organizationId);
214    
215                    return new Summary(title, content, portletURL);
216            }
217    
218            @Override
219            protected void doReindex(Object obj) throws Exception {
220                    if (obj instanceof List<?>) {
221                            List<Organization> organizations = (List<Organization>)obj;
222    
223                            for (Organization organization : organizations) {
224                                    doReindex(organization);
225                            }
226                    }
227                    else if (obj instanceof Long) {
228                            long organizationId = (Long)obj;
229    
230                            Organization organization =
231                                    OrganizationLocalServiceUtil.getOrganization(organizationId);
232    
233                            doReindex(organization);
234                    }
235                    else if (obj instanceof long[]) {
236                            long[] organizationIds = (long[])obj;
237    
238                            Map<Long, Collection<Document>> documentsMap =
239                                    new HashMap<Long, Collection<Document>>();
240    
241                            for (long organizationId : organizationIds) {
242                                    Organization organization =
243                                            OrganizationLocalServiceUtil.fetchOrganization(
244                                                    organizationId);
245    
246                                    if (organization == null) {
247                                            continue;
248                                    }
249    
250                                    Document document = getDocument(organization);
251    
252                                    long companyId = organization.getCompanyId();
253    
254                                    Collection<Document> documents = documentsMap.get(companyId);
255    
256                                    if (documents == null) {
257                                            documents = new ArrayList<Document>();
258    
259                                            documentsMap.put(companyId, documents);
260                                    }
261    
262                                    documents.add(document);
263                            }
264    
265                            for (Map.Entry<Long, Collection<Document>> entry :
266                                            documentsMap.entrySet()) {
267    
268                                    long companyId = entry.getKey();
269                                    Collection<Document> documents = entry.getValue();
270    
271                                    SearchEngineUtil.updateDocuments(
272                                            getSearchEngineId(), companyId, documents);
273                            }
274                    }
275                    else if (obj instanceof Organization) {
276                            Organization organization = (Organization)obj;
277    
278                            Document document = getDocument(organization);
279    
280                            SearchEngineUtil.updateDocument(
281                                    getSearchEngineId(), organization.getCompanyId(), document);
282                    }
283            }
284    
285            @Override
286            protected void doReindex(String className, long classPK) throws Exception {
287                    Organization organization =
288                            OrganizationLocalServiceUtil.getOrganization(classPK);
289    
290                    doReindex(organization);
291            }
292    
293            @Override
294            protected void doReindex(String[] ids) throws Exception {
295                    long companyId = GetterUtil.getLong(ids[0]);
296    
297                    reindexOrganizations(companyId);
298            }
299    
300            @Override
301            protected String getPortletId(SearchContext searchContext) {
302                    return PORTLET_ID;
303            }
304    
305            protected void reindexOrganizations(long companyId) throws Exception {
306                    final Collection<Document> documents = new ArrayList<Document>();
307    
308                    ActionableDynamicQuery actionableDynamicQuery =
309                            new OrganizationActionableDynamicQuery() {
310    
311                            @Override
312                            protected void performAction(Object object) throws PortalException {
313                                    Organization organization = (Organization)object;
314    
315                                    Document document = getDocument(organization);
316    
317                                    documents.add(document);
318                            }
319    
320                    };
321    
322                    actionableDynamicQuery.setCompanyId(companyId);
323    
324                    actionableDynamicQuery.performActions();
325    
326                    SearchEngineUtil.updateDocuments(
327                            getSearchEngineId(), companyId, documents);
328            }
329    
330    }