aclservice/impl/src/main/java/org/opendaylight/netvirt/aclservice/recovery/AclInstanceRecoveryHandler.java

   1 /*
   2  * Copyright (c) 2018 Ericsson India Global Services Pvt Ltd. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.netvirt.aclservice.recovery;
   9
  10 import java.util.ArrayList;
  11 import java.util.Collection;
  12 import java.util.List;
  13 import java.util.Optional;
  14 import javax.inject.Inject;
  15 import javax.inject.Singleton;
  16 import org.opendaylight.mdsal.binding.api.DataBroker;
  17 import org.opendaylight.netvirt.aclservice.api.utils.AclInterface;
  18 import org.opendaylight.netvirt.aclservice.listeners.AclInterfaceListener;
  19 import org.opendaylight.netvirt.aclservice.utils.AclDataUtil;
  20 import org.opendaylight.netvirt.aclservice.utils.AclServiceUtils;
  21 import org.opendaylight.serviceutils.srm.ServiceRecoveryInterface;
  22 import org.opendaylight.serviceutils.srm.ServiceRecoveryRegistry;
  23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.Interface;
  24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.InterfaceBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.aclservice.rev160608.InterfaceAcl;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.aclservice.rev160608.InterfaceAclBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.serviceutils.srm.types.rev180626.NetvirtAclInstance;
  29 import org.slf4j.Logger;
  30 import org.slf4j.LoggerFactory;
  31
  32
  33
  34 @Singleton
  35 public class AclInstanceRecoveryHandler implements ServiceRecoveryInterface {
  36
  37     private static final Logger LOG = LoggerFactory.getLogger(AclInstanceRecoveryHandler.class);
  38     private final DataBroker dataBroker;
  39     private final AclDataUtil aclDataUtil;
  40     private final AclInterfaceListener aclInterfaceListener;
  41
  42     @Inject
  43     public AclInstanceRecoveryHandler(ServiceRecoveryRegistry serviceRecoveryRegistry, DataBroker dataBroker,
  44              AclDataUtil aclDataUtil, AclInterfaceListener aclInterfaceListener) {
  45         this.dataBroker = dataBroker;
  46         this.aclDataUtil = aclDataUtil;
  47         this.aclInterfaceListener = aclInterfaceListener;
  48         serviceRecoveryRegistry.registerServiceRecoveryRegistry(buildServiceRegistryKey(), this);
  49     }
  50
  51     @Override
  52     public void recoverService(String entityId) {
  53         LOG.info("Recover ACL instance {}", entityId);
  54         Uuid aclId = new Uuid(entityId);
  55         Collection<AclInterface> aclInterfaces = aclDataUtil.getInterfaceList(aclId);
  56         for (AclInterface aclInterface : aclInterfaces) {
  57             String aclInterfaceId = aclInterface.getInterfaceId();
  58             Optional<Interface> interfaceOptional = AclServiceUtils.getInterface(dataBroker,
  59                     aclInterfaceId);
  60             if (interfaceOptional.isPresent()) {
  61                 Interface interfaceBefore = interfaceOptional.get();
  62                 LOG.debug("Starting Recovery of acl Instance {} for interface {}", entityId, interfaceBefore.getName());
  63                 InterfaceAcl interfaceAclBefore = interfaceBefore.augmentation(InterfaceAcl.class);
  64                 List<Uuid> sgList = interfaceAclBefore.getSecurityGroups() != null ? new ArrayList<>(
  65                     interfaceAclBefore.getSecurityGroups()) : new ArrayList<>();
  66                 sgList.remove(aclId);
  67                 InterfaceAcl interfaceAclAfter = new InterfaceAclBuilder(interfaceAclBefore).setSecurityGroups(sgList)
  68                         .build();
  69                 Interface interfaceAfter = new InterfaceBuilder(interfaceBefore)
  70                         .addAugmentation(InterfaceAcl.class,interfaceAclAfter).build();
  71                 aclInterfaceListener.update(null, interfaceBefore, interfaceAfter);
  72                 aclInterfaceListener.update(null, interfaceAfter, interfaceBefore);
  73             } else {
  74                 LOG.error("Interfaces not present for aclInterface {} ", aclInterfaceId);
  75             }
  76         }
  77     }
  78
  79     private String buildServiceRegistryKey() {
  80         return NetvirtAclInstance.class.toString();
  81     }
  82 }