1 files changed, 149 insertions, 0 deletions
diff --git a/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/IoamPotWriterCustomizer.java b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/IoamPotWriterCustomizer.java
new file mode 100644
index 000000000..6d80493a3
--- /dev/null
+++ b/ioam/impl/src/main/java/io/fd/hc2vpp/vppioam/impl/config/IoamPotWriterCustomizer.java
@@ -0,0 +1,149 @@
+/*
+ * Copyright (c) 2016 Cisco and/or its affiliates.
+ *
+ * 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 io.fd.hc2vpp.vppioam.impl.config;
+
+
+import io.fd.hc2vpp.common.translate.util.JvppReplyConsumer;
+import io.fd.hc2vpp.vppioam.impl.util.FutureJVppIoampotCustomizer;
+import io.fd.honeycomb.translate.spi.write.ListWriterCustomizer;
+import io.fd.honeycomb.translate.write.WriteContext;
+import io.fd.honeycomb.translate.write.WriteFailedException;
+import io.fd.vpp.jvpp.ioampot.dto.PotProfileAdd;
+import io.fd.vpp.jvpp.ioampot.dto.PotProfileAddReply;
+import io.fd.vpp.jvpp.ioampot.dto.PotProfileDel;
+import io.fd.vpp.jvpp.ioampot.dto.PotProfileDelReply;
+import io.fd.vpp.jvpp.ioampot.future.FutureJVppIoampot;
+import java.nio.charset.StandardCharsets;
+import javax.annotation.Nonnull;
+import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.ioam.sb.pot.rev160615.pot.profile.PotProfileList;
+import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.ioam.sb.pot.rev160615.pot.profiles.PotProfileSet;
+import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.ioam.sb.pot.rev160615.pot.profiles.PotProfileSetKey;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+
+public class IoamPotWriterCustomizer extends FutureJVppIoampotCustomizer implements
+        ListWriterCustomizer<PotProfileSet,PotProfileSetKey>, JvppReplyConsumer {
+
+    private static final Logger LOG = LoggerFactory.getLogger(IoamPotWriterCustomizer.class);
+
+    public IoamPotWriterCustomizer(@Nonnull FutureJVppIoampot futureJVppIoampot) {
+        super(futureJVppIoampot);
+    }
+
+
+    /**
+     * Handle write operation. C from CRUD.
+     *
+     * @param id           Identifier(from root) of data being written
+     * @param dataAfter    New data to be written
+     * @param writeContext Write context can be used to store any useful information and then utilized by other customizers
+     * @throws WriteFailedException if write was unsuccessful
+     */
+    @Override
+    public void writeCurrentAttributes(@Nonnull InstanceIdentifier<PotProfileSet> id,
+                                       @Nonnull PotProfileSet dataAfter, @Nonnull WriteContext writeContext)
+            throws WriteFailedException {
+        try {
+            addPotProfile(dataAfter,id);
+        } catch (WriteFailedException exCreate) {
+            LOG.error("Add POT profile failed", exCreate);
+            throw new WriteFailedException.CreateFailedException(id, dataAfter, exCreate);
+        }
+
+        LOG.info("POT profile added iid={}, added {}", id, dataAfter);
+    }
+
+    /**
+     * Handle update operation. U from CRUD.
+     *
+     * @param id           Identifier(from root) of data being written
+     * @param dataBefore   Old data
+     * @param dataAfter    New, updated data
+     * @param writeContext Write context can be used to store any useful information and then utilized by other customizers
+     * @throws WriteFailedException if update was unsuccessful
+     */
+    @Override
+    public void updateCurrentAttributes(@Nonnull InstanceIdentifier<PotProfileSet> id,
+                                        @Nonnull PotProfileSet dataBefore, @Nonnull PotProfileSet dataAfter,
+                                        @Nonnull WriteContext writeContext) throws WriteFailedException {
+        try {
+            delPotProfile(dataBefore,id);
+            addPotProfile(dataAfter,id);
+        } catch (WriteFailedException exUpdate) {
+            LOG.error("Update POT Profile failed", exUpdate);
+            throw new WriteFailedException.UpdateFailedException(id,dataBefore,dataAfter,exUpdate);
+        }
+
+        LOG.info("POT profile updated iid={}, added {}", id, dataAfter);
+    }
+
+    /**
+     * Handle delete operation. D from CRUD.
+     *
+     * @param id           Identifier(from root) of data being written
+     * @param dataBefore   Old data being deleted
+     * @param writeContext Write context can be used to store any useful information and then utilized by other customizers
+     * @throws WriteFailedException if delete was unsuccessful
+     */
+    @Override
+    public void deleteCurrentAttributes(@Nonnull InstanceIdentifier<PotProfileSet> id,
+                                        @Nonnull PotProfileSet dataBefore, @Nonnull WriteContext writeContext)
+            throws WriteFailedException {
+        try {
+            delPotProfile(dataBefore,id);
+        } catch (WriteFailedException exDelete) {
+            LOG.error("Del POT Profile failed", exDelete);
+            throw new WriteFailedException.DeleteFailedException(id, exDelete);
+        }
+
+        LOG.info("POT profile deleted iid={}, added {}", id, dataBefore);
+    }
+
+    private void addPotProfile(PotProfileSet potProfileSet, InstanceIdentifier<PotProfileSet> id) throws WriteFailedException{
+        for(PotProfileList potProfileList:potProfileSet.getPotProfileList()) {
+            writePotProfileList(potProfileList,potProfileSet.getName(),id);
+        }
+    }
+
+    private PotProfileAddReply writePotProfileList(PotProfileList potProfileList, String name,
+                                                   InstanceIdentifier<PotProfileSet> id) throws WriteFailedException{
+        PotProfileAdd request = new PotProfileAdd();
+            request.id = potProfileList.getIndex().getValue().byteValue();
+            request.validator = (byte) (potProfileList.isValidator() ? 1 : 0);
+            request.secretKey = 1;
+            request.secretShare = potProfileList.getSecretShare().longValue();
+            request.prime = potProfileList.getPrimeNumber().longValue();
+            request.secretKey = potProfileList.getValidatorKey().longValue();
+            request.maxBits = 64;
+            request.lpc = potProfileList.getLpc().longValue();
+            request.polynomialPublic = potProfileList.getPublicPolynomial().longValue();
+            request.listNameLen = (byte) name.getBytes(StandardCharsets.UTF_8).length;
+            request.listName = name.getBytes(StandardCharsets.UTF_8);
+
+            return getReplyForWrite(getFutureJVppIoampot().potProfileAdd(request).toCompletableFuture(), id);
+    }
+
+    private PotProfileDelReply delPotProfile(PotProfileSet potProfileSet, InstanceIdentifier<PotProfileSet> id)
+            throws WriteFailedException{
+        PotProfileDel request = new PotProfileDel();
+        request.listNameLen = (byte)potProfileSet.getName().getBytes(StandardCharsets.UTF_8).length;
+        request.listName = potProfileSet.getName().getBytes(StandardCharsets.UTF_8);
+
+        return getReplyForWrite(getFutureJVppIoampot().potProfileDel(request).toCompletableFuture(),id);
+    }
+}