openflow-protocol-impl/src/main/java/org/opendaylight/openflowjava/protocol/impl/util/TypeToClassInitHelper.java

   1 /*\r
   2  * Copyright (c) 2013 Pantheon Technologies s.r.o. and others.  All rights reserved.\r
   3  *\r
   4  * This program and the accompanying materials are made available under the\r
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,\r
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html\r
   7  */\r
   8 package org.opendaylight.openflowjava.protocol.impl.util;\r
   9 \r
  10 import java.util.Map;\r
  11 \r
  12 import org.opendaylight.openflowjava.protocol.impl.deserialization.TypeToClassKey;\r
  13 \r
  14 /**\r
  15  * @author michal.polkorab\r
  16  *\r
  17  */\r
  18 public class TypeToClassInitHelper {\r
  19 \r
  20     private short version;\r
  21     private Map<TypeToClassKey, Class<?>> messageClassMap;\r
  22 \r
  23     /**\r
  24      * Constructor\r
  25      * @param version protocol wire version\r
  26      * @param messageClassMap map which stores type to class mapping\r
  27      */\r
  28     public TypeToClassInitHelper(short version, Map<TypeToClassKey, \r
  29             Class<?>> messageClassMap) {\r
  30         this.version = version;\r
  31         this.messageClassMap = messageClassMap;\r
  32     }\r
  33 \r
  34     /**\r
  35      * Registers Class int the type to class mapping\r
  36      * @param type code value for message type / class\r
  37      * @param clazz corresponding code -> class\r
  38      */\r
  39     public void registerTypeToClass(short type, Class<?> clazz) {\r
  40         messageClassMap.put(new TypeToClassKey(version, type), clazz);\r
  41     }\r
  42 }\r