*/
package org.opendaylight.protocol.bgp.rib.spi;
-import com.google.common.base.Preconditions;
-import java.util.HashSet;
-import java.util.Set;
+import static com.google.common.base.Preconditions.checkArgument;
+import static java.util.Objects.requireNonNull;
+
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;
-import org.opendaylight.protocol.concepts.AbstractRegistration;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.AddressFamily;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.SubsequentAddressFamily;
-import org.opendaylight.yangtools.sal.binding.generator.impl.GeneratedClassLoadingStrategy;
-import org.opendaylight.yangtools.sal.binding.generator.impl.ModuleInfoBackedContext;
-import org.opendaylight.yangtools.yang.binding.YangModuleInfo;
-import org.opendaylight.yangtools.yang.binding.util.BindingReflections;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.Route;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.Tables;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.tables.Routes;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.AddressFamily;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev200120.SubsequentAddressFamily;
+import org.opendaylight.yangtools.yang.binding.ChildOf;
+import org.opendaylight.yangtools.yang.binding.ChoiceIn;
+import org.opendaylight.yangtools.yang.binding.DataObject;
+import org.opendaylight.yangtools.yang.binding.Identifiable;
+import org.opendaylight.yangtools.yang.binding.Identifier;
+import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifierWithPredicates;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class SimpleRIBExtensionProviderContext implements RIBExtensionProviderContext {
-
private static final Logger LOG = LoggerFactory.getLogger(SimpleRIBExtensionProviderContext.class);
- private final ConcurrentMap<TablesKey, AdjRIBsFactory> factories = new ConcurrentHashMap<>();
- private final ConcurrentMap<TablesKey, RIBSupport> supports = new ConcurrentHashMap<>();
- private final ModuleInfoBackedContext classLoadingStrategy = ModuleInfoBackedContext.create();
-
- @Override
- public final synchronized AbstractRegistration registerAdjRIBsInFactory(final Class<? extends AddressFamily> afi,
- final Class<? extends SubsequentAddressFamily> safi, final AdjRIBsFactory factory) {
- final TablesKey key = new TablesKey(afi, safi);
-
- if (this.factories.containsKey(key)) {
- throw new IllegalArgumentException("Specified AFI/SAFI combination is already registered");
- }
-
- this.factories.put(key, factory);
-
- return new AbstractRegistration() {
- @Override
- protected void removeRegistration() {
- synchronized (SimpleRIBExtensionProviderContext.this) {
- SimpleRIBExtensionProviderContext.this.factories.remove(key);
- }
- }
- };
- }
-
- @Override
- public final synchronized AdjRIBsFactory getAdjRIBsInFactory(final Class<? extends AddressFamily> afi,
- final Class<? extends SubsequentAddressFamily> safi) {
- return this.factories.get(new TablesKey(afi, safi));
- }
+ private final ConcurrentMap<TablesKey, RIBSupport<?, ?, ?, ?>> supports = new ConcurrentHashMap<>();
+ private final ConcurrentMap<NodeIdentifierWithPredicates, RIBSupport<?, ?, ?, ?>> domSupports =
+ new ConcurrentHashMap<>();
@Override
- public <T extends RIBSupport> RIBSupportRegistration<T> registerRIBSupport(final Class<? extends AddressFamily> afi,
- final Class<? extends SubsequentAddressFamily> safi, final T support) {
+ public <T extends RIBSupport<?, ?, ?, ?>> RIBSupportRegistration<T> registerRIBSupport(
+ final Class<? extends AddressFamily> afi, final Class<? extends SubsequentAddressFamily> safi,
+ final T support) {
final TablesKey key = new TablesKey(afi, safi);
- final RIBSupport prev = this.supports.putIfAbsent(key, support);
- addClassLoadingSupport(afi, safi, support);
- Preconditions.checkArgument(prev == null, "AFI %s SAFI %s is already registered with %s", afi, safi, prev);
-
- return new AbstractRIBSupportRegistration<T>(support) {
+ final RIBSupport<?, ?, ?, ?> prev = this.supports.putIfAbsent(key, support);
+ checkArgument(prev == null, "AFI %s SAFI %s is already registered with %s", afi, safi, prev);
+ this.domSupports.put(RibSupportUtils.toYangTablesKey(afi, safi), support);
+ return new AbstractRIBSupportRegistration<>(support) {
@Override
protected void removeRegistration() {
+ // FIXME: clean up registrations, too
SimpleRIBExtensionProviderContext.this.supports.remove(key);
}
};
}
- private void addClassLoadingSupport(Class<?> afi, Class<?> safi, RIBSupport s) {
- final Set<YangModuleInfo> moduleInfos =
- getModuleInfos(afi, safi, s.routesListClass(), s.routesContainerClass(), s.routesCaseClass());
- if(!moduleInfos.isEmpty()) {
- classLoadingStrategy.addModuleInfos(moduleInfos);
- }
- }
-
- private static Set<YangModuleInfo> getModuleInfos(Class<?>... clazzes) {
- final Set<YangModuleInfo> moduleInfos = new HashSet<>();
- for(Class<?> clz : clazzes) {
- try {
- moduleInfos.add(BindingReflections.getModuleInfo(clz));
- } catch (Exception e) {
- LOG.debug("Could not find module info for class {}", clz, e);
- }
- }
- return moduleInfos;
- }
-
@Override
- public RIBSupport getRIBSupport(final Class<? extends AddressFamily> afi, final Class<? extends SubsequentAddressFamily> safi) {
+ public <C extends Routes & DataObject & ChoiceIn<Tables>, S extends ChildOf<C>,
+ R extends Route & ChildOf<S> & Identifiable<I>, I extends Identifier<R>> RIBSupport<C, S, R, I> getRIBSupport(
+ final Class<? extends AddressFamily> afi, final Class<? extends SubsequentAddressFamily> safi) {
return getRIBSupport(new TablesKey(afi, safi));
}
@Override
- public RIBSupport getRIBSupport(final TablesKey key) {
- return this.supports.get(Preconditions.checkNotNull(key));
+ @SuppressWarnings("unchecked")
+ public <C extends Routes & DataObject & ChoiceIn<Tables>, S extends ChildOf<C>,
+ R extends Route & ChildOf<S> & Identifiable<I>, I extends Identifier<R>> RIBSupport<C, S, R, I> getRIBSupport(
+ final TablesKey key) {
+ return (RIBSupport<C, S, R, I>) this.supports.get(requireNonNull(key));
}
@Override
- public GeneratedClassLoadingStrategy getClassLoadingStrategy() {
- return classLoadingStrategy;
+ @SuppressWarnings("unchecked")
+ public <C extends Routes & DataObject & ChoiceIn<Tables>, S extends ChildOf<C>,
+ R extends Route & ChildOf<S> & Identifiable<I>, I extends Identifier<R>> RIBSupport<C, S, R, I> getRIBSupport(
+ final NodeIdentifierWithPredicates key) {
+ return (RIBSupport<C, S, R, I>) this.domSupports.get(key);
}
}