Serializer for DefaultAnnotations. This allows Annotated objects to be correctly
used as keys in ConsistentMaps. Fixes ONOS-2128.
Change-Id: Ia21a25712351b99bc1b79dc231be8187cf3d3a0b
diff --git a/core/store/serializers/src/main/java/org/onosproject/store/serializers/AnnotationsSerializer.java b/core/store/serializers/src/main/java/org/onosproject/store/serializers/AnnotationsSerializer.java
new file mode 100644
index 0000000..febb4ad
--- /dev/null
+++ b/core/store/serializers/src/main/java/org/onosproject/store/serializers/AnnotationsSerializer.java
@@ -0,0 +1,32 @@
+package org.onosproject.store.serializers;
+
+import org.onosproject.net.DefaultAnnotations;
+
+import com.esotericsoftware.kryo.Kryo;
+import com.esotericsoftware.kryo.Serializer;
+import com.esotericsoftware.kryo.io.Input;
+import com.esotericsoftware.kryo.io.Output;
+
+import java.util.HashMap;
+
+public class AnnotationsSerializer extends Serializer<DefaultAnnotations> {
+
+ public AnnotationsSerializer() {
+ super(false, true);
+ }
+
+ @Override
+ public void write(Kryo kryo, Output output, DefaultAnnotations object) {
+ kryo.writeObject(output, object.asMap());
+ }
+
+ @Override
+ public DefaultAnnotations read(Kryo kryo, Input input, Class<DefaultAnnotations> type) {
+ DefaultAnnotations.Builder b = DefaultAnnotations.builder();
+ HashMap<String, String> map = kryo.readObject(input, HashMap.class);
+ map.forEach((k, v) -> b.set(k, v));
+
+ return b.build();
+ }
+
+}