001
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
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 }