2 * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.protocol.bgp.parser.spi.pojo;
10 import com.google.common.base.Preconditions;
11 import com.google.common.cache.Cache;
12 import com.google.common.cache.CacheBuilder;
13 import java.util.concurrent.atomic.AtomicReference;
14 import org.opendaylight.protocol.bgp.parser.spi.AttributeParser;
15 import org.opendaylight.protocol.bgp.parser.spi.AttributeSerializer;
16 import org.opendaylight.protocol.bgp.parser.spi.BGPExtensionProviderContext;
17 import org.opendaylight.protocol.bgp.parser.spi.CapabilityParser;
18 import org.opendaylight.protocol.bgp.parser.spi.CapabilitySerializer;
19 import org.opendaylight.protocol.bgp.parser.spi.MessageParser;
20 import org.opendaylight.protocol.bgp.parser.spi.MessageSerializer;
21 import org.opendaylight.protocol.bgp.parser.spi.NlriParser;
22 import org.opendaylight.protocol.bgp.parser.spi.NlriSerializer;
23 import org.opendaylight.protocol.bgp.parser.spi.ParameterParser;
24 import org.opendaylight.protocol.bgp.parser.spi.ParameterSerializer;
25 import org.opendaylight.protocol.util.ReferenceCache;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.BgpParameters;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.AddressFamily;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.SubsequentAddressFamily;
29 import org.opendaylight.yangtools.yang.binding.DataObject;
30 import org.opendaylight.yangtools.yang.binding.Notification;
32 public class SimpleBGPExtensionProviderContext extends SimpleBGPExtensionConsumerContext implements BGPExtensionProviderContext {
34 public static final int DEFAULT_MAXIMUM_CACHED_OBJECTS = 100000;
36 private final AtomicReference<Cache<Object, Object>> cacheRef;
37 private final ReferenceCache referenceCache = new ReferenceCache() {
39 public <T> T getSharedReference(final T object) {
40 final Cache<Object, Object> cache = SimpleBGPExtensionProviderContext.this.cacheRef.get();
42 @SuppressWarnings("unchecked")
43 final T ret = (T) cache.getIfPresent(object);
45 cache.put(object, object);
52 private final int maximumCachedObjects;
54 public SimpleBGPExtensionProviderContext() {
55 this(DEFAULT_MAXIMUM_CACHED_OBJECTS);
58 public SimpleBGPExtensionProviderContext(final int maximumCachedObjects) {
59 this.maximumCachedObjects = maximumCachedObjects;
61 final Cache<Object, Object> cache = CacheBuilder.newBuilder().maximumSize(maximumCachedObjects).build();
62 this.cacheRef = new AtomicReference<Cache<Object,Object>>(cache);
66 public AutoCloseable registerAddressFamily(final Class<? extends AddressFamily> clazz, final int number) {
67 return this.getAddressFamilyRegistry().registerAddressFamily(clazz, number);
71 public AutoCloseable registerAttributeParser(final int attributeType, final AttributeParser parser) {
72 return this.getAttributeRegistry().registerAttributeParser(attributeType, parser);
76 public AutoCloseable registerAttributeSerializer(final Class<? extends DataObject> attributeClass, final AttributeSerializer serializer) {
77 return this.getAttributeRegistry().registerAttributeSerializer(attributeClass, serializer);
81 public AutoCloseable registerCapabilityParser(final int capabilityType, final CapabilityParser parser) {
82 return this.getCapabilityRegistry().registerCapabilityParser(capabilityType, parser);
86 public AutoCloseable registerCapabilitySerializer(final Class<? extends DataObject> capabilityClass, final CapabilitySerializer serializer) {
87 return this.getCapabilityRegistry().registerCapabilitySerializer(capabilityClass, serializer);
91 public AutoCloseable registerMessageParser(final int messageType, final MessageParser parser) {
92 return this.getMessageRegistry().registerMessageParser(messageType, parser);
96 public AutoCloseable registerMessageSerializer(final Class<? extends Notification> messageClass, final MessageSerializer serializer) {
97 return this.getMessageRegistry().registerMessageSerializer(messageClass, serializer);
101 public AutoCloseable registerNlriParser(final Class<? extends AddressFamily> afi, final Class<? extends SubsequentAddressFamily> safi,
102 final NlriParser parser) {
103 return this.getNlriRegistry().registerNlriParser(afi, safi, parser);
107 public AutoCloseable registerNlriSerializer(final Class<? extends DataObject> nlriClass, final NlriSerializer serializer) {
108 return this.getNlriRegistry().registerNlriSerializer(nlriClass,serializer);
112 public AutoCloseable registerParameterParser(final int parameterType, final ParameterParser parser) {
113 return this.getParameterRegistry().registerParameterParser(parameterType, parser);
117 public AutoCloseable registerParameterSerializer(final Class<? extends BgpParameters> paramClass, final ParameterSerializer serializer) {
118 return this.getParameterRegistry().registerParameterSerializer(paramClass, serializer);
122 public AutoCloseable registerSubsequentAddressFamily(final Class<? extends SubsequentAddressFamily> clazz, final int number) {
123 return this.getSubsequentAddressFamilyRegistry().registerSubsequentAddressFamily(clazz, number);
127 public ReferenceCache getReferenceCache() {
128 return this.referenceCache;
131 public final synchronized int getMaximumCachedObjects() {
132 return this.maximumCachedObjects;
135 public final synchronized void setMaximumCachedObjects(final int maximumCachedObjects) {
136 Preconditions.checkArgument(maximumCachedObjects >= 0);
138 Cache<Object, Object> newCache = CacheBuilder.newBuilder().maximumSize(maximumCachedObjects).build();
139 newCache.putAll(this.cacheRef.get().asMap());
140 this.cacheRef.set(newCache);