bgp/parser-spi/src/main/java/org/opendaylight/protocol/bgp/parser/spi/pojo/AbstractFamilyRegistry.java

   1 /*
   2  * Copyright (c) 2013 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 package org.opendaylight.protocol.bgp.parser.spi.pojo;
   9
  10 import com.google.common.base.Preconditions;
  11
  12 import java.util.Map;
  13 import java.util.concurrent.ConcurrentHashMap;
  14
  15 import org.opendaylight.protocol.concepts.AbstractRegistration;
  16
  17 abstract class AbstractFamilyRegistry<C, N> {
  18     private final Map<Class<? extends C>, N> classToNumber = new ConcurrentHashMap<>();
  19     private final Map<N, Class<? extends C>> numberToClass = new ConcurrentHashMap<>();
  20
  21     protected synchronized AutoCloseable registerFamily(final Class<? extends C> clazz, final N number) {
  22         Preconditions.checkNotNull(clazz);
  23
  24         final Class<?> c = this.numberToClass.get(number);
  25         Preconditions.checkState(c == null, "Number " + number + " already registered to " + c);
  26
  27         final N n = this.classToNumber.get(clazz);
  28         Preconditions.checkState(n == null, "Class " + clazz + " already registered to " + n);
  29
  30         this.numberToClass.put(number, clazz);
  31         this.classToNumber.put(clazz, number);
  32
  33         final Object lock = this;
  34         return new AbstractRegistration() {
  35
  36             @Override
  37             protected void removeRegistration() {
  38                 synchronized (lock) {
  39                     AbstractFamilyRegistry.this.classToNumber.remove(clazz);
  40                     AbstractFamilyRegistry.this.numberToClass.remove(number);
  41                 }
  42             }
  43         };
  44     }
  45
  46     protected Class<? extends C> classForFamily(final N number) {
  47         return this.numberToClass.get(number);
  48     }
  49
  50     protected N numberForClass(final Class<? extends C> clazz) {
  51         return this.classToNumber.get(clazz);
  52     }
  53 }