2 * Copyright (c) 2014, 2015 Red Hat, Inc. and others. All rights reserved.
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
9 package org.opendaylight.neutron.transcriber;
11 import com.google.common.base.Optional;
12 import com.google.common.collect.ImmutableBiMap;
13 import com.google.common.util.concurrent.CheckedFuture;
14 import java.util.ArrayList;
15 import java.util.HashSet;
16 import java.util.List;
18 import java.util.concurrent.ExecutionException;
19 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
20 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
21 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
22 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
23 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
24 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
25 import org.opendaylight.neutron.spi.INeutronLoadBalancerPoolCRUD;
26 import org.opendaylight.neutron.spi.NeutronLoadBalancerPool;
27 import org.opendaylight.neutron.spi.NeutronLoadBalancerPoolMember;
28 import org.opendaylight.neutron.spi.NeutronLoadBalancer_SessionPersistence;
29 import org.opendaylight.neutron.spi.Neutron_ID;
30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
31 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev150712.ProtocolBase;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev150712.ProtocolHttp;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev150712.ProtocolHttps;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev150712.ProtocolTcp;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.constants.rev150712.ProtocolTerminatedHttps;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev150712.lbaas.attributes.Pools;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev150712.lbaas.attributes.pools.Pool;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev150712.lbaas.attributes.pools.PoolBuilder;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev150712.lbaas.attributes.pools.pool.Members;
41 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev150712.lbaas.attributes.pools.pool.members.Member;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev150712.lbaas.attributes.pools.pool.members.MemberBuilder;
43 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.lbaasv2.rev150712.pool.attributes.SessionPersistenceBuilder;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.neutron.rev150712.Neutron;
45 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
46 import org.slf4j.Logger;
47 import org.slf4j.LoggerFactory;
49 public final class NeutronLoadBalancerPoolInterface
50 extends AbstractNeutronInterface<Pool, Pools, NeutronLoadBalancerPool>
51 implements INeutronLoadBalancerPoolCRUD {
52 private static final Logger LOGGER = LoggerFactory.getLogger(NeutronLoadBalancerPoolInterface.class);
54 private static final ImmutableBiMap<Class<? extends ProtocolBase>,
55 String> PROTOCOL_MAP = new ImmutableBiMap.Builder<Class<? extends ProtocolBase>, String>()
56 .put(ProtocolHttp.class, "HTTP").put(ProtocolHttps.class, "HTTPS").put(ProtocolTcp.class, "TCP")
57 .put(ProtocolTerminatedHttps.class, "TERMINATED_HTTPS").build();
59 NeutronLoadBalancerPoolInterface(DataBroker db) {
60 super(PoolBuilder.class, db);
64 protected List<Pool> getDataObjectList(Pools pools) {
65 return pools.getPool();
69 protected InstanceIdentifier<Pool> createInstanceIdentifier(Pool pool) {
70 return InstanceIdentifier.create(Neutron.class).child(Pools.class).child(Pool.class, pool.getKey());
74 protected InstanceIdentifier<Pools> createInstanceIdentifier() {
75 return InstanceIdentifier.create(Neutron.class).child(Pools.class);
79 protected Pool toMd(NeutronLoadBalancerPool pool) {
80 final PoolBuilder poolBuilder = new PoolBuilder();
81 poolBuilder.setAdminStateUp(pool.getLoadBalancerPoolAdminIsStateIsUp());
82 if (pool.getNeutronLoadBalancerPoolHealthMonitorID() != null) {
83 poolBuilder.setHealthmonitorId(toUuid(pool.getNeutronLoadBalancerPoolHealthMonitorID()));
85 if (pool.getLoadBalancerPoolLbAlgorithm() != null) {
86 poolBuilder.setLbAlgorithm(pool.getLoadBalancerPoolLbAlgorithm());
88 if (pool.getLoadBalancerPoolListeners() != null) {
89 final List<Uuid> listListener = new ArrayList<Uuid>();
90 for (final Neutron_ID neutron_id : pool.getLoadBalancerPoolListeners()) {
91 listListener.add(toUuid(neutron_id.getID()));
93 poolBuilder.setListeners(listListener);
95 // because members are another container, we don't want to copy
96 // it over, so just skip it here
97 if (pool.getName() != null) {
98 poolBuilder.setName(pool.getName());
100 if (pool.getLoadBalancerPoolProtocol() != null) {
101 final ImmutableBiMap<String, Class<? extends ProtocolBase>> mapper = PROTOCOL_MAP.inverse();
102 poolBuilder.setProtocol((Class<? extends ProtocolBase>) mapper.get(pool.getLoadBalancerPoolProtocol()));
104 if (pool.getLoadBalancerPoolSessionPersistence() != null) {
105 final NeutronLoadBalancer_SessionPersistence sessionPersistence = pool
106 .getLoadBalancerPoolSessionPersistence();
107 final SessionPersistenceBuilder sessionPersistenceBuilder = new SessionPersistenceBuilder();
108 sessionPersistenceBuilder.setCookieName(sessionPersistence.getCookieName());
109 sessionPersistenceBuilder.setType(sessionPersistence.getType());
110 poolBuilder.setSessionPersistence(sessionPersistenceBuilder.build());
112 if (pool.getTenantID() != null) {
113 poolBuilder.setTenantId(toUuid(pool.getTenantID()));
115 if (pool.getID() != null) {
116 poolBuilder.setUuid(toUuid(pool.getID()));
118 LOGGER.warn("Attempting to write neutron load balancer pool without UUID");
120 return poolBuilder.build();
123 protected NeutronLoadBalancerPool fromMd(Pool pool) {
124 final NeutronLoadBalancerPool answer = new NeutronLoadBalancerPool();
125 if (pool.isAdminStateUp() != null) {
126 answer.setLoadBalancerPoolAdminStateIsUp(pool.isAdminStateUp());
128 if (pool.getHealthmonitorId() != null) {
129 answer.setNeutronLoadBalancerPoolHealthMonitorID(pool.getHealthmonitorId().getValue());
131 if (pool.getLbAlgorithm() != null) {
132 answer.setLoadBalancerPoolLbAlgorithm(pool.getLbAlgorithm());
134 if (pool.getListeners() != null) {
135 final List<Neutron_ID> ids = new ArrayList<Neutron_ID>();
136 for (final Uuid id : pool.getListeners()) {
137 ids.add(new Neutron_ID(id.getValue()));
139 answer.setLoadBalancerPoolListeners(ids);
141 if (pool.getMembers() != null) {
142 final List<NeutronLoadBalancerPoolMember> members = new ArrayList<NeutronLoadBalancerPoolMember>();
143 for (final Member member : pool.getMembers().getMember()) {
144 members.add(fromMemberMd(member));
146 answer.setLoadBalancerPoolMembers(members);
148 if (pool.getName() != null) {
149 answer.setName(pool.getName());
151 if (pool.getProtocol() != null) {
152 answer.setLoadBalancerPoolProtocol(PROTOCOL_MAP.get(pool.getProtocol()));
154 if (pool.getSessionPersistence() != null) {
155 final NeutronLoadBalancer_SessionPersistence sessionPersistence =
156 new NeutronLoadBalancer_SessionPersistence();
157 sessionPersistence.setCookieName(pool.getSessionPersistence().getCookieName());
158 sessionPersistence.setType(pool.getSessionPersistence().getType());
160 answer.setLoadBalancerSessionPersistence(sessionPersistence);
162 if (pool.getTenantId() != null) {
163 answer.setTenantID(pool.getTenantId());
165 if (pool.getUuid() != null) {
166 answer.setID(pool.getUuid().getValue());
171 public boolean neutronLoadBalancerPoolMemberExists(String poolUuid, String uuid) {
172 final Member member = readMemberMd(createMemberInstanceIdentifier(toMd(poolUuid), toMemberMd(uuid)));
173 if (member == null) {
179 public NeutronLoadBalancerPoolMember getNeutronLoadBalancerPoolMember(String poolUuid, String uuid) {
180 final Member member = readMemberMd(createMemberInstanceIdentifier(toMd(poolUuid), toMemberMd(uuid)));
181 if (member == null) {
184 return fromMemberMd(member);
187 public List<NeutronLoadBalancerPoolMember> getAllNeutronLoadBalancerPoolMembers(String poolUuid) {
188 final Set<NeutronLoadBalancerPoolMember> allLoadBalancerPoolMembers = new HashSet<
189 NeutronLoadBalancerPoolMember>();
190 final Members members = readMd(createMembersInstanceIdentifier(toMd(poolUuid)));
191 if (members != null) {
192 for (final Member member : members.getMember()) {
193 allLoadBalancerPoolMembers.add(fromMemberMd(member));
196 LOGGER.debug("Exiting getLoadBalancerPoolMembers, Found {} OpenStackLoadBalancerPoolMember",
197 allLoadBalancerPoolMembers.size());
198 final List<NeutronLoadBalancerPoolMember> ans = new ArrayList<NeutronLoadBalancerPoolMember>();
199 ans.addAll(allLoadBalancerPoolMembers);
203 public boolean addNeutronLoadBalancerPoolMember(String poolUuid, NeutronLoadBalancerPoolMember input) {
204 if (neutronLoadBalancerPoolMemberExists(poolUuid, input.getID())) {
207 addMemberMd(toMd(poolUuid), input);
211 public boolean removeNeutronLoadBalancerPoolMember(String poolUuid, String uuid) {
212 if (!neutronLoadBalancerPoolMemberExists(poolUuid, uuid)) {
215 return removeMemberMd(toMd(poolUuid), toMemberMd(uuid));
218 public boolean updateNeutronLoadBalancerPoolMember(String poolUuid, String uuid,
219 NeutronLoadBalancerPoolMember delta) {
220 if (!neutronLoadBalancerPoolMemberExists(poolUuid, uuid)) {
223 updateMemberMd(toMd(poolUuid), delta);
227 public boolean neutronLoadBalancerPoolMemberInUse(String poolUuid, String loadBalancerPoolMemberID) {
228 return !neutronLoadBalancerPoolMemberExists(poolUuid, loadBalancerPoolMemberID);
231 protected InstanceIdentifier<Member> createMemberInstanceIdentifier(Pool pool, Member item) {
232 return InstanceIdentifier.create(Neutron.class).child(Pools.class).child(Pool.class, pool.getKey())
233 .child(Members.class).child(Member.class, item.getKey());
236 protected InstanceIdentifier<Members> createMembersInstanceIdentifier(Pool pool) {
237 return InstanceIdentifier.create(Neutron.class).child(Pools.class).child(Pool.class, pool.getKey())
238 .child(Members.class);
241 static NeutronLoadBalancerPoolMember fromMemberMd(Member member) {
242 final NeutronLoadBalancerPoolMember answer = new NeutronLoadBalancerPoolMember();
243 if (member.isAdminStateUp() != null) {
244 answer.setPoolMemberAdminStateIsUp(member.isAdminStateUp());
246 if (member.getAddress() != null) {
247 answer.setPoolMemberAddress(String.valueOf(member.getAddress().getValue()));
249 if (member.getProtocolPort() != null) {
250 answer.setPoolMemberProtoPort(member.getProtocolPort());
252 if (member.getUuid() != null) {
253 answer.setID(member.getUuid().getValue());
254 answer.setPoolID(member.getUuid().getValue());
256 if (member.getSubnetId() != null) {
257 answer.setPoolMemberSubnetID(member.getSubnetId().getValue());
259 if (member.getTenantId() != null) {
260 answer.setTenantID(member.getTenantId());
262 if (member.getWeight() != null) {
263 answer.setPoolMemberWeight(member.getWeight());
268 protected Member toMemberMd(NeutronLoadBalancerPoolMember member) {
269 final MemberBuilder memberBuilder = new MemberBuilder();
270 memberBuilder.setAdminStateUp(member.getPoolMemberAdminStateIsUp());
271 if (member.getPoolMemberAddress() != null) {
272 final IpAddress ipAddress = new IpAddress(member.getPoolMemberAddress().toCharArray());
273 memberBuilder.setAddress(ipAddress);
275 if (member.getPoolMemberProtoPort() != null) {
276 memberBuilder.setProtocolPort(member.getPoolMemberProtoPort());
278 if (member.getID() != null) {
279 memberBuilder.setUuid(toUuid(member.getID()));
281 if (member.getPoolMemberSubnetID() != null) {
282 memberBuilder.setSubnetId(toUuid(member.getPoolMemberSubnetID()));
284 if (member.getTenantID() != null) {
285 memberBuilder.setTenantId(toUuid(member.getTenantID()));
287 if (member.getPoolMemberWeight() != null) {
288 memberBuilder.setWeight(member.getPoolMemberWeight());
290 return memberBuilder.build();
293 protected Member toMemberMd(String uuid) {
294 final MemberBuilder memberBuilder = new MemberBuilder();
295 memberBuilder.setUuid(toUuid(uuid));
296 return memberBuilder.build();
300 T extends org.opendaylight.yangtools.yang.binding.DataObject> T readMemberMd(InstanceIdentifier<T> path) {
302 final ReadOnlyTransaction transaction = getDataBroker().newReadOnlyTransaction();
303 final CheckedFuture<Optional<T>,
304 ReadFailedException> future = transaction.read(LogicalDatastoreType.CONFIGURATION, path);
305 if (future != null) {
306 Optional<T> optional;
308 optional = future.checkedGet();
309 if (optional.isPresent()) {
310 result = optional.get();
312 } catch (final ReadFailedException e) {
313 LOGGER.warn("Failed to read {}", path, e);
320 protected boolean addMemberMd(Pool pool, NeutronLoadBalancerPoolMember neutronObject) {
321 // TODO think about adding existence logic
322 return updateMemberMd(pool, neutronObject);
325 protected boolean updateMemberMd(Pool pool, NeutronLoadBalancerPoolMember neutronObject) {
326 final WriteTransaction transaction = getDataBroker().newWriteOnlyTransaction();
327 final Member item = toMemberMd(neutronObject);
328 final InstanceIdentifier<Member> iid = createMemberInstanceIdentifier(pool, item);
329 transaction.put(LogicalDatastoreType.CONFIGURATION, iid, item, true);
330 final CheckedFuture<Void, TransactionCommitFailedException> future = transaction.submit();
333 } catch (InterruptedException | ExecutionException e) {
334 LOGGER.warn("Transation failed ", e);
340 protected boolean removeMemberMd(Pool pool, Member item) {
341 final WriteTransaction transaction = getDataBroker().newWriteOnlyTransaction();
342 final InstanceIdentifier<Member> iid = createMemberInstanceIdentifier(pool, item);
343 transaction.delete(LogicalDatastoreType.CONFIGURATION, iid);
344 final CheckedFuture<Void, TransactionCommitFailedException> future = transaction.submit();
347 } catch (InterruptedException | ExecutionException e) {
348 LOGGER.warn("Transation failed ", e);