e81d5a5ba87fc64082a89f2da7ec7017caa000be
[ovsdb.git] / of-extension / nx-ofjava / src / main / java / org / opendaylight / ovs / nx / ofjava / NiciraExtensionsRegistrator.java
1 /*
2  * Copyright (C) 2014 Red Hat, Inc.
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  * Authors : Madhu Venugopal
9  */
10 package org.opendaylight.ovs.nx.ofjava;
11 import org.opendaylight.openflowjava.nx.api.NiciraExtensionCodecRegistrator;
12 import org.opendaylight.ovs.nx.ofjava.codec.action.NiciraActionCodecs;
13 import org.opendaylight.ovs.nx.ofjava.codec.action.ResubmitCodec;
14 import org.opendaylight.ovs.nx.ofjava.codec.action.SetNsiCodec;
15 import org.opendaylight.ovs.nx.ofjava.codec.action.SetNspCodec;
16
17 import com.google.common.base.Preconditions;
18
19 public class NiciraExtensionsRegistrator implements AutoCloseable {
20
21     private final NiciraExtensionCodecRegistrator registrator;
22     public NiciraExtensionsRegistrator(NiciraExtensionCodecRegistrator registrator) {
23         Preconditions.checkNotNull(registrator);
24         this.registrator = registrator;
25     }
26
27     public void registerNiciraExtensions() {
28         registrator.registerActionDeserializer(ResubmitCodec.RESUBMIT_DESERIALIZER_KEY, NiciraActionCodecs.RESUBMIT_CODEC);
29         registrator.registerActionDeserializer(ResubmitCodec.RESUBMIT_TABLE_DESERIALIZER_KEY, NiciraActionCodecs.RESUBMIT_CODEC);
30         registrator.registerActionSerializer(ResubmitCodec.SERIALIZER_KEY, NiciraActionCodecs.RESUBMIT_CODEC);
31
32         registrator.registerActionDeserializer(SetNspCodec.DESERIALIZER_KEY, NiciraActionCodecs.SET_NSP_CODEC);
33         registrator.registerActionSerializer(SetNspCodec.SERIALIZER_KEY, NiciraActionCodecs.SET_NSP_CODEC);
34
35         registrator.registerActionDeserializer(SetNsiCodec.DESERIALIZER_KEY, NiciraActionCodecs.SET_NSI_CODEC);
36         registrator.registerActionSerializer(SetNsiCodec.SERIALIZER_KEY, NiciraActionCodecs.SET_NSI_CODEC);
37     }
38
39     public void unregisterExtensions() {
40         registrator.unregisterActionDeserializer(ResubmitCodec.RESUBMIT_DESERIALIZER_KEY);
41         registrator.unregisterActionDeserializer(ResubmitCodec.RESUBMIT_TABLE_DESERIALIZER_KEY);
42         registrator.unregisterActionSerializer(ResubmitCodec.SERIALIZER_KEY);
43         registrator.unregisterActionDeserializer(SetNsiCodec.DESERIALIZER_KEY);
44         registrator.unregisterActionSerializer(SetNsiCodec.SERIALIZER_KEY);
45         registrator.unregisterActionDeserializer(SetNspCodec.DESERIALIZER_KEY);
46         registrator.unregisterActionSerializer(SetNspCodec.SERIALIZER_KEY);
47     }
48
49     @Override
50     public void close() throws Exception {
51         unregisterExtensions();
52     }
53
54 }