Merge dev/fluorine work across to master
[unimgr.git] / legato-api / src / main / java / org / opendaylight / unimgr / mef / legato / LegatoCosProfileController.java
diff --git a/legato-api/src/main/java/org/opendaylight/unimgr/mef/legato/LegatoCosProfileController.java b/legato-api/src/main/java/org/opendaylight/unimgr/mef/legato/LegatoCosProfileController.java
new file mode 100644 (file)
index 0000000..5429cd2
--- /dev/null
@@ -0,0 +1,129 @@
+/*
+ * Copyright (c) 2018 Xoriant Corporation and others. All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.unimgr.mef.legato;
+
+import com.google.common.base.Optional;
+
+import java.util.Collections;
+
+import org.opendaylight.controller.md.sal.binding.api.DataBroker;
+import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
+import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
+import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
+import org.opendaylight.unimgr.api.UnimgrDataTreeChangeListener;
+import org.opendaylight.unimgr.mef.legato.util.LegatoConstants;
+import org.opendaylight.unimgr.mef.legato.util.LegatoUtils;
+import org.opendaylight.yang.gen.v1.urn.mef.yang.mef.global.rev171215.MefGlobal;
+import org.opendaylight.yang.gen.v1.urn.mef.yang.mef.global.rev171215.mef.global.CosProfiles;
+import org.opendaylight.yang.gen.v1.urn.mef.yang.mef.global.rev171215.mef.global.CosProfilesBuilder;
+import org.opendaylight.yang.gen.v1.urn.mef.yang.mef.global.rev171215.mef.global.cos.profiles.Profile;
+import org.opendaylight.yang.gen.v1.urn.mef.yang.mef.global.rev171215.mef.global.cos.profiles.ProfileKey;
+import org.opendaylight.yangtools.concepts.ListenerRegistration;
+import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+
+/**
+ * @author Arif.Hussain@Xoriant.Com
+ */
+public class LegatoCosProfileController extends UnimgrDataTreeChangeListener<Profile> {
+
+    public LegatoCosProfileController(DataBroker dataBroker) {
+        super(dataBroker);
+        registerListener();
+    }
+
+    private static final Logger LOG = LoggerFactory.getLogger(LegatoCosProfileController.class);
+    private static final InstanceIdentifier<Profile> PROFILE_IID = InstanceIdentifier.builder(MefGlobal.class)
+            .child(CosProfiles.class).child(Profile.class).build();
+    private ListenerRegistration<LegatoCosProfileController> dataTreeChangeListenerRegistration;
+
+    public void registerListener() {
+        LOG.info("Initializing LegatoSlsProfileController:init() ");
+        dataTreeChangeListenerRegistration = dataBroker.registerDataTreeChangeListener(
+                new DataTreeIdentifier<Profile>(LogicalDatastoreType.CONFIGURATION, PROFILE_IID), this);
+    }
+
+    @Override
+    public void close() throws Exception {
+        if (dataTreeChangeListenerRegistration != null) {
+            dataTreeChangeListenerRegistration.close();
+        }
+    }
+
+    @Override
+    public void add(DataTreeModification<Profile> newDataObject) {
+        LOG.info("  Node Added  " + newDataObject.getRootNode().getIdentifier());
+        addToOperationalDB(newDataObject.getRootNode().getDataAfter());
+    }
+
+    public void addToOperationalDB(Profile profile) {
+        try {
+            assert profile != null;
+            CosProfiles cosProfiles = new CosProfilesBuilder().setProfile(Collections.singletonList(profile)).build();
+            InstanceIdentifier<CosProfiles> profilesTx = InstanceIdentifier.create(MefGlobal.class)
+                .child(CosProfiles.class);
+            LegatoUtils.addToOperationalDB(cosProfiles, profilesTx, dataBroker);
+        } catch (Exception ex) {
+            LOG.error("error: ", ex);
+        }
+    }
+
+    @Override
+    public void remove(DataTreeModification<Profile> removedDataObject) {
+        // TODO Auto-generated method stub
+        if (removedDataObject.getRootNode() != null && removedDataObject.getRootPath() != null) {
+            LOG.info("  Node removed  " + removedDataObject.getRootNode().getIdentifier());
+            try {
+                assert removedDataObject.getRootNode().getDataBefore() != null;
+                deleteFromOperationalDB(removedDataObject.getRootNode().getDataBefore());
+            } catch (Exception ex) {
+                LOG.error("error: ", ex);
+            }
+        }
+    }
+
+
+    public void deleteFromOperationalDB(Profile profile) {
+        assert profile != null;
+        LegatoUtils.deleteFromOperationalDB(InstanceIdentifier.create(MefGlobal.class)
+                .child(CosProfiles.class).child(Profile.class, new ProfileKey(profile.getId())),
+                dataBroker);
+    }
+
+    @Override
+    public void update(DataTreeModification<Profile> modifiedDataObject) {
+        // TODO Auto-generated method stub
+        if (modifiedDataObject.getRootNode() != null && modifiedDataObject.getRootPath() != null) {
+            LOG.info("  Node modified  " + modifiedDataObject.getRootNode().getIdentifier());
+            try {
+                assert modifiedDataObject.getRootNode().getDataAfter() != null;
+                updateFromOperationalDB(modifiedDataObject.getRootNode().getDataAfter());
+            } catch (Exception ex) {
+                LOG.error("error: ", ex);
+            }
+        }
+    }
+
+    @SuppressWarnings("unchecked")
+    public void updateFromOperationalDB(Profile profile) {
+        assert profile != null;
+        InstanceIdentifier<Profile> instanceIdentifier = InstanceIdentifier.create(MefGlobal.class)
+                .child(CosProfiles.class).child(Profile.class, new ProfileKey(profile.getId()));
+        Optional<Profile> optionalProfile =
+                (Optional<Profile>) LegatoUtils.readProfile(LegatoConstants.COS_PROFILES,
+                        dataBroker, LogicalDatastoreType.CONFIGURATION, instanceIdentifier);
+        if (optionalProfile.isPresent()) {
+            LegatoUtils.deleteFromOperationalDB(instanceIdentifier, dataBroker);
+            addToOperationalDB(optionalProfile.get());
+        }
+    }
+
+}