openstack/net-virt/src/main/java/org/opendaylight/netvirt/openstack/netvirt/translator/iaware/impl/NeutronLoadBalancerPoolMemberChangeListener.java

   1 /*
   2  * Copyright (C) 2015 Red Hat, 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.netvirt.openstack.netvirt.translator.iaware.impl;
  10
  11 import java.util.Map.Entry;
  12
  13 import org.opendaylight.controller.md.sal.binding.api.ClusteredDataChangeListener;
  14 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  15 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
  16 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker;
  17 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
  18 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  19 import org.opendaylight.netvirt.openstack.netvirt.translator.NeutronLoadBalancerPoolMember;
  20 import org.opendaylight.netvirt.openstack.netvirt.translator.iaware.INeutronLoadBalancerPoolMemberAware;
  21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev150712.lbaas.attributes.Pools;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev150712.lbaas.attributes.pools.Pool;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev150712.lbaas.attributes.pools.PoolKey;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev150712.lbaas.attributes.pools.pool.Members;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev150712.lbaas.attributes.pools.pool.members.Member;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.rev150712.Neutron;
  28 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  29 import org.opendaylight.yangtools.yang.binding.DataObject;
  30 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  31 import org.slf4j.Logger;
  32 import org.slf4j.LoggerFactory;
  33
  34 public class NeutronLoadBalancerPoolMemberChangeListener implements ClusteredDataChangeListener, AutoCloseable {
  35     private static final Logger LOG = LoggerFactory.getLogger(NeutronLoadBalancerPoolMemberChangeListener.class);
  36
  37     private ListenerRegistration<DataChangeListener> registration;
  38     private DataBroker db;
  39
  40     public NeutronLoadBalancerPoolMemberChangeListener(DataBroker db){
  41         this.db = db;
  42         InstanceIdentifier<Member> path = InstanceIdentifier
  43                 .create(Neutron.class)
  44                 .child(Pools.class)
  45                 .child(Pool.class)
  46                 .child(Members.class)
  47                 .child(Member.class);
  48         LOG.debug("Register listener for Neutron Load Balancer Pool Member model data changes");
  49         registration =
  50                 this.db.registerDataChangeListener(LogicalDatastoreType.CONFIGURATION, path, this,
  51                         AsyncDataBroker.DataChangeScope.ONE);
  52     }
  53
  54     @Override
  55     public void close() throws Exception {
  56         registration.close();
  57     }
  58
  59     @Override
  60     public void onDataChanged(AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> changes) {
  61         LOG.trace("Data changes : {}", changes);
  62
  63         Object[] subscribers = NeutronIAwareUtil.getInstances(INeutronLoadBalancerPoolMemberAware.class, this);
  64         createPoolMember(changes, subscribers);
  65         updatePoolMember(changes, subscribers);
  66         deletePoolMember(changes, subscribers);
  67     }
  68
  69     private void createPoolMember(
  70             AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> changes,
  71             Object[] subscribers) {
  72         for (Entry<InstanceIdentifier<?>, DataObject> newPoolMember : changes.getCreatedData().entrySet()) {
  73                 if(newPoolMember.getValue() instanceof Member){
  74                 NeutronLoadBalancerPoolMember neutronLBPoolMember = fromMd(newPoolMember.getKey(), (Member) newPoolMember.getValue());
  75                 for (Object entry : subscribers) {
  76                     INeutronLoadBalancerPoolMemberAware subscriber = (INeutronLoadBalancerPoolMemberAware) entry;
  77                     subscriber.neutronLoadBalancerPoolMemberCreated(neutronLBPoolMember);
  78                 }
  79                 }
  80         }
  81     }
  82     private void updatePoolMember(
  83             AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> changes,
  84             Object[] subscribers) {
  85         for (Entry<InstanceIdentifier<?>, DataObject> updatePoolMember : changes.getUpdatedData().entrySet()) {
  86                 if(updatePoolMember.getValue() instanceof Member){
  87                 NeutronLoadBalancerPoolMember neutronLBPoolMember =
  88                         fromMd(updatePoolMember.getKey(), (Member) updatePoolMember.getValue());
  89                 for(Object entry: subscribers){
  90                     INeutronLoadBalancerPoolMemberAware subscriber = (INeutronLoadBalancerPoolMemberAware) entry;
  91                     subscriber.neutronLoadBalancerPoolMemberUpdated(neutronLBPoolMember);
  92                 }
  93                 }
  94         }
  95     }
  96     private void deletePoolMember(
  97             AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> changes,
  98             Object[] subscribers) {
  99         for (InstanceIdentifier<?> deletedPoolMemberPath : changes.getRemovedPaths()) {
 100                 if(deletedPoolMemberPath.getTargetType().equals(Member.class)){
 101                 NeutronLoadBalancerPoolMember neutronLBPoolMember =
 102                         fromMd(deletedPoolMemberPath, (Member) changes.getOriginalData().get(deletedPoolMemberPath));
 103                 for(Object entry: subscribers){
 104                     INeutronLoadBalancerPoolMemberAware subscriber = (INeutronLoadBalancerPoolMemberAware) entry;
 105                     subscriber.neutronLoadBalancerPoolMemberDeleted(neutronLBPoolMember);
 106                 }
 107                 }
 108         }
 109     }
 110
 111     /*
 112      * This method is borrowed from NeutronLoadBalancerPoolMember.java class of Neutron Northbound class.
 113      * in the original location, this method is called extractFields.
 114      * We will be utilizing similar code from other classes from the same package of neutron project.
 115      */
 116     private NeutronLoadBalancerPoolMember fromMd(InstanceIdentifier<?> iid, Member member) {
 117         NeutronLoadBalancerPoolMember result = new NeutronLoadBalancerPoolMember();
 118
 119         final PoolKey poolsKey = iid.firstKeyOf(Pool.class, PoolKey.class);
 120         if (poolsKey != null) {
 121             result.setPoolID(poolsKey.getUuid().getValue());
 122         }
 123
 124         result.setID(member.getUuid().getValue());
 125         if (member.isAdminStateUp() != null) {
 126             result.setPoolMemberAdminStateIsUp(member.isAdminStateUp());
 127         }
 128
 129         final IpAddress memberIpAddress = member.getAddress();
 130         if (memberIpAddress != null) {
 131             if (memberIpAddress.getIpv4Address() != null) {
 132                 result.setPoolMemberAddress(memberIpAddress.getIpv4Address().getValue());
 133             } else if (memberIpAddress.getIpv6Address() != null) {
 134                 result.setPoolMemberAddress(memberIpAddress.getIpv6Address().getValue());
 135             }
 136         }
 137
 138         if (member.getProtocolPort() != null) {
 139             result.setPoolMemberProtoPort(member.getProtocolPort());
 140         }
 141         if (member.getSubnetId() != null) {
 142             result.setPoolMemberSubnetID(member.getSubnetId().getValue());
 143         }
 144         if (member.getTenantId() != null) {
 145             result.setPoolMemberTenantID(member.getTenantId().getValue());
 146         }
 147         if (member.getWeight() != null) {
 148             result.setPoolMemberWeight(member.getWeight());
 149         }
 150
 151         return result;
 152     }
 153 }
 154