groupbasedpolicy/src/main/java/org/opendaylight/groupbasedpolicy/resolver/FollowedTenantListener.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. 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
   9 package org.opendaylight.groupbasedpolicy.resolver;
  10
  11 import java.io.Closeable;
  12 import java.io.IOException;
  13 import java.util.Collection;
  14
  15 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  16 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
  17 import org.opendaylight.controller.md.sal.binding.api.DataTreeChangeListener;
  18 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
  19 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
  20 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.common.rev140421.TenantId;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.Renderers;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.Renderer;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.Interests;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.interests.FollowedTenants;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.groupbasedpolicy.renderer.rev151103.renderers.renderer.interests.followed.tenants.FollowedTenant;
  27 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  28 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  29
  30 import com.google.common.collect.HashMultiset;
  31 import com.google.common.collect.Multiset;
  32
  33 public class FollowedTenantListener implements DataTreeChangeListener<FollowedTenant>, Closeable {
  34
  35     private final PolicyResolver policyResolver;
  36     private final ListenerRegistration<FollowedTenantListener> listenerRegistration;
  37     private final Multiset<TenantId> countedFollowedTenants = HashMultiset.create();
  38
  39     public FollowedTenantListener(DataBroker dataProvider, PolicyResolver policyResolver) {
  40         this.policyResolver = policyResolver;
  41         listenerRegistration = dataProvider.registerDataTreeChangeListener(
  42                 new DataTreeIdentifier<>(LogicalDatastoreType.OPERATIONAL,
  43                         InstanceIdentifier.builder(Renderers.class)
  44                             .child(Renderer.class)
  45                             .child(Interests.class)
  46                             .child(FollowedTenants.class)
  47                             .child(FollowedTenant.class)
  48                             .build()),
  49                 this);
  50     }
  51
  52     @Override
  53     public void onDataTreeChanged(Collection<DataTreeModification<FollowedTenant>> changes) {
  54         for (DataTreeModification<FollowedTenant> change : changes) {
  55             DataObjectModification<FollowedTenant> rootNode = change.getRootNode();
  56             TenantId tenantId = change.getRootPath().getRootIdentifier().firstKeyOf(FollowedTenant.class).getId();
  57             switch (rootNode.getModificationType()) {
  58                 case WRITE:
  59                     policyResolver.subscribeTenant(tenantId);
  60                     countedFollowedTenants.add(tenantId);
  61                     break;
  62                 case DELETE:
  63                     countedFollowedTenants.remove(tenantId);
  64                     if (countedFollowedTenants.count(tenantId) == 0) {
  65                         policyResolver.unsubscribeTenant(tenantId);
  66                     }
  67                     break;
  68                 case SUBTREE_MODIFIED:
  69                     // NOOP
  70                     break;
  71             }
  72         }
  73     }
  74
  75     @Override
  76     public void close() throws IOException {
  77         if (listenerRegistration != null)
  78             listenerRegistration.close();
  79     }
  80
  81 }