001    /**
002     * Copyright (c) 2000-present 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.portal.kernel.search;
016    
017    import com.liferay.portal.kernel.log.Log;
018    import com.liferay.portal.kernel.log.LogFactoryUtil;
019    import com.liferay.portal.kernel.util.ClassUtil;
020    import com.liferay.registry.Registry;
021    import com.liferay.registry.RegistryUtil;
022    import com.liferay.registry.ServiceReference;
023    import com.liferay.registry.ServiceRegistration;
024    import com.liferay.registry.ServiceTracker;
025    import com.liferay.registry.ServiceTrackerCustomizer;
026    import com.liferay.registry.collections.StringServiceRegistrationMap;
027    
028    import java.util.ArrayList;
029    import java.util.Collections;
030    import java.util.HashMap;
031    import java.util.HashSet;
032    import java.util.List;
033    import java.util.Map;
034    import java.util.Set;
035    import java.util.concurrent.ConcurrentHashMap;
036    
037    /**
038     * @author Raymond Aug??
039     */
040    public class IndexerRegistryUtil {
041    
042            public static Indexer getIndexer(Class<?> clazz) {
043                    return _instance._indexers.get(clazz.getName());
044            }
045    
046            public static Indexer getIndexer(String className) {
047                    return _instance._indexers.get(className);
048            }
049    
050            public static List<Indexer> getIndexers() {
051                    List<Indexer> indexers = new ArrayList<Indexer>(
052                            _instance._indexers.values());
053    
054                    return Collections.unmodifiableList(indexers);
055            }
056    
057            public static Indexer nullSafeGetIndexer(Class<?> clazz) {
058                    return _instance._nullSafeGetIndexer(clazz.getName());
059            }
060    
061            public static Indexer nullSafeGetIndexer(String className) {
062                    return _instance._nullSafeGetIndexer(className);
063            }
064    
065            public static void register(Indexer indexer) {
066                    _instance._register(null, indexer);
067            }
068    
069            public static void register(String className, Indexer indexer) {
070                    _instance._register(className, indexer);
071            }
072    
073            public static void unregister(Indexer indexer) {
074                    _instance._unregister(indexer);
075            }
076    
077            public static void unregister(String className) {
078                    _instance._unregister(className);
079            }
080    
081            private IndexerRegistryUtil() {
082                    Registry registry = RegistryUtil.getRegistry();
083    
084                    _serviceTracker = registry.trackServices(
085                            Indexer.class, new IndexerServiceTrackerCustomizer());
086    
087                    _serviceTracker.open();
088            }
089    
090            private Set<String> _aggregrateClassNames(
091                    String[] classNames, String... moreClassNames) {
092    
093                    Set<String> set = new HashSet<String>();
094    
095                    if (classNames != null) {
096                            for (String className : classNames) {
097                                    if (className == null) {
098                                            continue;
099                                    }
100    
101                                    set.add(className);
102                            }
103                    }
104    
105                    if (moreClassNames != null) {
106                            for (String className : moreClassNames) {
107                                    if (className == null) {
108                                            continue;
109                                    }
110    
111                                    set.add(className);
112                            }
113                    }
114    
115                    return set;
116            }
117    
118            private Indexer _nullSafeGetIndexer(String className) {
119                    Indexer indexer = _indexers.get(className);
120    
121                    if (indexer != null) {
122                            return indexer;
123                    }
124    
125                    if (_log.isWarnEnabled()) {
126                            _log.warn("No indexer found for " + className);
127                    }
128    
129                    return _dummyIndexer;
130            }
131    
132            private void _register(String className, Indexer indexer) {
133                    Registry registry = RegistryUtil.getRegistry();
134    
135                    Map<String, Object> properties = new HashMap<String, Object>();
136    
137                    Set<String> classNames = _aggregrateClassNames(
138                            indexer.getClassNames(), ClassUtil.getClassName(indexer),
139                            className);
140    
141                    properties.put(
142                            "indexer.classNames",
143                            classNames.toArray(new String[classNames.size()]));
144    
145                    properties.put("javax.portlet.name", indexer.getPortletId());
146    
147                    ServiceRegistration<Indexer> serviceRegistration =
148                            registry.registerService(Indexer.class, indexer, properties);
149    
150                    for (String curClassName : classNames) {
151                            _serviceRegistrations.put(curClassName, serviceRegistration);
152                    }
153            }
154    
155            private void _unregister(Indexer indexer) {
156                    Set<String> classNames = _aggregrateClassNames(
157                            indexer.getClassNames(), ClassUtil.getClassName(indexer));
158    
159                    for (String className : classNames) {
160                            _unregister(className);
161                    }
162            }
163    
164            private void _unregister(String className) {
165                    ServiceRegistration<Indexer> serviceRegistration =
166                            _serviceRegistrations.remove(className);
167    
168                    if (serviceRegistration != null) {
169                            serviceRegistration.unregister();
170                    }
171            }
172    
173            private static final Log _log = LogFactoryUtil.getLog(
174                    IndexerRegistryUtil.class);
175    
176            private static final IndexerRegistryUtil _instance =
177                    new IndexerRegistryUtil();
178    
179            private static final Indexer _dummyIndexer = new DummyIndexer();
180    
181            private final Map<String, Indexer> _indexers =
182                    new ConcurrentHashMap<String, Indexer>();
183            private final StringServiceRegistrationMap<Indexer> _serviceRegistrations =
184                    new StringServiceRegistrationMap<Indexer>();
185            private final ServiceTracker<Indexer, Indexer> _serviceTracker;
186    
187            private class IndexerServiceTrackerCustomizer
188                    implements ServiceTrackerCustomizer<Indexer, Indexer> {
189    
190                    @Override
191                    public Indexer addingService(
192                            ServiceReference<Indexer> serviceReference) {
193    
194                            Registry registry = RegistryUtil.getRegistry();
195    
196                            Indexer indexer = registry.getService(serviceReference);
197    
198                            Set<String> classNames = _aggregrateClassNames(
199                                    (String[])serviceReference.getProperty("indexer.classNames"),
200                                    indexer.getClassNames());
201    
202                            for (String className : classNames) {
203                                    _indexers.put(className, indexer);
204                            }
205    
206                            return indexer;
207                    }
208    
209                    @Override
210                    public void modifiedService(
211                            ServiceReference<Indexer> serviceReference, Indexer indexer) {
212                    }
213    
214                    @Override
215                    public void removedService(
216                            ServiceReference<Indexer> serviceReference, Indexer indexer) {
217    
218                            Registry registry = RegistryUtil.getRegistry();
219    
220                            registry.ungetService(serviceReference);
221    
222                            Set<String> classNames = _aggregrateClassNames(
223                                    (String[])serviceReference.getProperty("indexer.classNames"),
224                                    indexer.getClassNames());
225    
226                            for (String className : classNames) {
227                                    _indexers.remove(className);
228                            }
229                    }
230    
231            }
232    
233    }