Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2014-2016 Open Networking Laboratory |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
| 17 | package org.onosproject.provider.bmv2.device.impl; |
| 18 | |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 19 | import com.google.common.collect.Maps; |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 20 | import org.apache.felix.scr.annotations.Component; |
| 21 | import org.apache.felix.scr.annotations.Reference; |
| 22 | import org.apache.felix.scr.annotations.ReferenceCardinality; |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 23 | import org.jboss.netty.util.HashedWheelTimer; |
| 24 | import org.jboss.netty.util.Timeout; |
| 25 | import org.jboss.netty.util.TimerTask; |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 26 | import org.onlab.util.Timer; |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 27 | import org.onosproject.bmv2.api.runtime.Bmv2Device; |
Carmelo Cascone | 442a962 | 2016-05-03 11:16:20 -0700 | [diff] [blame] | 28 | import org.onosproject.bmv2.api.runtime.Bmv2RuntimeException; |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 29 | import org.onosproject.bmv2.api.service.Bmv2Controller; |
| 30 | import org.onosproject.bmv2.api.service.Bmv2DeviceContextService; |
| 31 | import org.onosproject.bmv2.api.service.Bmv2DeviceListener; |
Carmelo Cascone | ee4cd7e | 2016-06-16 18:28:43 -0700 | [diff] [blame] | 32 | import org.onosproject.bmv2.api.service.Bmv2TableEntryService; |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 33 | import org.onosproject.common.net.AbstractDeviceProvider; |
| 34 | import org.onosproject.core.ApplicationId; |
| 35 | import org.onosproject.core.CoreService; |
| 36 | import org.onosproject.incubator.net.config.basics.ConfigException; |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 37 | import org.onosproject.net.Device; |
| 38 | import org.onosproject.net.DeviceId; |
| 39 | import org.onosproject.net.MastershipRole; |
| 40 | import org.onosproject.net.PortNumber; |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 41 | import org.onosproject.net.config.ConfigFactory; |
| 42 | import org.onosproject.net.config.NetworkConfigEvent; |
| 43 | import org.onosproject.net.config.NetworkConfigListener; |
| 44 | import org.onosproject.net.config.NetworkConfigRegistry; |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 45 | import org.onosproject.net.device.DeviceDescription; |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 46 | import org.onosproject.net.device.DeviceDescriptionDiscovery; |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 47 | import org.onosproject.net.device.DeviceService; |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 48 | import org.onosproject.net.device.PortDescription; |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 49 | import org.onosproject.net.device.PortStatistics; |
| 50 | import org.onosproject.net.driver.DefaultDriverData; |
| 51 | import org.onosproject.net.driver.DefaultDriverHandler; |
| 52 | import org.onosproject.net.driver.Driver; |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 53 | import org.onosproject.net.provider.ProviderId; |
| 54 | import org.slf4j.Logger; |
| 55 | |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 56 | import java.util.Collection; |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 57 | import java.util.List; |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 58 | import java.util.Objects; |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 59 | import java.util.concurrent.ConcurrentMap; |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 60 | import java.util.concurrent.ExecutorService; |
| 61 | import java.util.concurrent.Executors; |
| 62 | import java.util.concurrent.TimeUnit; |
| 63 | |
| 64 | import static org.onlab.util.Tools.groupedThreads; |
Carmelo Cascone | 442a962 | 2016-05-03 11:16:20 -0700 | [diff] [blame] | 65 | import static org.onosproject.bmv2.api.runtime.Bmv2Device.*; |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 66 | import static org.onosproject.net.config.basics.SubjectFactories.APP_SUBJECT_FACTORY; |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 67 | import static org.onosproject.provider.bmv2.device.impl.Bmv2PortStatisticsGetter.getPortStatistics; |
| 68 | import static org.onosproject.provider.bmv2.device.impl.Bmv2PortStatisticsGetter.initCounters; |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 69 | import static org.slf4j.LoggerFactory.getLogger; |
| 70 | |
| 71 | /** |
| 72 | * BMv2 device provider. |
| 73 | */ |
| 74 | @Component(immediate = true) |
| 75 | public class Bmv2DeviceProvider extends AbstractDeviceProvider { |
| 76 | |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 77 | private static final String APP_NAME = "org.onosproject.bmv2"; |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 78 | |
| 79 | private static final int POLL_INTERVAL = 5_000; // milliseconds |
| 80 | |
| 81 | private final Logger log = getLogger(this.getClass()); |
| 82 | |
| 83 | private final ExecutorService executorService = Executors |
| 84 | .newFixedThreadPool(16, groupedThreads("onos/bmv2", "device-discovery", log)); |
| 85 | |
| 86 | private final NetworkConfigListener cfgListener = new InternalNetworkConfigListener(); |
| 87 | |
| 88 | private final ConfigFactory cfgFactory = new InternalConfigFactory(); |
| 89 | |
| 90 | private final ConcurrentMap<DeviceId, DeviceDescription> activeDevices = Maps.newConcurrentMap(); |
| 91 | |
| 92 | private final DevicePoller devicePoller = new DevicePoller(); |
| 93 | |
| 94 | private final InternalDeviceListener deviceListener = new InternalDeviceListener(); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 95 | |
| 96 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) |
| 97 | protected NetworkConfigRegistry netCfgService; |
| 98 | |
| 99 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) |
| 100 | protected CoreService coreService; |
| 101 | |
| 102 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) |
| 103 | protected DeviceService deviceService; |
| 104 | |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 105 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 106 | protected Bmv2Controller controller; |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 107 | |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 108 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) |
| 109 | protected Bmv2DeviceContextService contextService; |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 110 | |
Carmelo Cascone | ee4cd7e | 2016-06-16 18:28:43 -0700 | [diff] [blame] | 111 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) |
| 112 | protected Bmv2TableEntryService tableEntryService; |
| 113 | |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 114 | private ApplicationId appId; |
| 115 | |
| 116 | /** |
| 117 | * Creates a Bmv2 device provider with the supplied identifier. |
| 118 | */ |
| 119 | public Bmv2DeviceProvider() { |
| 120 | super(new ProviderId("bmv2", "org.onosproject.provider.device")); |
| 121 | } |
| 122 | |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 123 | @Override |
| 124 | protected void activate() { |
| 125 | appId = coreService.registerApplication(APP_NAME); |
| 126 | netCfgService.registerConfigFactory(cfgFactory); |
| 127 | netCfgService.addListener(cfgListener); |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 128 | controller.addDeviceListener(deviceListener); |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 129 | devicePoller.start(); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 130 | super.activate(); |
| 131 | } |
| 132 | |
| 133 | @Override |
| 134 | protected void deactivate() { |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 135 | devicePoller.stop(); |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 136 | controller.removeDeviceListener(deviceListener); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 137 | try { |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 138 | activeDevices.forEach((did, value) -> { |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 139 | executorService.execute(() -> disconnectDevice(did)); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 140 | }); |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 141 | executorService.awaitTermination(1000, TimeUnit.MILLISECONDS); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 142 | } catch (InterruptedException e) { |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 143 | log.error("Device discovery threads did not terminate"); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 144 | } |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 145 | executorService.shutdownNow(); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 146 | netCfgService.unregisterConfigFactory(cfgFactory); |
| 147 | netCfgService.removeListener(cfgListener); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 148 | super.deactivate(); |
| 149 | } |
| 150 | |
| 151 | @Override |
| 152 | public void triggerProbe(DeviceId deviceId) { |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 153 | // Asynchronously trigger probe task. |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 154 | executorService.execute(() -> executeProbe(deviceId)); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 155 | } |
| 156 | |
| 157 | private void executeProbe(DeviceId did) { |
| 158 | boolean reachable = isReachable(did); |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 159 | log.debug("Probed device: id={}, reachable={}", did.toString(), reachable); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 160 | if (reachable) { |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 161 | discoverDevice(did); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 162 | } else { |
| 163 | disconnectDevice(did); |
| 164 | } |
| 165 | } |
| 166 | |
| 167 | @Override |
| 168 | public void roleChanged(DeviceId deviceId, MastershipRole newRole) { |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 169 | log.debug("roleChanged() is not yet implemented"); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 170 | // TODO: implement mastership handling |
| 171 | } |
| 172 | |
| 173 | @Override |
| 174 | public boolean isReachable(DeviceId deviceId) { |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 175 | return controller.isReacheable(deviceId); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 176 | } |
| 177 | |
| 178 | @Override |
| 179 | public void changePortState(DeviceId deviceId, PortNumber portNumber, boolean enable) { |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 180 | log.warn("changePortState() not supported"); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 181 | } |
| 182 | |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 183 | private void discoverDevice(DeviceId did) { |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 184 | log.debug("Starting device discovery... deviceId={}", did); |
| 185 | activeDevices.compute(did, (k, lastDescription) -> { |
| 186 | DeviceDescription thisDescription = getDeviceDescription(did); |
| 187 | if (thisDescription != null) { |
| 188 | boolean descriptionChanged = lastDescription != null && |
| 189 | (!Objects.equals(thisDescription, lastDescription) || |
| 190 | !Objects.equals(thisDescription.annotations(), lastDescription.annotations())); |
| 191 | if (descriptionChanged || !deviceService.isAvailable(did)) { |
Carmelo Cascone | fbc577b | 2016-06-17 23:19:09 -0700 | [diff] [blame^] | 192 | if (deviceService.getDevice(did) == null) { |
Carmelo Cascone | c18e82c | 2016-06-16 14:22:36 -0700 | [diff] [blame] | 193 | // Device is a first timer. |
| 194 | log.info("Setting DEFAULT context for {}", did); |
Carmelo Cascone | fbc577b | 2016-06-17 23:19:09 -0700 | [diff] [blame^] | 195 | // It is important to do this before connecting the device so other |
| 196 | // services won't find a null context. |
Carmelo Cascone | c18e82c | 2016-06-16 14:22:36 -0700 | [diff] [blame] | 197 | contextService.setContext(did, contextService.defaultContext()); |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 198 | } |
Carmelo Cascone | fbc577b | 2016-06-17 23:19:09 -0700 | [diff] [blame^] | 199 | resetDeviceState(did); |
| 200 | initPortCounters(did); |
| 201 | providerService.deviceConnected(did, thisDescription); |
| 202 | updatePortsAndStats(did); |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 203 | } |
| 204 | return thisDescription; |
| 205 | } else { |
| 206 | log.warn("Unable to get device description for {}", did); |
| 207 | return lastDescription; |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 208 | } |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 209 | }); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 210 | } |
| 211 | |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 212 | private DeviceDescription getDeviceDescription(DeviceId did) { |
| 213 | Device device = deviceService.getDevice(did); |
| 214 | DeviceDescriptionDiscovery discovery = null; |
| 215 | if (device == null) { |
| 216 | // Device not yet in the core. Manually get a driver. |
| 217 | Driver driver = driverService.getDriver(MANUFACTURER, HW_VERSION, SW_VERSION); |
| 218 | if (driver.hasBehaviour(DeviceDescriptionDiscovery.class)) { |
| 219 | discovery = driver.createBehaviour(new DefaultDriverHandler(new DefaultDriverData(driver, did)), |
| 220 | DeviceDescriptionDiscovery.class); |
| 221 | } |
| 222 | } else if (device.is(DeviceDescriptionDiscovery.class)) { |
| 223 | discovery = device.as(DeviceDescriptionDiscovery.class); |
| 224 | } |
| 225 | if (discovery == null) { |
| 226 | log.warn("No DeviceDescriptionDiscovery behavior for device {}", did); |
| 227 | return null; |
| 228 | } else { |
| 229 | return discovery.discoverDeviceDetails(); |
Carmelo Cascone | a2f510e | 2016-05-03 18:36:45 -0700 | [diff] [blame] | 230 | } |
| 231 | } |
| 232 | |
| 233 | private void resetDeviceState(DeviceId did) { |
| 234 | try { |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 235 | controller.getAgent(did).resetState(); |
Carmelo Cascone | ee4cd7e | 2016-06-16 18:28:43 -0700 | [diff] [blame] | 236 | // Tables emptied. Reset all bindings. |
| 237 | tableEntryService.unbindAll(did); |
Carmelo Cascone | a2f510e | 2016-05-03 18:36:45 -0700 | [diff] [blame] | 238 | } catch (Bmv2RuntimeException e) { |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 239 | log.warn("Unable to reset {}: {}", did, e.toString()); |
Carmelo Cascone | 442a962 | 2016-05-03 11:16:20 -0700 | [diff] [blame] | 240 | } |
| 241 | } |
| 242 | |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 243 | private void initPortCounters(DeviceId did) { |
| 244 | try { |
| 245 | initCounters(controller.getAgent(did)); |
| 246 | } catch (Bmv2RuntimeException e) { |
| 247 | log.warn("Unable to init counter on {}: {}", did, e.explain()); |
| 248 | } |
| 249 | } |
| 250 | |
| 251 | private void updatePortsAndStats(DeviceId did) { |
Carmelo Cascone | 442a962 | 2016-05-03 11:16:20 -0700 | [diff] [blame] | 252 | Device device = deviceService.getDevice(did); |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 253 | if (device.is(DeviceDescriptionDiscovery.class)) { |
| 254 | DeviceDescriptionDiscovery discovery = device.as(DeviceDescriptionDiscovery.class); |
| 255 | List<PortDescription> portDescriptions = discovery.discoverPortDetails(); |
| 256 | if (portDescriptions != null) { |
| 257 | providerService.updatePorts(did, portDescriptions); |
| 258 | } |
Carmelo Cascone | 442a962 | 2016-05-03 11:16:20 -0700 | [diff] [blame] | 259 | } else { |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 260 | log.warn("No DeviceDescriptionDiscovery behavior for device {}", did); |
| 261 | } |
| 262 | try { |
| 263 | Collection<PortStatistics> portStats = getPortStatistics(controller.getAgent(did), |
| 264 | deviceService.getPorts(did)); |
| 265 | providerService.updatePortStatistics(did, portStats); |
| 266 | } catch (Bmv2RuntimeException e) { |
| 267 | log.warn("Unable to get port statistics for {}: {}", did, e.explain()); |
Carmelo Cascone | 442a962 | 2016-05-03 11:16:20 -0700 | [diff] [blame] | 268 | } |
| 269 | } |
| 270 | |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 271 | private void disconnectDevice(DeviceId did) { |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 272 | log.debug("Trying to disconnect device from core... deviceId={}", did); |
Carmelo Cascone | 6256d01 | 2016-06-17 13:49:52 -0700 | [diff] [blame] | 273 | if (deviceService.isAvailable(did)) { |
| 274 | providerService.deviceDisconnected(did); |
| 275 | } |
Carmelo Cascone | fbc577b | 2016-06-17 23:19:09 -0700 | [diff] [blame^] | 276 | activeDevices.remove(did); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 277 | } |
| 278 | |
| 279 | /** |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 280 | * Internal net-cfg config factory. |
| 281 | */ |
| 282 | private class InternalConfigFactory extends ConfigFactory<ApplicationId, Bmv2ProviderConfig> { |
| 283 | |
| 284 | InternalConfigFactory() { |
| 285 | super(APP_SUBJECT_FACTORY, Bmv2ProviderConfig.class, "devices", true); |
| 286 | } |
| 287 | |
| 288 | @Override |
| 289 | public Bmv2ProviderConfig createConfig() { |
| 290 | return new Bmv2ProviderConfig(); |
| 291 | } |
| 292 | } |
| 293 | |
| 294 | /** |
| 295 | * Internal net-cfg event listener. |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 296 | */ |
| 297 | private class InternalNetworkConfigListener implements NetworkConfigListener { |
| 298 | |
| 299 | @Override |
| 300 | public void event(NetworkConfigEvent event) { |
| 301 | Bmv2ProviderConfig cfg = netCfgService.getConfig(appId, Bmv2ProviderConfig.class); |
| 302 | if (cfg != null) { |
| 303 | try { |
| 304 | cfg.getDevicesInfo().stream().forEach(info -> { |
Carmelo Cascone | c18e82c | 2016-06-16 14:22:36 -0700 | [diff] [blame] | 305 | // FIXME: require also bmv2 internal device id from net-cfg (now is default 0) |
Carmelo Cascone | c0fbbee | 2016-04-27 18:03:36 -0700 | [diff] [blame] | 306 | Bmv2Device bmv2Device = new Bmv2Device(info.ip().toString(), info.port(), 0); |
| 307 | triggerProbe(bmv2Device.asDeviceId()); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 308 | }); |
| 309 | } catch (ConfigException e) { |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 310 | log.error("Unable to read config: " + e); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 311 | } |
| 312 | } else { |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 313 | log.error("Unable to read config (was null)"); |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 314 | } |
| 315 | } |
| 316 | |
| 317 | @Override |
| 318 | public boolean isRelevant(NetworkConfigEvent event) { |
| 319 | return event.configClass().equals(Bmv2ProviderConfig.class) && |
| 320 | (event.type() == NetworkConfigEvent.Type.CONFIG_ADDED || |
| 321 | event.type() == NetworkConfigEvent.Type.CONFIG_UPDATED); |
| 322 | } |
| 323 | } |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 324 | |
| 325 | /** |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 326 | * Listener triggered by the BMv2 controller each time a hello message is received. |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 327 | */ |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 328 | private class InternalDeviceListener implements Bmv2DeviceListener { |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 329 | @Override |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 330 | public void handleHello(Bmv2Device device, int instanceId, String jsonConfigMd5) { |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 331 | log.debug("Received hello from {}", device); |
Carmelo Cascone | c0fbbee | 2016-04-27 18:03:36 -0700 | [diff] [blame] | 332 | triggerProbe(device.asDeviceId()); |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 333 | } |
| 334 | } |
| 335 | |
| 336 | /** |
Carmelo Cascone | fbc577b | 2016-06-17 23:19:09 -0700 | [diff] [blame^] | 337 | * Task that periodically trigger device probes to check for device status and update port information. |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 338 | */ |
| 339 | private class DevicePoller implements TimerTask { |
| 340 | |
| 341 | private final HashedWheelTimer timer = Timer.getTimer(); |
| 342 | private Timeout timeout; |
| 343 | |
| 344 | @Override |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 345 | public void run(Timeout tout) throws Exception { |
| 346 | if (tout.isCancelled()) { |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 347 | return; |
| 348 | } |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 349 | activeDevices.keySet() |
| 350 | .stream() |
| 351 | // Filter out devices not yet created in the core. |
| 352 | .filter(did -> deviceService.getDevice(did) != null) |
| 353 | .forEach(did -> executorService.execute(() -> pollingTask(did))); |
| 354 | tout.getTimer().newTimeout(this, POLL_INTERVAL, TimeUnit.MILLISECONDS); |
| 355 | } |
| 356 | |
| 357 | private void pollingTask(DeviceId deviceId) { |
| 358 | if (isReachable(deviceId)) { |
| 359 | updatePortsAndStats(deviceId); |
| 360 | } else { |
| 361 | disconnectDevice(deviceId); |
| 362 | } |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 363 | } |
| 364 | |
| 365 | /** |
| 366 | * Starts the collector. |
| 367 | */ |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 368 | synchronized void start() { |
| 369 | log.info("Starting device poller..."); |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 370 | timeout = timer.newTimeout(this, 1, TimeUnit.SECONDS); |
| 371 | } |
| 372 | |
| 373 | /** |
| 374 | * Stops the collector. |
| 375 | */ |
| 376 | synchronized void stop() { |
Carmelo Cascone | 0831efb | 2016-05-31 14:50:19 -0700 | [diff] [blame] | 377 | log.info("Stopping device poller..."); |
Carmelo Cascone | 5fa651e | 2016-04-27 17:35:57 -0700 | [diff] [blame] | 378 | timeout.cancel(); |
| 379 | } |
| 380 | } |
Carmelo Cascone | 3bb71c1 | 2016-04-06 21:30:44 -0700 | [diff] [blame] | 381 | } |