OpenConfig BGP more defensive
[bgpcep.git] / bgp / openconfig-impl / src / main / java / org / opendaylight / protocol / bgp / openconfig / impl / BGPConfigStateStoreImpl.java
1 /*
2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
3  *
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
7  */
8
9 package org.opendaylight.protocol.bgp.openconfig.impl;
10
11 import java.util.HashMap;
12 import java.util.Map;
13 import org.opendaylight.protocol.bgp.openconfig.impl.comparator.OpenConfigComparatorFactory;
14 import org.opendaylight.protocol.bgp.openconfig.impl.spi.BGPConfigHolder;
15 import org.opendaylight.protocol.bgp.openconfig.impl.spi.BGPConfigStateStore;
16 import org.opendaylight.yangtools.yang.binding.DataObject;
17
18 final class BGPConfigStateStoreImpl implements BGPConfigStateStore {
19
20     final Map<Class<? extends DataObject>, BGPConfigHolder<? extends DataObject>> configHolderMap = new HashMap<>();
21
22     @SuppressWarnings("unchecked")
23     @Override
24     public <T extends DataObject> BGPConfigHolder<T> getBGPConfigHolder(final Class<T> clazz) {
25         return (BGPConfigHolder<T>) configHolderMap.get(clazz);
26     }
27
28     @Override
29     public <T extends DataObject> void registerBGPConfigHolder(final Class<T> clazz) {
30         configHolderMap.put(clazz, new BGPConfigHolderImpl<T>(OpenConfigComparatorFactory.getComparator(clazz)));
31     }
32
33
34 }