openflow-protocol-impl/src/main/java/org/opendaylight/openflowjava/protocol/impl/serialization/SerializerRegistryImpl.java

   1 /*
   2  * Copyright (c) 2013 Pantheon Technologies s.r.o. 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.openflowjava.protocol.impl.serialization;
  10
  11 import java.util.HashMap;
  12 import java.util.Map;
  13
  14 import org.opendaylight.openflowjava.protocol.api.extensibility.MessageTypeKey;
  15 import org.opendaylight.openflowjava.protocol.api.extensibility.OFGeneralSerializer;
  16 import org.opendaylight.openflowjava.protocol.api.extensibility.SerializerRegistry;
  17 import org.opendaylight.openflowjava.protocol.api.extensibility.SerializerRegistryInjector;
  18 import org.opendaylight.openflowjava.protocol.api.util.EncodeConstants;
  19 import org.opendaylight.openflowjava.protocol.impl.util.OF10MatchSerializer;
  20 import org.opendaylight.openflowjava.protocol.impl.util.OF13MatchSerializer;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev130731.match.grouping.Match;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.oxm.rev130731.match.v10.grouping.MatchV10;
  23
  24 /**
  25  * Stores and handles serializers
  26  * @author michal.polkorab
  27  * @author timotej.kubas
  28  */
  29 public class SerializerRegistryImpl implements SerializerRegistry {
  30
  31     private static final short OF10 = EncodeConstants.OF10_VERSION_ID;
  32     private static final short OF13 = EncodeConstants.OF13_VERSION_ID;
  33     private Map<MessageTypeKey<?>, OFGeneralSerializer> registry;
  34
  35
  36     @Override
  37     public void init() {
  38         registry = new HashMap<>();
  39         // Openflow message type serializers
  40         MessageFactoryInitializer.registerMessageSerializers(this);
  41
  42         // match structure serializers
  43         registerSerializer(new MessageTypeKey<>(OF10, MatchV10.class), new OF10MatchSerializer());
  44         registerSerializer(new MessageTypeKey<>(OF13, Match.class), new OF13MatchSerializer());
  45
  46         // match entry serializers
  47         MatchEntriesInitializer.registerMatchEntrySerializers(this);
  48         // action serializers
  49         ActionsInitializer.registerActionSerializers(this);
  50         // instruction serializers
  51         InstructionsInitializer.registerInstructionSerializers(this);
  52     }
  53
  54     /**
  55      * @param msgTypeKey
  56      * @return encoder for current type of message (msgTypeKey)
  57      */
  58     @Override
  59     @SuppressWarnings("unchecked")
  60     public <KEY_TYPE, SERIALIZER_TYPE extends OFGeneralSerializer> SERIALIZER_TYPE getSerializer(
  61             MessageTypeKey<KEY_TYPE> msgTypeKey) {
  62         OFGeneralSerializer serializer = registry.get(msgTypeKey);
  63         if (serializer == null) {
  64             throw new NullPointerException("Serializer for key: " + msgTypeKey.toString()
  65                     + " was not found");
  66         }
  67         return (SERIALIZER_TYPE) serializer;
  68     }
  69
  70     @Override
  71     public <KEY_TYPE> void registerSerializer(
  72             MessageTypeKey<KEY_TYPE> msgTypeKey, OFGeneralSerializer serializer) {
  73         if ((msgTypeKey == null) || (serializer == null)) {
  74             throw new NullPointerException("MessageTypeKey or Serializer is null");
  75         }
  76         if (serializer instanceof SerializerRegistryInjector) {
  77             ((SerializerRegistryInjector) serializer).injectSerializerRegistry(this);
  78         }
  79         registry.put(msgTypeKey, serializer);
  80     }
  81
  82     @Override
  83     public <KEY_TYPE> boolean unregisterSerializer(MessageTypeKey<KEY_TYPE> msgTypeKey) {
  84         if (msgTypeKey == null) {
  85             throw new NullPointerException("MessageTypeKey is null");
  86         }
  87         OFGeneralSerializer serializer = registry.remove(msgTypeKey);
  88         if (serializer == null) {
  89             return false;
  90         }
  91         return true;
  92     }
  93 }