| package org.apache.felix.dm.lambda.impl; |
| |
| import java.util.ArrayList; |
| import java.util.Dictionary; |
| import java.util.HashMap; |
| import java.util.List; |
| import java.util.Map; |
| import java.util.Objects; |
| import java.util.function.Supplier; |
| import java.util.stream.Stream; |
| |
| import org.apache.felix.dm.BundleDependency; |
| import org.apache.felix.dm.Component; |
| import org.apache.felix.dm.DependencyManager; |
| import org.apache.felix.dm.lambda.BundleDependencyBuilder; |
| import org.apache.felix.dm.lambda.callbacks.CbBundle; |
| import org.apache.felix.dm.lambda.callbacks.CbComponentBundle; |
| import org.apache.felix.dm.lambda.callbacks.CbTypeBundle; |
| import org.apache.felix.dm.lambda.callbacks.CbTypeComponentBundle; |
| import org.osgi.framework.Bundle; |
| |
| @SuppressWarnings("unchecked") |
| public class BundleDependencyBuilderImpl implements BundleDependencyBuilder { |
| private String m_added; |
| private String m_changed; |
| private String m_removed; |
| private Object m_instance; |
| private boolean m_autoConfig = true; |
| private boolean m_autoConfigInvoked = false; |
| private boolean m_required = true; |
| private Bundle m_bundle; |
| private String m_filter; |
| private int m_stateMask = -1; |
| private boolean m_propagate; |
| private Object m_propagateInstance; |
| private String m_propagateMethod; |
| private Supplier<Dictionary<?, ?>> m_propagateSupplier; |
| private final Component m_component; |
| |
| enum Cb { |
| ADD, |
| CHG, |
| REM |
| }; |
| |
| private final Map<Cb, List<MethodRef<Object>>> m_refs = new HashMap<>(); |
| |
| @FunctionalInterface |
| interface MethodRef<I> { |
| public void accept(I instance, Component c, Bundle bundle); |
| } |
| |
| /** |
| * Class used to call a supplier that returns Propagated properties |
| */ |
| private class Propagate { |
| @SuppressWarnings("unused") |
| Dictionary<?, ?> propagate() { |
| return m_propagateSupplier.get(); |
| } |
| } |
| |
| public BundleDependencyBuilderImpl (Component component) { |
| m_component = component; |
| } |
| |
| @Override |
| public BundleDependencyBuilder autoConfig(boolean autoConfig) { |
| m_autoConfig = autoConfig; |
| m_autoConfigInvoked = true; |
| return this; |
| } |
| |
| @Override |
| public BundleDependencyBuilder autoConfig() { |
| autoConfig(true); |
| return this; |
| } |
| |
| @Override |
| public BundleDependencyBuilder required(boolean required) { |
| m_required = required; |
| return this; |
| } |
| |
| @Override |
| public BundleDependencyBuilder required() { |
| required(true); |
| return this; |
| } |
| |
| @Override |
| public BundleDependencyBuilder bundle(Bundle bundle) { |
| m_bundle = bundle; |
| return this; |
| } |
| |
| @Override |
| public BundleDependencyBuilder filter(String filter) throws IllegalArgumentException { |
| m_filter = filter; |
| return this; |
| } |
| |
| @Override |
| public BundleDependencyBuilder mask(int mask) { |
| m_stateMask = mask; |
| return this; |
| } |
| |
| @Override |
| public BundleDependencyBuilder propagate(boolean propagate) { |
| m_propagate = propagate; |
| return this; |
| } |
| |
| @Override |
| public BundleDependencyBuilder propagate() { |
| propagate(true); |
| return this; |
| } |
| |
| @Override |
| public BundleDependencyBuilder propagate(Object instance, String method) { |
| if (m_propagateSupplier != null || m_propagate) throw new IllegalStateException("Propagate callback already set."); |
| Objects.nonNull(method); |
| Objects.nonNull(instance); |
| m_propagateInstance = instance; |
| m_propagateMethod = method; |
| return this; |
| } |
| |
| @Override |
| public BundleDependencyBuilder propagate(Supplier<Dictionary<?, ?>> instance) { |
| if (m_propagateInstance != null || m_propagate) throw new IllegalStateException("Propagate callback already set."); |
| m_propagateSupplier = instance; |
| return this; |
| } |
| |
| public BundleDependencyBuilder cb(String ... callbacks) { |
| return cb(null, callbacks); |
| } |
| |
| @Override |
| public BundleDependencyBuilder cb(Object callbackInstance, String ... callbacks) { |
| switch (callbacks.length) { |
| case 1: |
| cbi(callbackInstance, callbacks[0], null, null); |
| break; |
| |
| case 2: |
| cbi(callbackInstance, callbacks[0], null, callbacks[1]); |
| break; |
| |
| case 3: |
| cbi(callbackInstance, callbacks[0], callbacks[1], callbacks[2]); |
| break; |
| |
| default: |
| throw new IllegalArgumentException("wrong number of arguments: " + callbacks.length + ". " + |
| "Possible arguments: [add], [add, remove] or [add, change, remove]"); |
| } |
| |
| return this; |
| } |
| |
| private BundleDependencyBuilder cbi(Object callbackInstance, String added, String changed, String removed) { |
| requiresNoMethodRefs(); |
| m_instance = callbackInstance; |
| m_added = added != null ? added : m_added; |
| m_changed = changed != null ? changed : m_changed; |
| m_removed = removed != null ? removed : m_removed; |
| if (! m_autoConfigInvoked) m_autoConfig = false; |
| return this; |
| } |
| |
| @Override |
| public <T> BundleDependencyBuilder cb(CbTypeBundle<T> add) { |
| return cb(add, null, null); |
| } |
| |
| @Override |
| public <T> BundleDependencyBuilder cb(CbTypeBundle<T> add, CbTypeBundle<T> remove) { |
| return cb(add, null, remove); |
| } |
| |
| @Override |
| public <T> BundleDependencyBuilder cb(CbTypeBundle<T> add, CbTypeBundle<T> change, CbTypeBundle<T> remove) { |
| if (add != null) { |
| setComponentCallbackRef(Cb.ADD, Helpers.getLambdaArgType(add, 0), (inst, component, bundle) -> add.accept ((T) inst, bundle)); |
| } |
| if (change != null) { |
| setComponentCallbackRef(Cb.CHG, Helpers.getLambdaArgType(change, 0), (inst, component, bundle) -> change.accept ((T) inst, bundle)); |
| } |
| if (remove != null) { |
| setComponentCallbackRef(Cb.REM, Helpers.getLambdaArgType(remove, 0), (inst, component, bundle) -> remove.accept ((T) inst, bundle)); |
| } |
| return this; |
| } |
| |
| @Override |
| public <T> BundleDependencyBuilder cb(CbTypeComponentBundle<T> add) { |
| return cb(add, null, null); |
| } |
| |
| @Override |
| public <T> BundleDependencyBuilder cb(CbTypeComponentBundle<T> add, CbTypeComponentBundle<T> remove) { |
| return cb(add, null, remove); |
| } |
| |
| @Override |
| public <T> BundleDependencyBuilder cb(CbTypeComponentBundle<T> add, CbTypeComponentBundle<T> change, CbTypeComponentBundle<T> remove) { |
| if (add != null) { |
| setComponentCallbackRef(Cb.ADD, Helpers.getLambdaArgType(add, 0), (inst, component, bundle) -> add.accept ((T) inst, component, bundle)); |
| } |
| if (change != null) { |
| setComponentCallbackRef(Cb.CHG, Helpers.getLambdaArgType(change, 0), (inst, component, bundle) -> change.accept ((T) inst, component, bundle)); |
| } |
| if (remove != null) { |
| setComponentCallbackRef(Cb.REM, Helpers.getLambdaArgType(remove, 0), (inst, component, bundle) -> remove.accept ((T) inst, component, bundle)); |
| } |
| return this; |
| } |
| |
| @Override |
| public BundleDependencyBuilder cbi(CbBundle add) { |
| return cbi(add, null, null); |
| } |
| |
| @Override |
| public BundleDependencyBuilder cbi(CbBundle add, CbBundle remove) { |
| return cbi(add, null, remove); |
| } |
| |
| @Override |
| public BundleDependencyBuilder cbi(CbBundle add, CbBundle change, CbBundle remove) { |
| if (add != null) setInstanceCallbackRef(Cb.ADD, (inst, component, bundle) -> add.accept(bundle)); |
| if (change != null) setInstanceCallbackRef(Cb.CHG, (inst, component, bundle) -> change.accept(bundle)); |
| if (remove != null) setInstanceCallbackRef(Cb.REM, (inst, component, bundle) -> remove.accept(bundle)); |
| return this; |
| } |
| |
| @Override |
| public BundleDependencyBuilder cbi(CbComponentBundle add) { |
| return cbi(add, null, null); |
| } |
| |
| @Override |
| public BundleDependencyBuilder cbi(CbComponentBundle add, CbComponentBundle remove) { |
| return cbi(add, null, remove); |
| } |
| |
| @Override |
| public BundleDependencyBuilder cbi(CbComponentBundle add, CbComponentBundle change, CbComponentBundle remove) { |
| if (add != null) setInstanceCallbackRef(Cb.ADD, (inst, component, bundle) -> add.accept(component, bundle)); |
| if (change != null) setInstanceCallbackRef(Cb.CHG, (inst, component, bundle) -> change.accept(component, bundle)); |
| if (remove != null) setInstanceCallbackRef(Cb.REM, (inst, component, bundle) -> remove.accept(component, bundle)); |
| return this; |
| } |
| |
| @Override |
| public BundleDependency build() { |
| DependencyManager dm = m_component.getDependencyManager(); |
| |
| BundleDependency dep = dm.createBundleDependency(); |
| dep.setRequired(m_required); |
| |
| if (m_filter != null) { |
| dep.setFilter(m_filter); |
| } |
| |
| if (m_bundle != null) { |
| dep.setBundle(m_bundle); |
| } |
| |
| if (m_stateMask != -1) { |
| dep.setStateMask(m_stateMask); |
| } |
| |
| if (m_propagate) { |
| dep.setPropagate(true); |
| } else if (m_propagateInstance != null) { |
| dep.setPropagate(m_propagateInstance, m_propagateMethod); |
| } else if (m_propagateSupplier != null) { |
| dep.setPropagate(new Propagate(), "propagate"); |
| } |
| |
| if (m_added != null || m_changed != null || m_removed != null) { |
| dep.setCallbacks(m_instance, m_added, m_changed, m_removed); |
| } else if (m_refs.size() > 0) { |
| Object cb = createCallbackInstance(); |
| dep.setCallbacks(cb, "add", "change", "remove"); |
| } |
| |
| dep.setAutoConfig(m_autoConfig); |
| return dep; |
| } |
| |
| private <T> BundleDependencyBuilder setInstanceCallbackRef(Cb cbType, MethodRef<T> ref) { |
| requiresNoStringCallbacks(); |
| if (! m_autoConfigInvoked) m_autoConfig = false; |
| List<MethodRef<Object>> list = m_refs.computeIfAbsent(cbType, l -> new ArrayList<>()); |
| list.add((instance, component, bundle) -> ref.accept(null, component, bundle)); |
| return this; |
| } |
| |
| private <T> BundleDependencyBuilder setComponentCallbackRef(Cb cbType, Class<T> type, MethodRef<T> ref) { |
| requiresNoStringCallbacks(); |
| if (! m_autoConfigInvoked) m_autoConfig = false; |
| List<MethodRef<Object>> list = m_refs.computeIfAbsent(cbType, l -> new ArrayList<>()); |
| list.add((instance, component, bundle) -> { |
| Object componentImpl = Stream.of(component.getInstances()) |
| .filter(impl -> Helpers.getClass(impl).equals(type)) |
| .findFirst() |
| .orElseThrow(() -> new IllegalStateException("The method reference " + ref + " does not match any available component impl classes.")); |
| ref.accept((T) componentImpl, component, bundle); |
| }); |
| return this; |
| } |
| |
| @SuppressWarnings("unused") |
| private Object createCallbackInstance() { |
| Object cb = null; |
| |
| cb = new Object() { |
| void add(Component c, Bundle bundle) { |
| invokeMethodRefs(Cb.ADD, c, bundle); |
| } |
| |
| void change(Component c, Bundle bundle) { |
| invokeMethodRefs(Cb.CHG, c, bundle); |
| } |
| |
| void remove(Component c, Bundle bundle) { |
| invokeMethodRefs(Cb.REM, c, bundle); |
| } |
| }; |
| |
| return cb; |
| } |
| |
| private void invokeMethodRefs(Cb cbType, Component c, Bundle bundle) { |
| m_refs.computeIfPresent(cbType, (k, mrefs) -> { |
| mrefs.forEach(mref -> mref.accept(null, c, bundle)); |
| return mrefs; |
| }); |
| } |
| |
| private void requiresNoStringCallbacks() { |
| if (m_added != null || m_changed != null || m_removed != null) { |
| throw new IllegalStateException("can't mix method references and string callbacks."); |
| } |
| } |
| |
| private void requiresNoMethodRefs() { |
| if (m_refs.size() > 0) { |
| throw new IllegalStateException("can't mix method references and string callbacks."); |
| } |
| } |
| } |