[VOL-4246] Feature parity with the previous implementation
Change-Id: I3741edb3c1b88b1cf8b5e6d4ff0900132e2e5e6a
diff --git a/impl/src/main/java/org/opencord/olt/impl/ServiceKeySerializer.java b/impl/src/main/java/org/opencord/olt/impl/ServiceKeySerializer.java
new file mode 100644
index 0000000..4035bdd
--- /dev/null
+++ b/impl/src/main/java/org/opencord/olt/impl/ServiceKeySerializer.java
@@ -0,0 +1,58 @@
+/*
+ * Copyright 2021-present Open Networking Foundation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.opencord.olt.impl;
+
+import com.esotericsoftware.kryo.Kryo;
+import com.esotericsoftware.kryo.io.Input;
+import com.esotericsoftware.kryo.io.Output;
+import com.esotericsoftware.kryo.Serializer;
+import org.onosproject.net.ConnectPoint;
+import org.onosproject.net.DeviceId;
+import org.onosproject.net.PortNumber;
+import org.opencord.sadis.UniTagInformation;
+
+/**
+ * ServiceKeySerializer is a custom serializer to store a ServiceKey in an Atomix ditributed map.
+ */
+class ServiceKeySerializer extends Serializer<ServiceKey> {
+
+ ServiceKeySerializer() {
+ // non-null, immutable
+ super(false, true);
+ }
+
+ @Override
+ public void write(Kryo kryo, Output output, ServiceKey object) {
+ kryo.writeClassAndObject(output, object.getPort().connectPoint().port());
+ output.writeString(object.getPort().name());
+ output.writeString(object.getPort().connectPoint().deviceId().toString());
+ kryo.writeClassAndObject(output, object.getService());
+ }
+
+ @Override
+ public ServiceKey read(Kryo kryo, Input input, Class<ServiceKey> type) {
+ PortNumber port = (PortNumber) kryo.readClassAndObject(input);
+ final String portName = input.readString();
+ final String devId = input.readString();
+
+ UniTagInformation uti = (UniTagInformation) kryo.readClassAndObject(input);
+ ConnectPoint cp = new ConnectPoint(DeviceId.deviceId(devId), port);
+ AccessDevicePort adp = new AccessDevicePort(cp, portName);
+
+ return new ServiceKey(adp, uti);
+ }
+}