2 * Copyright (c) 2015 Cisco Systems, 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
8 package org.opendaylight.openflowplugin.impl.device;
10 import com.google.common.annotations.VisibleForTesting;
11 import com.google.common.base.Preconditions;
12 import com.google.common.util.concurrent.SettableFuture;
13 import org.opendaylight.controller.md.sal.binding.api.BindingTransactionChain;
14 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
15 import org.opendaylight.controller.md.sal.binding.api.ReadTransaction;
16 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
17 import org.opendaylight.controller.md.sal.common.api.data.AsyncTransaction;
18 import org.opendaylight.controller.md.sal.common.api.data.TransactionChain;
19 import org.opendaylight.controller.md.sal.common.api.data.TransactionChainListener;
20 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
21 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
22 import org.opendaylight.openflowplugin.api.openflow.device.DeviceState;
23 import org.opendaylight.openflowplugin.api.openflow.device.RequestContext;
24 import org.opendaylight.openflowplugin.api.openflow.device.RequestFutureContext;
25 import org.opendaylight.openflowplugin.api.openflow.device.Xid;
26 import org.opendaylight.openflowplugin.api.openflow.device.XidGenerator;
27 import org.opendaylight.openflowplugin.api.openflow.md.core.SwitchConnectionDistinguisher;
28 import org.opendaylight.openflowplugin.openflow.md.core.session.SwitchConnectionCookieOFImpl;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.OfHeader;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.table.types.rev131026.TableFeatures;
31 import org.opendaylight.yangtools.yang.binding.ChildOf;
32 import org.opendaylight.yangtools.yang.binding.DataObject;
33 import org.slf4j.Logger;
34 import org.slf4j.LoggerFactory;
35 import javax.annotation.Nonnull;
36 import java.math.BigInteger;
37 import java.util.HashMap;
43 public class DeviceContextImpl implements DeviceContext, TransactionChainListener {
45 private static final Logger LOG = LoggerFactory.getLogger(DeviceContextImpl.class);
47 private final ConnectionContext primaryConnectionContext;
48 private final DeviceState deviceState;
49 private final DataBroker dataBroker;
50 private final XidGenerator xidGenerator;
52 private final Map<Xid, RequestFutureContext> requests;
53 private final Map<SwitchConnectionDistinguisher, ConnectionContext> auxiliaryConnectionContexts;
54 private BindingTransactionChain txChainFactory;
57 DeviceContextImpl(@Nonnull final ConnectionContext primaryConnectionContext,
58 @Nonnull final DeviceState deviceState, @Nonnull final DataBroker dataBroker) {
59 this.primaryConnectionContext = Preconditions.checkNotNull(primaryConnectionContext);
60 this.deviceState = Preconditions.checkNotNull(deviceState);
61 this.dataBroker = Preconditions.checkNotNull(dataBroker);
62 xidGenerator = new XidGenerator();
63 txChainFactory = dataBroker.createTransactionChain(DeviceContextImpl.this);
64 auxiliaryConnectionContexts = new HashMap<>();
65 requests = new HashMap<>();
69 public <M extends ChildOf<DataObject>> void onMessage(final M message, final RequestContext requestContext) {
70 // TODO Auto-generated method stub
75 public void addAuxiliaryConenctionContext(final ConnectionContext connectionContext) {
76 final SwitchConnectionDistinguisher connectionDistinguisher = new SwitchConnectionCookieOFImpl(connectionContext.getFeatures().getAuxiliaryId());
77 auxiliaryConnectionContexts.put(connectionDistinguisher, connectionContext);
81 public void removeAuxiliaryConenctionContext(final ConnectionContext connectionContext) {
82 // TODO Auto-generated method stub
87 public DeviceState getDeviceState() {
92 public ReadTransaction getReadTransaction() {
93 return dataBroker.newReadOnlyTransaction();
97 public WriteTransaction getWriteTransaction() {
98 // FIXME : we wana to have only one WriteTransaction exposed in one time
99 // so thing about blocking notification mechanism for wait to new transaction
100 return txChainFactory.newWriteOnlyTransaction();
104 public TableFeatures getCapabilities() {
105 // TODO Auto-generated method stub
110 public ConnectionContext getPrimaryConnectionContext() {
111 return primaryConnectionContext;
115 public ConnectionContext getAuxiliaryConnectiobContexts(final BigInteger cookie) {
116 return auxiliaryConnectionContexts.get(new SwitchConnectionCookieOFImpl(cookie.longValue()));
120 public Xid getNextXid() {
121 return xidGenerator.generate();
125 public Map<Xid, RequestFutureContext> getRequests() {
130 public void hookRequestCtx(final Xid xid, final RequestFutureContext requestFutureContext) {
131 requests.put(xid, requestFutureContext);
135 public void processReply(final Xid xid, final OfHeader ofHeader) {
136 final SettableFuture replyFuture = getRequests().get(xid).getFuture();
137 replyFuture.set(ofHeader);
141 public void onTransactionChainFailed(final TransactionChain<?, ?> chain,
142 final AsyncTransaction<?, ?> transaction, final Throwable cause) {
143 txChainFactory.close();
144 txChainFactory = dataBroker.createTransactionChain(DeviceContextImpl.this);
148 public void onTransactionChainSuccessful(final TransactionChain<?, ?> chain) {
149 // NOOP - only yet, here is probably place for notification to get new WriteTransaction