BUG 5746 - Ovsdb QoS and Queue model enhancements
[ovsdb.git] / southbound / southbound-impl / src / main / java / org / opendaylight / ovsdb / southbound / ovsdb / transact / QueueUpdateCommand.java
1 /*
2  * Copyright (c) 2016 Intel Corporation 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.ovsdb.southbound.ovsdb.transact;
9
10 import static org.opendaylight.ovsdb.lib.operations.Operations.op;
11
12 import java.util.Collection;
13 import java.util.HashMap;
14 import java.util.HashSet;
15 import java.util.List;
16 import java.util.Map;
17 import java.util.Map.Entry;
18 import java.util.Set;
19
20 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
21 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
22 import org.opendaylight.ovsdb.lib.notation.UUID;
23 import org.opendaylight.ovsdb.lib.operations.TransactionBuilder;
24 import org.opendaylight.ovsdb.lib.schema.typed.TyperUtils;
25 import org.opendaylight.ovsdb.schema.openvswitch.Queue;
26 import org.opendaylight.ovsdb.southbound.SouthboundConstants;
27 import org.opendaylight.ovsdb.southbound.SouthboundMapper;
28 import org.opendaylight.ovsdb.southbound.SouthboundUtil;
29 import org.opendaylight.ovsdb.utils.yang.YangUtils;
30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Uri;
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.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentation;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.Queues;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.QueuesKey;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.queues.QueuesExternalIds;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.queues.QueuesOtherConfig;
37 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
38 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
39 import org.opendaylight.yangtools.yang.binding.DataObject;
40 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
41 import org.slf4j.Logger;
42 import org.slf4j.LoggerFactory;
43
44 public class QueueUpdateCommand implements TransactCommand {
45     private static final Logger LOG = LoggerFactory.getLogger(QueueUpdateCommand.class);
46
47     @Override
48     public void execute(TransactionBuilder transaction, BridgeOperationalState state,
49                         AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> events) {
50         execute(transaction, state, TransactUtils.extractCreatedOrUpdated(events, OvsdbNodeAugmentation.class));
51     }
52
53     @Override
54     public void execute(TransactionBuilder transaction, BridgeOperationalState state,
55                         Collection<DataTreeModification<Node>> modifications) {
56         execute(transaction, state, TransactUtils.extractCreatedOrUpdated(modifications, OvsdbNodeAugmentation.class));
57     }
58
59     private void execute(TransactionBuilder transaction, BridgeOperationalState state,
60                          Map<InstanceIdentifier<OvsdbNodeAugmentation>, OvsdbNodeAugmentation> createdOrUpdated) {
61         for (Entry<InstanceIdentifier<OvsdbNodeAugmentation>, OvsdbNodeAugmentation> ovsdbNodeEntry:
62             createdOrUpdated.entrySet()) {
63             updateQueue(transaction, state, ovsdbNodeEntry.getKey(), ovsdbNodeEntry.getValue());
64         }
65     }
66
67     private void updateQueue(
68             TransactionBuilder transaction, BridgeOperationalState state,
69             InstanceIdentifier<OvsdbNodeAugmentation> iid, OvsdbNodeAugmentation ovsdbNode) {
70
71         List<Queues> queueList = ovsdbNode.getQueues();
72
73         if (!state.getBridgeNode(iid).isPresent()) {
74             return;
75         }
76         OvsdbNodeAugmentation operNode = state.getBridgeNode(iid).get().getAugmentation(OvsdbNodeAugmentation.class);
77         List<Queues> operQueues = operNode.getQueues();
78
79         if (queueList != null) {
80             for (Queues queueEntry : queueList) {
81                 Queue queue = TyperUtils.getTypedRowWrapper(transaction.getDatabaseSchema(), Queue.class);
82
83                 if (queueEntry.getDscp() != null) {
84                     try {
85                         Set<Long> dscpSet = new HashSet<>();
86                             if (dscpSet.add(new Long(queueEntry.getDscp().toString()))) {
87                             queue.setDscp(dscpSet);
88                         }
89                     } catch (NumberFormatException e) {
90                         LOG.warn("Invalid DSCP {} setting for Queue {}", queueEntry.getDscp(), queueEntry, e);
91                     }
92                 }
93
94                 Map<String, String> externalIdsMap = new HashMap<>();
95                 try {
96                     YangUtils.copyYangKeyValueListToMap(externalIdsMap, queueEntry.getQueuesExternalIds(),
97                             QueuesExternalIds::getQueuesExternalIdKey, QueuesExternalIds::getQueuesExternalIdValue);
98                 } catch (NullPointerException e) {
99                     LOG.warn("Incomplete Queue external IDs", e);
100                 }
101                 externalIdsMap.put(SouthboundConstants.IID_EXTERNAL_ID_KEY,
102                         SouthboundUtil.serializeInstanceIdentifier(
103                         SouthboundMapper.createInstanceIdentifier(iid.firstKeyOf(Node.class, NodeKey.class).getNodeId())
104                         .augmentation(OvsdbNodeAugmentation.class)
105                         .child(Queues.class, new QueuesKey(queueEntry.getQueueId()))));
106                 queue.setExternalIds(externalIdsMap);
107
108                 try {
109                     queue.setOtherConfig(YangUtils.convertYangKeyValueListToMap(queueEntry.getQueuesOtherConfig(),
110                             QueuesOtherConfig::getQueueOtherConfigKey, QueuesOtherConfig::getQueueOtherConfigValue));
111                 } catch (NullPointerException e) {
112                     LOG.warn("Incomplete Queue other_config", e);
113                 }
114
115                 Uuid operQueueUuid = getQueueEntryUuid(operQueues, queueEntry.getQueueId());
116                 if (operQueueUuid == null) {
117                     UUID namedUuid = new UUID(SouthboundConstants.QUEUE_NAMED_UUID_PREFIX +
118                             TransactUtils.bytesToHexString(queueEntry.getQueueId().getValue().getBytes()));
119                     transaction.add(op.insert(queue).withId(namedUuid.toString())).build();
120                 } else {
121                     UUID uuid = new UUID(operQueueUuid.getValue());
122                     //transaction.add(op.update(queue)).build();
123                     Queue extraQueue = TyperUtils.getTypedRowWrapper(
124                             transaction.getDatabaseSchema(), Queue.class, null);
125                     extraQueue.getUuidColumn().setData(uuid);
126                     transaction.add(op.update(queue.getSchema())
127                             .where(extraQueue.getUuidColumn().getSchema().opEqual(uuid)).build());
128                 }
129                 transaction.build();
130             }
131         }
132     }
133
134     private Uuid getQueueEntryUuid(List<Queues> operQueues, Uri queueId) {
135         if (operQueues != null && !operQueues.isEmpty()) {
136             for (Queues queueEntry : operQueues) {
137                 if (queueEntry.getQueueId().equals(queueId)) {
138                     return queueEntry.getQueueUuid();
139                 }
140             }
141         }
142         return null;
143     }
144 }