| /** |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| package org.apache.aries.rsa.discovery.local; |
| |
| import java.util.ArrayList; |
| import java.util.Collection; |
| import java.util.Dictionary; |
| import java.util.HashMap; |
| import java.util.Hashtable; |
| import java.util.Iterator; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.Map.Entry; |
| import java.util.concurrent.ConcurrentHashMap; |
| |
| import org.apache.aries.rsa.util.StringPlus; |
| import org.osgi.framework.Bundle; |
| import org.osgi.framework.BundleEvent; |
| import org.osgi.framework.BundleListener; |
| import org.osgi.framework.Filter; |
| import org.osgi.framework.FrameworkUtil; |
| import org.osgi.framework.ServiceReference; |
| import org.osgi.service.remoteserviceadmin.EndpointDescription; |
| import org.osgi.service.remoteserviceadmin.EndpointEvent; |
| import org.osgi.service.remoteserviceadmin.EndpointEventListener; |
| |
| public class LocalDiscovery implements BundleListener { |
| |
| // this is effectively a set which allows for multiple service descriptions with the |
| // same interface name but different properties and takes care of itself with respect to concurrency |
| final Map<EndpointDescription, Bundle> endpointDescriptions = new ConcurrentHashMap<>(); |
| final Map<EndpointEventListener, Collection<String>> listenerToFilters = new HashMap<>(); |
| final Map<String, Collection<EndpointEventListener>> filterToListeners = new HashMap<>(); |
| |
| EndpointDescriptionBundleParser bundleParser; |
| |
| public LocalDiscovery() { |
| this.bundleParser = new EndpointDescriptionBundleParser(); |
| } |
| |
| public void processExistingBundles(Bundle[] bundles) { |
| if (bundles == null) { |
| return; |
| } |
| |
| for (Bundle b : bundles) { |
| if (b.getState() == Bundle.ACTIVE) { |
| findDeclaredRemoteServices(b); |
| } |
| } |
| } |
| |
| void addListener(ServiceReference<EndpointEventListener> endpointListenerRef, EndpointEventListener endpointListener) { |
| List<String> filters = StringPlus.normalize(endpointListenerRef.getProperty(EndpointEventListener.ENDPOINT_LISTENER_SCOPE)); |
| if (filters.isEmpty()) { |
| return; |
| } |
| |
| synchronized (listenerToFilters) { |
| listenerToFilters.put(endpointListener, filters); |
| for (String filter : filters) { |
| Collection<EndpointEventListener> listeners = filterToListeners.get(filter); |
| if (listeners == null) { |
| listeners = new ArrayList<>(); |
| filterToListeners.put(filter, listeners); |
| } |
| listeners.add(endpointListener); |
| } |
| } |
| |
| triggerCallbacks(filters, endpointListener); |
| } |
| |
| /** |
| * If the tracker was removed or the scope was changed this doesn't require |
| * additional callbacks on the tracker. Its the responsibility of the tracker |
| * itself to clean up any orphans. See Remote Service Admin spec 122.6.3 |
| * @param endpointListener |
| */ |
| void removeListener(EndpointEventListener endpointListener) { |
| synchronized (listenerToFilters) { |
| Collection<String> filters = listenerToFilters.remove(endpointListener); |
| if (filters == null) { |
| return; |
| } |
| |
| for (String filter : filters) { |
| Collection<EndpointEventListener> listeners = filterToListeners.get(filter); |
| if (listeners != null) { |
| listeners.remove(endpointListener); |
| if (listeners.isEmpty()) { |
| filterToListeners.remove(filter); |
| } |
| } |
| } |
| } |
| } |
| |
| private Map<String, Collection<EndpointEventListener>> getMatchingListeners(EndpointDescription endpoint) { |
| // return a copy of matched filters/listeners so that caller doesn't need to hold locks while triggering events |
| Map<String, Collection<EndpointEventListener>> matched = new HashMap<>(); |
| synchronized (listenerToFilters) { |
| for (Entry<String, Collection<EndpointEventListener>> entry : filterToListeners.entrySet()) { |
| String filter = entry.getKey(); |
| if (LocalDiscovery.matchFilter(filter, endpoint)) { |
| matched.put(filter, new ArrayList<>(entry.getValue())); |
| } |
| } |
| } |
| return matched; |
| } |
| |
| // BundleListener method |
| public void bundleChanged(BundleEvent be) { |
| switch (be.getType()) { |
| case BundleEvent.STARTED: |
| findDeclaredRemoteServices(be.getBundle()); |
| break; |
| case BundleEvent.STOPPED: |
| removeServicesDeclaredInBundle(be.getBundle()); |
| break; |
| default: |
| } |
| } |
| |
| private void findDeclaredRemoteServices(Bundle bundle) { |
| List<EndpointDescription> endpoints = bundleParser.getAllEndpointDescriptions(bundle); |
| for (EndpointDescription endpoint : endpoints) { |
| endpointDescriptions.put(endpoint, bundle); |
| EndpointEvent event = new EndpointEvent(EndpointEvent.ADDED, endpoint); |
| triggerCallbacks(event); |
| } |
| } |
| |
| private void removeServicesDeclaredInBundle(Bundle bundle) { |
| for (Iterator<Entry<EndpointDescription, Bundle>> i = endpointDescriptions.entrySet().iterator(); |
| i.hasNext();) { |
| Entry<EndpointDescription, Bundle> entry = i.next(); |
| if (bundle.equals(entry.getValue())) { |
| EndpointEvent event = new EndpointEvent(EndpointEvent.REMOVED, entry.getKey()); |
| triggerCallbacks(event); |
| i.remove(); |
| } |
| } |
| } |
| |
| private void triggerCallbacks(EndpointEvent event) { |
| EndpointDescription endpoint = event.getEndpoint(); |
| for (Map.Entry<String, Collection<EndpointEventListener>> entry : getMatchingListeners(endpoint).entrySet()) { |
| String filter = entry.getKey(); |
| for (EndpointEventListener listener : entry.getValue()) { |
| triggerCallbacks(listener, filter, event); |
| } |
| } |
| } |
| |
| private void triggerCallbacks(EndpointEventListener endpointListener, String filter, EndpointEvent event) { |
| if (!LocalDiscovery.matchFilter(filter, event.getEndpoint())) { |
| return; |
| } |
| endpointListener.endpointChanged(event, filter); |
| } |
| |
| private void triggerCallbacks(Collection<String> filters, EndpointEventListener endpointListener) { |
| for (String filter : filters) { |
| for (EndpointDescription endpoint : endpointDescriptions.keySet()) { |
| EndpointEvent event = new EndpointEvent(EndpointEvent.ADDED, endpoint); |
| triggerCallbacks(endpointListener, filter, event); |
| } |
| } |
| } |
| |
| private static boolean matchFilter(String filter, EndpointDescription endpoint) { |
| if (filter == null) { |
| return false; |
| } |
| |
| try { |
| Filter f = FrameworkUtil.createFilter(filter); |
| Dictionary<String, Object> dict = new Hashtable<>(endpoint.getProperties()); |
| return f.match(dict); |
| } catch (Exception e) { |
| return false; |
| } |
| } |
| |
| } |