Merge "BUG-4283 experimenter msg support - registration part"
[openflowplugin.git] / openflowplugin-impl / src / main / java / org / opendaylight / openflowplugin / impl / device / DeviceContextImpl.java
1 /**
2  * Copyright (c) 2015 Cisco Systems, 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 package org.opendaylight.openflowplugin.impl.device;
9
10 import com.google.common.annotations.VisibleForTesting;
11 import com.google.common.base.Preconditions;
12 import com.google.common.util.concurrent.FutureCallback;
13 import com.google.common.util.concurrent.Futures;
14 import com.google.common.util.concurrent.ListenableFuture;
15 import io.netty.util.HashedWheelTimer;
16 import io.netty.util.Timeout;
17 import java.math.BigInteger;
18 import java.util.Collection;
19 import java.util.HashMap;
20 import java.util.HashSet;
21 import java.util.List;
22 import java.util.Map;
23 import java.util.concurrent.ConcurrentHashMap;
24 import javax.annotation.Nonnull;
25 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
26 import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
27 import org.opendaylight.controller.md.sal.binding.api.NotificationService;
28 import org.opendaylight.controller.md.sal.binding.api.ReadTransaction;
29 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
30 import org.opendaylight.openflowjava.protocol.api.connection.ConnectionAdapter;
31 import org.opendaylight.openflowjava.protocol.api.connection.OutboundQueue;
32 import org.opendaylight.openflowplugin.api.openflow.connection.ConnectionContext;
33 import org.opendaylight.openflowplugin.api.openflow.connection.OutboundQueueProvider;
34 import org.opendaylight.openflowplugin.api.openflow.device.DeviceContext;
35 import org.opendaylight.openflowplugin.api.openflow.device.DeviceState;
36 import org.opendaylight.openflowplugin.api.openflow.device.MessageTranslator;
37 import org.opendaylight.openflowplugin.api.openflow.device.RequestContext;
38 import org.opendaylight.openflowplugin.api.openflow.device.TranslatorLibrary;
39 import org.opendaylight.openflowplugin.api.openflow.device.Xid;
40 import org.opendaylight.openflowplugin.api.openflow.device.handlers.DeviceContextClosedHandler;
41 import org.opendaylight.openflowplugin.api.openflow.device.handlers.MultiMsgCollector;
42 import org.opendaylight.openflowplugin.api.openflow.md.core.SwitchConnectionDistinguisher;
43 import org.opendaylight.openflowplugin.api.openflow.md.core.TranslatorKey;
44 import org.opendaylight.openflowplugin.api.openflow.registry.ItemLifeCycleRegistry;
45 import org.opendaylight.openflowplugin.api.openflow.registry.flow.DeviceFlowRegistry;
46 import org.opendaylight.openflowplugin.api.openflow.registry.flow.FlowDescriptor;
47 import org.opendaylight.openflowplugin.api.openflow.registry.flow.FlowRegistryKey;
48 import org.opendaylight.openflowplugin.api.openflow.registry.group.DeviceGroupRegistry;
49 import org.opendaylight.openflowplugin.api.openflow.registry.meter.DeviceMeterRegistry;
50 import org.opendaylight.openflowplugin.api.openflow.rpc.ItemLifeCycleKeeper;
51 import org.opendaylight.openflowplugin.api.openflow.rpc.RpcContext;
52 import org.opendaylight.openflowplugin.api.openflow.rpc.listener.ItemLifecycleListener;
53 import org.opendaylight.openflowplugin.api.openflow.statistics.ofpspecific.MessageSpy;
54 import org.opendaylight.openflowplugin.impl.common.ItemLifeCycleSourceImpl;
55 import org.opendaylight.openflowplugin.impl.common.NodeStaticReplyTranslatorUtil;
56 import org.opendaylight.openflowplugin.impl.device.listener.MultiMsgCollectorImpl;
57 import org.opendaylight.openflowplugin.impl.registry.flow.DeviceFlowRegistryImpl;
58 import org.opendaylight.openflowplugin.impl.registry.flow.FlowRegistryKeyFactory;
59 import org.opendaylight.openflowplugin.impl.registry.group.DeviceGroupRegistryImpl;
60 import org.opendaylight.openflowplugin.impl.registry.meter.DeviceMeterRegistryImpl;
61 import org.opendaylight.openflowplugin.openflow.md.core.session.SwitchConnectionCookieOFImpl;
62 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
63 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeConnector;
64 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.Table;
65 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.Flow;
66 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.FlowKey;
67 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorId;
68 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorRef;
69 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnector;
70 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorBuilder;
71 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey;
72 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
73 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.common.types.rev130731.PortReason;
74 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.Error;
75 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.FlowRemoved;
76 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.MultipartReply;
77 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.OfHeader;
78 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PacketIn;
79 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PacketInMessage;
80 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PortGrouping;
81 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.PortStatusMessage;
82 import org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service.rev130709.PacketReceived;
83 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.FlowCapableNodeConnectorStatisticsData;
84 import org.opendaylight.yang.gen.v1.urn.opendaylight.port.statistics.rev131214.FlowCapableNodeConnectorStatisticsDataBuilder;
85 import org.opendaylight.yangtools.yang.binding.ChildOf;
86 import org.opendaylight.yangtools.yang.binding.DataObject;
87 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
88 import org.opendaylight.yangtools.yang.binding.KeyedInstanceIdentifier;
89 import org.slf4j.Logger;
90 import org.slf4j.LoggerFactory;
91
92 /**
93  *
94  */
95 public class DeviceContextImpl implements DeviceContext {
96
97     private static final Logger LOG = LoggerFactory.getLogger(DeviceContextImpl.class);
98
99     // TODO: drain factor should be parametrized
100     public static final float REJECTED_DRAIN_FACTOR = 0.25f;
101     // TODO: low water mark factor should be parametrized
102     private static final float LOW_WATERMARK_FACTOR = 0.75f;
103     // TODO: high water mark factor should be parametrized
104     private static final float HIGH_WATERMARK_FACTOR = 0.95f;
105
106     private final ConnectionContext primaryConnectionContext;
107     private final DeviceState deviceState;
108     private final DataBroker dataBroker;
109     private final HashedWheelTimer hashedWheelTimer;
110     private final Map<SwitchConnectionDistinguisher, ConnectionContext> auxiliaryConnectionContexts;
111     private final TransactionChainManager transactionChainManager;
112     private final DeviceFlowRegistry deviceFlowRegistry;
113     private final DeviceGroupRegistry deviceGroupRegistry;
114     private final DeviceMeterRegistry deviceMeterRegistry;
115     private final Collection<DeviceContextClosedHandler> closeHandlers = new HashSet<>();
116     private final PacketInRateLimiter packetInLimiter;
117     private final MessageSpy messageSpy;
118     private final ItemLifeCycleKeeper flowLifeCycleKeeper;
119     private NotificationPublishService notificationPublishService;
120     private NotificationService notificationService;
121     private final OutboundQueue outboundQueueProvider;
122     private Timeout barrierTaskTimeout;
123     private final MessageTranslator<PortGrouping, FlowCapableNodeConnector> portStatusTranslator;
124     private final MessageTranslator<PacketInMessage, PacketReceived> packetInTranslator;
125     private final MessageTranslator<FlowRemoved, org.opendaylight.yang.gen.v1.urn.opendaylight.flow.service.rev130819.FlowRemoved> flowRemovedTranslator;
126     private final TranslatorLibrary translatorLibrary;
127     private Map<Long, NodeConnectorRef> nodeConnectorCache;
128     private ItemLifeCycleRegistry itemLifeCycleSourceRegistry;
129     private RpcContext rpcContext;
130
131     @VisibleForTesting
132     DeviceContextImpl(@Nonnull final ConnectionContext primaryConnectionContext,
133                       @Nonnull final DeviceState deviceState,
134                       @Nonnull final DataBroker dataBroker,
135                       @Nonnull final HashedWheelTimer hashedWheelTimer,
136                       @Nonnull final MessageSpy _messageSpy,
137                       @Nonnull final OutboundQueueProvider outboundQueueProvider,
138                       @Nonnull final TranslatorLibrary translatorLibrary,
139                       @Nonnull final TransactionChainManager transactionChainManager) {
140         this.primaryConnectionContext = Preconditions.checkNotNull(primaryConnectionContext);
141         this.deviceState = Preconditions.checkNotNull(deviceState);
142         this.dataBroker = Preconditions.checkNotNull(dataBroker);
143         this.hashedWheelTimer = Preconditions.checkNotNull(hashedWheelTimer);
144         this.outboundQueueProvider = Preconditions.checkNotNull(outboundQueueProvider);
145         this.transactionChainManager = Preconditions.checkNotNull(transactionChainManager);
146         auxiliaryConnectionContexts = new HashMap<>();
147         deviceFlowRegistry = new DeviceFlowRegistryImpl();
148         deviceGroupRegistry = new DeviceGroupRegistryImpl();
149         deviceMeterRegistry = new DeviceMeterRegistryImpl();
150         messageSpy = _messageSpy;
151
152         packetInLimiter = new PacketInRateLimiter(primaryConnectionContext.getConnectionAdapter(),
153                 /*initial*/ 1000, /*initial*/2000, messageSpy, REJECTED_DRAIN_FACTOR);
154
155         this.translatorLibrary = translatorLibrary;
156         portStatusTranslator = translatorLibrary.lookupTranslator(
157                 new TranslatorKey(deviceState.getVersion(), PortGrouping.class.getName()));
158         packetInTranslator = translatorLibrary.lookupTranslator(
159                 new TranslatorKey(deviceState.getVersion(), PacketIn.class.getName()));
160         flowRemovedTranslator = translatorLibrary.lookupTranslator(
161                 new TranslatorKey(deviceState.getVersion(), FlowRemoved.class.getName()));
162
163
164         nodeConnectorCache = new ConcurrentHashMap<>();
165
166         itemLifeCycleSourceRegistry = new ItemLifeCycleRegistryImpl();
167         flowLifeCycleKeeper = new ItemLifeCycleSourceImpl();
168         itemLifeCycleSourceRegistry.registerLifeCycleSource(flowLifeCycleKeeper);
169     }
170
171     /**
172      * This method is called from {@link DeviceManagerImpl} only. So we could say "posthandshake process finish"
173      * and we are able to set a scheduler for an automatic transaction submitting by time (0,5sec).
174      */
175     void initialSubmitTransaction() {
176         transactionChainManager.initialSubmitWriteTransaction();
177     }
178
179     /**
180      * This method is called fron
181      */
182     void cancelTransaction() {
183         transactionChainManager.cancelWriteTransaction();
184     }
185
186     @Override
187     public Long getReservedXid() {
188         return outboundQueueProvider.reserveEntry();
189     }
190
191     @Override
192     public <M extends ChildOf<DataObject>> void onMessage(final M message, final RequestContext<?> requestContext) {
193         // TODO Auto-generated method stub
194     }
195
196     @Override
197     public void addAuxiliaryConenctionContext(final ConnectionContext connectionContext) {
198         final SwitchConnectionDistinguisher connectionDistinguisher = createConnectionDistinguisher(connectionContext);
199         auxiliaryConnectionContexts.put(connectionDistinguisher, connectionContext);
200     }
201
202     private static SwitchConnectionDistinguisher createConnectionDistinguisher(final ConnectionContext connectionContext) {
203         return new SwitchConnectionCookieOFImpl(connectionContext.getFeatures().getAuxiliaryId());
204     }
205
206     @Override
207     public void removeAuxiliaryConenctionContext(final ConnectionContext connectionContext) {
208         // TODO Auto-generated method stub
209     }
210
211     @Override
212     public DeviceState getDeviceState() {
213         return deviceState;
214     }
215
216     @Override
217     public ReadTransaction getReadTransaction() {
218         return dataBroker.newReadOnlyTransaction();
219     }
220
221     @Override
222     public <T extends DataObject> void writeToTransaction(final LogicalDatastoreType store,
223                                                           final InstanceIdentifier<T> path, final T data) {
224         transactionChainManager.writeToTransaction(store, path, data);
225     }
226
227     @Override
228     public <T extends DataObject> void addDeleteToTxChain(final LogicalDatastoreType store, final InstanceIdentifier<T> path) {
229         transactionChainManager.addDeleteOperationTotTxChain(store, path);
230     }
231
232     @Override
233     public boolean submitTransaction() {
234         return transactionChainManager.submitWriteTransaction();
235     }
236
237     @Override
238     public ConnectionContext getPrimaryConnectionContext() {
239         return primaryConnectionContext;
240     }
241
242     @Override
243     public ConnectionContext getAuxiliaryConnectiobContexts(final BigInteger cookie) {
244         return auxiliaryConnectionContexts.get(new SwitchConnectionCookieOFImpl(cookie.longValue()));
245     }
246
247     @Override
248     public DeviceFlowRegistry getDeviceFlowRegistry() {
249         return deviceFlowRegistry;
250     }
251
252     @Override
253     public DeviceGroupRegistry getDeviceGroupRegistry() {
254         return deviceGroupRegistry;
255     }
256
257     @Override
258     public DeviceMeterRegistry getDeviceMeterRegistry() {
259         return deviceMeterRegistry;
260     }
261
262     @Override
263     public void processReply(final OfHeader ofHeader) {
264         if (ofHeader instanceof Error) {
265             messageSpy.spyMessage(ofHeader.getImplementedInterface(), MessageSpy.STATISTIC_GROUP.FROM_SWITCH_PUBLISHED_FAILURE);
266         } else {
267             messageSpy.spyMessage(ofHeader.getImplementedInterface(), MessageSpy.STATISTIC_GROUP.FROM_SWITCH_PUBLISHED_SUCCESS);
268         }
269     }
270
271     @Override
272     public void processReply(final Xid xid, final List<MultipartReply> ofHeaderList) {
273         for (final MultipartReply multipartReply : ofHeaderList) {
274             messageSpy.spyMessage(multipartReply.getImplementedInterface(), MessageSpy.STATISTIC_GROUP.FROM_SWITCH_PUBLISHED_FAILURE);
275         }
276     }
277
278     @Override
279     public void processFlowRemovedMessage(final FlowRemoved flowRemoved) {
280         final ItemLifecycleListener itemLifecycleListener = flowLifeCycleKeeper.getItemLifecycleListener();
281         if (itemLifecycleListener != null) {
282             //1. translate to general flow (table, priority, match, cookie)
283             final org.opendaylight.yang.gen.v1.urn.opendaylight.flow.service.rev130819.FlowRemoved flowRemovedNotification =
284                     flowRemovedTranslator.translate(flowRemoved, this, null);
285             //2. create registry key
286             FlowRegistryKey flowRegKey = FlowRegistryKeyFactory.create(flowRemovedNotification);
287             //3. lookup flowId
288             final FlowDescriptor flowDescriptor = deviceFlowRegistry.retrieveIdForFlow(flowRegKey);
289             //4. if flowId present:
290             if (flowDescriptor != null) {
291                 // a) construct flow path
292                 KeyedInstanceIdentifier<Flow, FlowKey> flowPath = getDeviceState().getNodeInstanceIdentifier()
293                         .augmentation(FlowCapableNode.class)
294                         .child(Table.class, flowDescriptor.getTableKey())
295                         .child(Flow.class, new FlowKey(flowDescriptor.getFlowId()));
296                 // b) notify listener
297                 itemLifecycleListener.onRemoved(flowPath);
298             } else {
299                 LOG.debug("flow id not found: nodeId={} tableId={}, priority={}",
300                         getDeviceState().getNodeId(), flowRegKey.getTableId(), flowRemovedNotification.getPriority());
301             }
302         }
303     }
304
305     @Override
306     public void processPortStatusMessage(final PortStatusMessage portStatus) {
307         messageSpy.spyMessage(portStatus.getImplementedInterface(), MessageSpy.STATISTIC_GROUP.FROM_SWITCH_PUBLISHED_SUCCESS);
308         final FlowCapableNodeConnector flowCapableNodeConnector = portStatusTranslator.translate(portStatus, this, null);
309
310         final KeyedInstanceIdentifier<NodeConnector, NodeConnectorKey> iiToNodeConnector = provideIIToNodeConnector(portStatus.getPortNo(), portStatus.getVersion());
311         if (portStatus.getReason().equals(PortReason.OFPPRADD) || portStatus.getReason().equals(PortReason.OFPPRMODIFY)) {
312             // because of ADD status node connector has to be created
313             final NodeConnectorBuilder nConnectorBuilder = new NodeConnectorBuilder().setKey(iiToNodeConnector.getKey());
314             nConnectorBuilder.addAugmentation(FlowCapableNodeConnectorStatisticsData.class, new FlowCapableNodeConnectorStatisticsDataBuilder().build());
315             nConnectorBuilder.addAugmentation(FlowCapableNodeConnector.class, flowCapableNodeConnector);
316             writeToTransaction(LogicalDatastoreType.OPERATIONAL, iiToNodeConnector, nConnectorBuilder.build());
317         } else if (portStatus.getReason().equals(PortReason.OFPPRDELETE)) {
318             addDeleteToTxChain(LogicalDatastoreType.OPERATIONAL, iiToNodeConnector);
319         }
320         submitTransaction();
321     }
322
323     private KeyedInstanceIdentifier<NodeConnector, NodeConnectorKey> provideIIToNodeConnector(final long portNo, final short version) {
324         final InstanceIdentifier<Node> iiToNodes = deviceState.getNodeInstanceIdentifier();
325         final BigInteger dataPathId = deviceState.getFeatures().getDatapathId();
326         final NodeConnectorId nodeConnectorId = NodeStaticReplyTranslatorUtil.nodeConnectorId(dataPathId.toString(), portNo, version);
327         return iiToNodes.child(NodeConnector.class, new NodeConnectorKey(nodeConnectorId));
328     }
329
330     @Override
331     public void processPacketInMessage(final PacketInMessage packetInMessage) {
332         messageSpy.spyMessage(packetInMessage.getImplementedInterface(), MessageSpy.STATISTIC_GROUP.FROM_SWITCH);
333         final ConnectionAdapter connectionAdapter = getPrimaryConnectionContext().getConnectionAdapter();
334         final PacketReceived packetReceived = packetInTranslator.translate(packetInMessage, this, null);
335
336         if (packetReceived == null) {
337             LOG.debug("Received a null packet from switch {}", connectionAdapter.getRemoteAddress());
338             return;
339         }
340
341         if (!packetInLimiter.acquirePermit()) {
342             LOG.debug("Packet limited");
343             // TODO: save packet into emergency slot if possible
344             // FIXME: some other counter
345             messageSpy.spyMessage(packetReceived.getImplementedInterface(), MessageSpy.STATISTIC_GROUP.FROM_SWITCH_TRANSLATE_SRC_FAILURE);
346             return;
347         }
348
349         final ListenableFuture<? extends Object> offerNotification = notificationPublishService.offerNotification(packetReceived);
350         if (NotificationPublishService.REJECTED.equals(offerNotification)) {
351             LOG.debug("notification offer rejected");
352             messageSpy.spyMessage(packetReceived.getImplementedInterface(), MessageSpy.STATISTIC_GROUP.FROM_SWITCH_TRANSLATE_SRC_FAILURE);
353             packetInLimiter.drainLowWaterMark();
354             packetInLimiter.releasePermit();
355             return;
356         }
357
358         Futures.addCallback(offerNotification, new FutureCallback<Object>() {
359             @Override
360             public void onSuccess(final Object result) {
361                 messageSpy.spyMessage(packetReceived.getImplementedInterface(), MessageSpy.STATISTIC_GROUP.FROM_SWITCH_PUBLISHED_SUCCESS);
362                 packetInLimiter.releasePermit();
363             }
364
365             @Override
366             public void onFailure(final Throwable t) {
367                 messageSpy.spyMessage(packetReceived.getImplementedInterface(), MessageSpy.STATISTIC_GROUP.FROM_SWITCH_NOTIFICATION_REJECTED);
368                 LOG.debug("notification offer failed: {}", t.getMessage());
369                 LOG.trace("notification offer failed..", t);
370                 packetInLimiter.releasePermit();
371             }
372         });
373     }
374
375     @Override
376     public TranslatorLibrary oook() {
377         return translatorLibrary;
378     }
379
380     @Override
381     public HashedWheelTimer getTimer() {
382         return hashedWheelTimer;
383     }
384
385     @Override
386     public void close() {
387         LOG.debug("closing deviceContext: {}, nodeId:{}",
388                 getPrimaryConnectionContext().getConnectionAdapter().getRemoteAddress(),
389                 getDeviceState().getNodeId());
390
391         tearDown();
392
393         primaryConnectionContext.closeConnection(false);
394     }
395
396     private void tearDown() {
397         deviceState.setValid(false);
398
399         for (final ConnectionContext connectionContext : auxiliaryConnectionContexts.values()) {
400             connectionContext.closeConnection(false);
401         }
402
403         deviceGroupRegistry.close();
404         deviceFlowRegistry.close();
405         deviceMeterRegistry.close();
406
407         itemLifeCycleSourceRegistry.clear();
408
409
410         for (final DeviceContextClosedHandler deviceContextClosedHandler : closeHandlers) {
411             deviceContextClosedHandler.onDeviceContextClosed(this);
412         }
413
414         LOG.info("Closing transaction chain manager without cleaning inventory operational");
415         transactionChainManager.close();
416     }
417
418     @Override
419     public void onDeviceDisconnectedFromCluster() {
420         LOG.info("Removing device from operational and closing transaction Manager for device:{}", getDeviceState().getNodeId());
421         transactionChainManager.cleanupPostClosure();
422     }
423
424     @Override
425     public void onDeviceDisconnected(final ConnectionContext connectionContext) {
426         if (getPrimaryConnectionContext().equals(connectionContext)) {
427             try {
428                 tearDown();
429             } catch (final Exception e) {
430                 LOG.trace("Error closing device context.");
431             }
432         } else {
433             LOG.debug("auxiliary connection dropped: {}, nodeId:{}",
434                     connectionContext.getConnectionAdapter().getRemoteAddress(),
435                     getDeviceState().getNodeId());
436             final SwitchConnectionDistinguisher connectionDistinguisher = createConnectionDistinguisher(connectionContext);
437             auxiliaryConnectionContexts.remove(connectionDistinguisher);
438         }
439     }
440
441     @Override
442     public void setCurrentBarrierTimeout(final Timeout timeout) {
443         barrierTaskTimeout = timeout;
444     }
445
446     @Override
447     public Timeout getBarrierTaskTimeout() {
448         return barrierTaskTimeout;
449     }
450
451     @Override
452     public void setNotificationService(final NotificationService notificationServiceParam) {
453         notificationService = notificationServiceParam;
454     }
455
456     @Override
457     public void setNotificationPublishService(final NotificationPublishService notificationPublishService) {
458         this.notificationPublishService = notificationPublishService;
459     }
460
461     @Override
462     public MessageSpy getMessageSpy() {
463         return messageSpy;
464     }
465
466     @Override
467     public void addDeviceContextClosedHandler(final DeviceContextClosedHandler deviceContextClosedHandler) {
468         closeHandlers.add(deviceContextClosedHandler);
469     }
470
471     @Override
472     public void onPublished() {
473         primaryConnectionContext.getConnectionAdapter().setPacketInFiltering(false);
474         for (final ConnectionContext switchAuxConnectionContext : auxiliaryConnectionContexts.values()) {
475             switchAuxConnectionContext.getConnectionAdapter().setPacketInFiltering(false);
476         }
477     }
478
479     @Override
480     public MultiMsgCollector getMultiMsgCollector(final RequestContext<List<MultipartReply>> requestContext) {
481         return new MultiMsgCollectorImpl(this, requestContext);
482     }
483
484     @Override
485     public NodeConnectorRef lookupNodeConnectorRef(Long portNumber) {
486         return nodeConnectorCache.get(portNumber);
487     }
488
489     @Override
490     public void storeNodeConnectorRef(final Long portNumber, final NodeConnectorRef nodeConnectorRef) {
491         nodeConnectorCache.put(
492                 Preconditions.checkNotNull(portNumber),
493                 Preconditions.checkNotNull(nodeConnectorRef));
494     }
495
496     @Override
497     public void updatePacketInRateLimit(long upperBound) {
498         packetInLimiter.changeWaterMarks((int) (LOW_WATERMARK_FACTOR * upperBound), (int) (HIGH_WATERMARK_FACTOR * upperBound));
499     }
500
501     @Override
502     public ItemLifeCycleRegistry getItemLifeCycleSourceRegistry() {
503         return itemLifeCycleSourceRegistry;
504     }
505
506     @Override
507     public void setRpcContext(RpcContext rpcContext) {
508         this.rpcContext = rpcContext;
509     }
510
511     @Override
512     public RpcContext getRpcContext() {
513         return rpcContext;
514     }
515
516
517 }