Make TransactionBuilder type-aware
[ovsdb.git] / hwvtepsouthbound / hwvtepsouthbound-impl / src / main / java / org / opendaylight / ovsdb / hwvtepsouthbound / transact / LogicalRouterUpdateCommand.java
1 /*
2  * Copyright © 2017 Hewlett Packard Enterprise, Co. 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.ovsdb.hwvtepsouthbound.transact;
10
11 import static org.opendaylight.ovsdb.lib.operations.Operations.op;
12
13 import com.google.common.base.Optional;
14 import java.util.Collection;
15 import java.util.HashMap;
16 import java.util.List;
17 import java.util.Map;
18 import java.util.Map.Entry;
19 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
20 import org.opendaylight.ovsdb.lib.notation.UUID;
21 import org.opendaylight.ovsdb.lib.operations.TransactionBuilder;
22 import org.opendaylight.ovsdb.schema.hardwarevtep.LogicalRouter;
23 import org.opendaylight.ovsdb.utils.mdsal.utils.TransactionType;
24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentation;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.Acls;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalRouters;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitches;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.logical.router.attributes.AclBindings;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.logical.router.attributes.StaticRoutes;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.logical.router.attributes.SwitchBindings;
32 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
33 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
34 import org.slf4j.Logger;
35 import org.slf4j.LoggerFactory;
36
37
38 public class LogicalRouterUpdateCommand extends AbstractTransactCommand<LogicalRouters, HwvtepGlobalAugmentation> {
39     private static final Logger LOG = LoggerFactory.getLogger(LogicalRouterUpdateCommand.class);
40
41     public LogicalRouterUpdateCommand(final HwvtepOperationalState state,
42             final Collection<DataTreeModification<Node>> changes) {
43         super(state, changes);
44     }
45
46     @Override
47     public void execute(final TransactionBuilder transaction) {
48         Map<InstanceIdentifier<Node>, List<LogicalRouters>> updateMap =
49                 extractUpdated(getChanges(),LogicalRouters.class);
50
51         for (Entry<InstanceIdentifier<Node>, List<LogicalRouters>> updated:
52             updateMap.entrySet()) {
53             updateLogicalRouter(transaction,  updated.getKey(), updated.getValue());
54         }
55     }
56
57     private void updateLogicalRouter(final TransactionBuilder transaction,
58             final InstanceIdentifier<Node> instanceIdentifier, final List<LogicalRouters> routerList) {
59         for (LogicalRouters lrouter: routerList) {
60             InstanceIdentifier<LogicalRouters> routerKey = instanceIdentifier
61                     .augmentation(HwvtepGlobalAugmentation.class).child(LogicalRouters.class, lrouter.key());
62             LOG.debug("Creating logical router named: {}", lrouter.getHwvtepNodeName());
63
64             final Optional<LogicalRouters> operationalRouterOptional =
65                     getOperationalState().getLogicalRouters(instanceIdentifier, lrouter.key());
66             LogicalRouter logicalRouter = transaction.getTypedRowWrapper(LogicalRouter.class);
67             setDescription(logicalRouter, lrouter);
68
69             setSwitchBindings(transaction, logicalRouter, lrouter.getSwitchBindings());
70             setStaticRoutes(logicalRouter, lrouter.getStaticRoutes());
71             setAclBindings(logicalRouter, lrouter.getAclBindings());
72
73             if (!operationalRouterOptional.isPresent()) {
74                 setName(logicalRouter, lrouter, operationalRouterOptional);
75                 LOG.trace("Creating LogicalRouter entry: {}", logicalRouter);
76                 transaction.add(op.insert(logicalRouter).withId(TransactUtils.getLogicalRouterId(lrouter)));
77                 transaction.add(op.comment("Logical Router: Creating " + lrouter.getHwvtepNodeName().getValue()));
78                 UUID lrUuid = new UUID(TransactUtils.getLogicalRouterId(lrouter));
79                 updateCurrentTxData(LogicalRouters.class, routerKey, lrUuid, lrouter);
80                 updateControllerTxHistory(TransactionType.ADD, logicalRouter);
81             } else {
82                 LogicalRouters updatedLRouter = operationalRouterOptional.get();
83                 String existingLogicalRouterName = updatedLRouter.getHwvtepNodeName().getValue();
84                 LogicalRouter extraLogicalRouter = transaction.getTypedRowWrapper(LogicalRouter.class);
85                 extraLogicalRouter.setName("");
86                 LOG.trace("Updating LogicalRouter entry: {}", logicalRouter);
87                 transaction.add(op.update(logicalRouter)
88                         .where(extraLogicalRouter.getNameColumn().getSchema().opEqual(existingLogicalRouterName))
89                         .build());
90                 transaction.add(op.comment("Logical Router: Updating " + existingLogicalRouterName));
91                 updateControllerTxHistory(TransactionType.UPDATE, logicalRouter);
92             }
93         }
94
95     }
96
97     private void setDescription(final LogicalRouter logicalRouter, final LogicalRouters inputRouter) {
98         if (inputRouter.getHwvtepNodeDescription() != null) {
99             logicalRouter.setDescription(inputRouter.getHwvtepNodeDescription());
100         } else {
101             LOG.warn("Logical router {} is missing a description string", inputRouter.getHwvtepNodeName());
102         }
103     }
104
105     private void setName(final LogicalRouter logicalRouter, final LogicalRouters inputRouter,
106             final Optional<LogicalRouters> inputRouterOptional) {
107         if (inputRouter.getHwvtepNodeName() != null) {
108             logicalRouter.setName(inputRouter.getHwvtepNodeName().getValue());
109         } else if (inputRouterOptional.isPresent() && inputRouterOptional.get().getHwvtepNodeName() != null) {
110             logicalRouter.setName(inputRouterOptional.get().getHwvtepNodeName().getValue());
111         }
112     }
113
114     private void setSwitchBindings(final TransactionBuilder transaction, final LogicalRouter logicalRouter,
115             final List<SwitchBindings> switchBindings) {
116         if (switchBindings != null) {
117             Map<String, UUID> bindingMap = new HashMap<>();
118             for (SwitchBindings switchBinding : switchBindings) {
119                 @SuppressWarnings("unchecked")
120                 InstanceIdentifier<LogicalSwitches> lswitchIid =
121                         (InstanceIdentifier<LogicalSwitches>)switchBinding.getLogicalSwitchRef().getValue();
122                 Optional<LogicalSwitches> operationalSwitchOptional =
123                         getOperationalState().getLogicalSwitches(lswitchIid);
124                 if (operationalSwitchOptional.isPresent()) {
125                     Uuid logicalSwitchUuid = operationalSwitchOptional.get().getLogicalSwitchUuid();
126                     bindingMap.put(switchBinding.getDestinationAddress().getIpv4Prefix().getValue(),
127                             new UUID(logicalSwitchUuid.getValue()));
128                 } else {
129                     bindingMap.put(switchBinding.getDestinationAddress().getIpv4Prefix().getValue(),
130                             TransactUtils.getLogicalSwitchUUID(transaction, getOperationalState(), lswitchIid));
131                 }
132             }
133             logicalRouter.setSwitchBinding(bindingMap);
134         }
135     }
136
137     private void setAclBindings(final LogicalRouter logicalRouter, final List<AclBindings> aclBindings) {
138         if (aclBindings != null) {
139             Map<String, UUID> bindingMap = new HashMap<>();
140             for (AclBindings aclBinding : aclBindings) {
141                 @SuppressWarnings("unchecked")
142                 InstanceIdentifier<Acls> aclIid =
143                         (InstanceIdentifier<Acls>)aclBinding.getAclRef().getValue();
144                 Optional<Acls> operationalAclOptional =
145                         getOperationalState().getAcls(aclIid);
146                 if (operationalAclOptional.isPresent()) {
147                     Uuid aclUuid = operationalAclOptional.get().getAclUuid();
148                     bindingMap.put(aclBinding.getRouterInterface().stringValue(),
149                             new UUID(aclUuid.getValue()));
150                 } else {
151                     bindingMap.put(aclBinding.getRouterInterface().stringValue(),
152                             TransactUtils.getAclUUID(aclIid));
153                 }
154             }
155             logicalRouter.setAclBinding(bindingMap);
156         }
157
158     }
159
160     private void setStaticRoutes(final LogicalRouter logicalRouter, final List<StaticRoutes> staticRoutes) {
161         if (staticRoutes != null) {
162             Map<String, String> staticRoutesMap = new HashMap<>();
163             for (StaticRoutes staticRoute : staticRoutes) {
164                 staticRoutesMap.put(staticRoute.getDestinationAddress().stringValue(),
165                         staticRoute.getNexthopAddress().stringValue());
166             }
167             logicalRouter.setStaticRoutes(staticRoutesMap);
168         }
169     }
170
171     @Override
172     protected List<LogicalRouters> getData(final HwvtepGlobalAugmentation augmentation) {
173         return augmentation.getLogicalRouters();
174     }
175 }
176