BUG-4084: Li:Save groups in operational DS based on barrier success
[openflowplugin.git] / openflowplugin-impl / src / main / java / org / opendaylight / openflowplugin / impl / services / SalGroupServiceImpl.java
index 2a6b134b702455415c595caf44f401f9de4bacd7..1e0dcd7e52c01c87982cf0125da9d5aed8a6f324 100644 (file)
@@ -8,8 +8,22 @@
 package org.opendaylight.openflowplugin.impl.services;
 
 import java.util.concurrent.Future;
+
+import com.google.common.annotations.VisibleForTesting;
+import com.google.common.util.concurrent.FutureCallback;
+import com.google.common.util.concurrent.Futures;
+import com.google.common.util.concurrent.ListenableFuture;
 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
 import org.opendaylight.openflowplugin.api.openflow.device.RequestContextStack;
+import org.opendaylight.openflowplugin.api.openflow.registry.flow.FlowDescriptor;
+import org.opendaylight.openflowplugin.api.openflow.rpc.ItemLifeCycleSource;
+import org.opendaylight.openflowplugin.api.openflow.rpc.listener.ItemLifecycleListener;
+import org.opendaylight.openflowplugin.impl.registry.flow.FlowDescriptorFactory;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.Flow;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.FlowBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.FlowKey;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.AddGroupInput;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.AddGroupOutput;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.RemoveGroupInput;
@@ -18,33 +32,122 @@ import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.Sal
 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.UpdateGroupInput;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.service.rev130918.UpdateGroupOutput;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.Group;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.GroupId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.groups.GroupBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.groups.GroupKey;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
+import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
 import org.opendaylight.yangtools.yang.common.RpcResult;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import javax.annotation.Nullable;
 
-public class SalGroupServiceImpl implements SalGroupService {
+public class SalGroupServiceImpl implements SalGroupService, ItemLifeCycleSource {
+    private static final Logger LOG = LoggerFactory.getLogger(SalGroupServiceImpl.class);
     private final GroupService<AddGroupInput, AddGroupOutput> addGroup;
     private final GroupService<Group, UpdateGroupOutput> updateGroup;
     private final GroupService<RemoveGroupInput, RemoveGroupOutput> removeGroup;
+    private final DeviceContext deviceContext;
+    private ItemLifecycleListener itemLifecycleListener;
 
     public SalGroupServiceImpl(final RequestContextStack requestContextStack, final DeviceContext deviceContext) {
+        this.deviceContext = deviceContext;
         addGroup = new GroupService<>(requestContextStack, deviceContext, AddGroupOutput.class);
         updateGroup = new GroupService<>(requestContextStack, deviceContext, UpdateGroupOutput.class);
         removeGroup = new GroupService<>(requestContextStack, deviceContext, RemoveGroupOutput.class);
     }
 
+    @Override
+    public void setItemLifecycleListener(@Nullable ItemLifecycleListener itemLifecycleListener) {
+        this.itemLifecycleListener = itemLifecycleListener;
+    }
+
     @Override
     public Future<RpcResult<AddGroupOutput>> addGroup(final AddGroupInput input) {
         addGroup.getDeviceContext().getDeviceGroupRegistry().store(input.getGroupId());
-        return addGroup.handleServiceCall(input);
+        final ListenableFuture<RpcResult<AddGroupOutput>> resultFuture = addGroup.handleServiceCall(input);
+        Futures.addCallback(resultFuture, new FutureCallback<RpcResult<AddGroupOutput>>() {
+            @Override
+            public void onSuccess(RpcResult<AddGroupOutput> result) {
+                if (result.isSuccessful()) {
+                    LOG.debug("group add finished without error, id={}", input.getGroupId().getValue());
+                    addIfNecessaryToDS(input.getGroupId(), input);
+                }
+            }
+
+            @Override
+            public void onFailure(Throwable t) {
+                LOG.error("group add failed for id={}. Exception: {}", input.getGroupId().getValue(), t);
+            }
+        });
+
+        return resultFuture;
     }
 
+
     @Override
     public Future<RpcResult<UpdateGroupOutput>> updateGroup(final UpdateGroupInput input) {
-        return updateGroup.handleServiceCall(input.getUpdatedGroup());
+        final ListenableFuture<RpcResult<UpdateGroupOutput>> resultFuture = updateGroup.handleServiceCall(input.getUpdatedGroup());
+        Futures.addCallback(resultFuture, new FutureCallback<RpcResult<UpdateGroupOutput>>() {
+            @Override
+            public void onSuccess(@Nullable RpcResult<UpdateGroupOutput> result) {
+                if (result.isSuccessful()) {
+                    LOG.debug("Group update succeded");
+                    removeIfNecessaryFromDS(input.getOriginalGroup().getGroupId());
+                    addIfNecessaryToDS(input.getUpdatedGroup().getGroupId(), input.getUpdatedGroup());
+                }
+            }
+
+            @Override
+            public void onFailure(Throwable t) {
+                LOG.debug("Group update failed for id={}. Exception: {}", input.getOriginalGroup().getGroupId(), t);
+            }
+        });
+        return resultFuture;
     }
 
     @Override
     public Future<RpcResult<RemoveGroupOutput>> removeGroup(final RemoveGroupInput input) {
         removeGroup.getDeviceContext().getDeviceGroupRegistry().markToBeremoved(input.getGroupId());
-        return removeGroup.handleServiceCall(input);
+        final ListenableFuture<RpcResult<RemoveGroupOutput>> resultFuture = removeGroup.handleServiceCall(input);
+        Futures.addCallback(resultFuture, new FutureCallback<RpcResult<RemoveGroupOutput>>() {
+            @Override
+            public void onSuccess(@Nullable RpcResult<RemoveGroupOutput> result) {
+                if (result.isSuccessful()) {
+                    LOG.debug("Group remove succeded");
+                    removeIfNecessaryFromDS(input.getGroupId());
+                }
+            }
+
+            @Override
+            public void onFailure(Throwable t) {
+                LOG.error("Group remove failed for id={}. Exception: {}", input.getGroupId(), t);
+            }
+        });
+        return resultFuture;
+    }
+
+    private void removeIfNecessaryFromDS(final GroupId groupId) {
+        if (itemLifecycleListener != null) {
+            KeyedInstanceIdentifier<org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.groups.Group, GroupKey> groupPath
+                    = createGroupPath(groupId,
+                    deviceContext.getDeviceState().getNodeInstanceIdentifier());
+            itemLifecycleListener.onRemoved(groupPath);
+        }
+    }
+
+    private void addIfNecessaryToDS(final GroupId groupId, final Group data) {
+        if (itemLifecycleListener != null) {
+            KeyedInstanceIdentifier<org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.groups.Group, GroupKey> groupPath
+                    = createGroupPath(groupId,
+                    deviceContext.getDeviceState().getNodeInstanceIdentifier());
+            itemLifecycleListener.onAdded(groupPath, new GroupBuilder(data).build());
+        }
+    }
+
+    static KeyedInstanceIdentifier<org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.groups.Group, GroupKey> createGroupPath(final GroupId groupId, final KeyedInstanceIdentifier<Node, NodeKey> nodePath) {
+        return nodePath.augmentation(FlowCapableNode.class).child(org.opendaylight.yang.gen.v1.urn.opendaylight.group.types.rev131018.groups.Group.class, new GroupKey(groupId));
     }
 }