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
  17 import org.opendaylight.mdsal.binding.api.DataBroker;
  18 import org.opendaylight.netvirt.aclservice.api.utils.AclInterface;
  19 import org.opendaylight.netvirt.aclservice.listeners.AclInterfaceListener;
  20 import org.opendaylight.netvirt.aclservice.utils.AclDataUtil;
  21 import org.opendaylight.netvirt.aclservice.utils.AclServiceUtils;
  22 import org.opendaylight.serviceutils.srm.ServiceRecoveryInterface;
  23 import org.opendaylight.serviceutils.srm.ServiceRecoveryRegistry;
  24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.Interface;
  25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.interfaces.rev140508.interfaces.InterfaceBuilder;
  26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.aclservice.rev160608.InterfaceAcl;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.netvirt.aclservice.rev160608.InterfaceAclBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.serviceutils.srm.types.rev180626.NetvirtAclInstance;
  30 import org.slf4j.Logger;
  31 import org.slf4j.LoggerFactory;
  32
  33
  34
  35 @Singleton
  36 public class AclInstanceRecoveryHandler implements ServiceRecoveryInterface {
  37
  38     private static final Logger LOG = LoggerFactory.getLogger(AclInstanceRecoveryHandler.class);
  39     private final DataBroker dataBroker;
  40     private final AclDataUtil aclDataUtil;
  41     private final AclInterfaceListener aclInterfaceListener;
  42
  43     @Inject
  44     public AclInstanceRecoveryHandler(ServiceRecoveryRegistry serviceRecoveryRegistry, DataBroker dataBroker,
  45              AclDataUtil aclDataUtil, AclInterfaceListener aclInterfaceListener) {
  46         this.dataBroker = dataBroker;
  47         this.aclDataUtil = aclDataUtil;
  48         this.aclInterfaceListener = aclInterfaceListener;
  49         serviceRecoveryRegistry.registerServiceRecoveryRegistry(buildServiceRegistryKey(), this);
  50     }
  51
  52     @Override
  53     public void recoverService(String entityId) {
  54         LOG.info("Recover ACL instance {}", entityId);
  55         Uuid aclId = new Uuid(entityId);
  56         Collection<AclInterface> aclInterfaces = aclDataUtil.getInterfaceList(aclId);
  57         for (AclInterface aclInterface : aclInterfaces) {
  58             String aclInterfaceId = aclInterface.getInterfaceId();
  59             Optional<Interface> interfaceOptional = AclServiceUtils.getInterface(dataBroker,
  60                     aclInterfaceId);
  61             if (interfaceOptional.isPresent()) {
  62                 Interface interfaceBefore = interfaceOptional.get();
  63                 LOG.debug("Starting Recovery of acl Instance {} for interface {}", entityId, interfaceBefore.getName());
  64                 InterfaceAcl interfaceAclBefore = interfaceBefore.augmentation(InterfaceAcl.class);
  65                 List<Uuid> sgList = interfaceAclBefore.getSecurityGroups() != null ? new ArrayList<>(
  66                     interfaceAclBefore.getSecurityGroups()) : new ArrayList<>();
  67                 sgList.remove(aclId);
  68                 InterfaceAcl interfaceAclAfter = new InterfaceAclBuilder(interfaceAclBefore).setSecurityGroups(sgList)
  69                         .build();
  70                 Interface interfaceAfter = new InterfaceBuilder(interfaceBefore)
  71                         .addAugmentation(InterfaceAcl.class,interfaceAclAfter).build();
  72                 aclInterfaceListener.update(null, interfaceBefore, interfaceAfter);
  73                 aclInterfaceListener.update(null, interfaceAfter, interfaceBefore);
  74             } else {
  75                 LOG.error("Interfaces not present for aclInterface {} ", aclInterfaceId);
  76             }
  77         }
  78     }
  79
  80     private String buildServiceRegistryKey() {
  81         return NetvirtAclInstance.class.toString();
  82     }
  83 }