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