tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 1 | package org.onlab.onos.net.trivial.impl; |
| 2 | |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 3 | import com.google.common.collect.Sets; |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 4 | import org.apache.felix.scr.annotations.Activate; |
| 5 | import org.apache.felix.scr.annotations.Component; |
| 6 | import org.apache.felix.scr.annotations.Deactivate; |
tom | 5f38b3a | 2014-08-27 23:50:54 -0700 | [diff] [blame] | 7 | import org.apache.felix.scr.annotations.Reference; |
| 8 | import org.apache.felix.scr.annotations.ReferenceCardinality; |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 9 | import org.apache.felix.scr.annotations.Service; |
tom | 96dfcab | 2014-08-28 09:26:03 -0700 | [diff] [blame] | 10 | import org.onlab.onos.event.AbstractListenerRegistry; |
| 11 | import org.onlab.onos.event.EventDeliveryService; |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 12 | import org.onlab.onos.net.ConnectPoint; |
| 13 | import org.onlab.onos.net.DeviceId; |
| 14 | import org.onlab.onos.net.Link; |
| 15 | import org.onlab.onos.net.link.LinkAdminService; |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 16 | import org.onlab.onos.net.link.LinkDescription; |
tom | 5f38b3a | 2014-08-27 23:50:54 -0700 | [diff] [blame] | 17 | import org.onlab.onos.net.link.LinkEvent; |
| 18 | import org.onlab.onos.net.link.LinkListener; |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 19 | import org.onlab.onos.net.link.LinkProvider; |
tom | 96dfcab | 2014-08-28 09:26:03 -0700 | [diff] [blame] | 20 | import org.onlab.onos.net.link.LinkProviderRegistry; |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 21 | import org.onlab.onos.net.link.LinkProviderService; |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 22 | import org.onlab.onos.net.link.LinkService; |
tom | 96dfcab | 2014-08-28 09:26:03 -0700 | [diff] [blame] | 23 | import org.onlab.onos.net.provider.AbstractProviderRegistry; |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 24 | import org.onlab.onos.net.provider.AbstractProviderService; |
| 25 | import org.slf4j.Logger; |
tom | 5f38b3a | 2014-08-27 23:50:54 -0700 | [diff] [blame] | 26 | |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 27 | import java.util.Set; |
| 28 | |
| 29 | import static com.google.common.base.Preconditions.checkNotNull; |
tom | 5f38b3a | 2014-08-27 23:50:54 -0700 | [diff] [blame] | 30 | import static org.slf4j.LoggerFactory.getLogger; |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 31 | |
| 32 | /** |
| 33 | * Provides basic implementation of the link SB & NB APIs. |
| 34 | */ |
| 35 | @Component(immediate = true) |
| 36 | @Service |
tom | 5f38b3a | 2014-08-27 23:50:54 -0700 | [diff] [blame] | 37 | public class SimpleLinkManager |
tom | 96dfcab | 2014-08-28 09:26:03 -0700 | [diff] [blame] | 38 | extends AbstractProviderRegistry<LinkProvider, LinkProviderService> |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 39 | implements LinkService, LinkAdminService, LinkProviderRegistry { |
| 40 | |
| 41 | private static final String DEVICE_ID_NULL = "Device ID cannot be null"; |
| 42 | private static final String LINK_DESC_NULL = "Link description cannot be null"; |
| 43 | private static final String CONNECT_POINT_NULL = "Connection point cannot be null"; |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 44 | |
tom | 5f38b3a | 2014-08-27 23:50:54 -0700 | [diff] [blame] | 45 | private final Logger log = getLogger(getClass()); |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 46 | |
tom | 96dfcab | 2014-08-28 09:26:03 -0700 | [diff] [blame] | 47 | private final AbstractListenerRegistry<LinkEvent, LinkListener> |
| 48 | listenerRegistry = new AbstractListenerRegistry<>(); |
tom | 5f38b3a | 2014-08-27 23:50:54 -0700 | [diff] [blame] | 49 | |
tom | 4c6606f | 2014-09-07 11:11:21 -0700 | [diff] [blame] | 50 | private final SimpleLinkStore store = new SimpleLinkStore(); |
| 51 | |
tom | 5f38b3a | 2014-08-27 23:50:54 -0700 | [diff] [blame] | 52 | @Reference(cardinality = ReferenceCardinality.MANDATORY_UNARY) |
tom | 0d39526 | 2014-09-07 16:53:40 -0700 | [diff] [blame] | 53 | protected EventDeliveryService eventDispatcher; |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 54 | |
| 55 | @Activate |
| 56 | public void activate() { |
tom | 96dfcab | 2014-08-28 09:26:03 -0700 | [diff] [blame] | 57 | eventDispatcher.addSink(LinkEvent.class, listenerRegistry); |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 58 | log.info("Started"); |
| 59 | } |
| 60 | |
| 61 | @Deactivate |
| 62 | public void deactivate() { |
tom | 5f38b3a | 2014-08-27 23:50:54 -0700 | [diff] [blame] | 63 | eventDispatcher.removeSink(LinkEvent.class); |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 64 | log.info("Stopped"); |
| 65 | } |
| 66 | |
| 67 | @Override |
tom | 5f38b3a | 2014-08-27 23:50:54 -0700 | [diff] [blame] | 68 | protected LinkProviderService createProviderService(LinkProvider provider) { |
| 69 | return new InternalLinkProviderService(provider); |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 70 | } |
| 71 | |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 72 | @Override |
| 73 | public int getLinkCount() { |
| 74 | return store.getLinkCount(); |
| 75 | } |
| 76 | |
| 77 | @Override |
| 78 | public Iterable<Link> getLinks() { |
| 79 | return store.getLinks(); |
| 80 | } |
| 81 | |
| 82 | @Override |
| 83 | public Set<Link> getDeviceLinks(DeviceId deviceId) { |
| 84 | checkNotNull(deviceId, DEVICE_ID_NULL); |
| 85 | return Sets.union(store.getDeviceEgressLinks(deviceId), |
| 86 | store.getDeviceIngressLinks(deviceId)); |
| 87 | } |
| 88 | |
| 89 | @Override |
| 90 | public Set<Link> getDeviceEgressLinks(DeviceId deviceId) { |
| 91 | checkNotNull(deviceId, DEVICE_ID_NULL); |
| 92 | return store.getDeviceEgressLinks(deviceId); |
| 93 | } |
| 94 | |
| 95 | @Override |
tom | d176fc4 | 2014-09-08 00:12:30 -0700 | [diff] [blame] | 96 | public Set<Link> getDeviceIngressLinks(DeviceId deviceId) { |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 97 | checkNotNull(deviceId, DEVICE_ID_NULL); |
| 98 | return store.getDeviceIngressLinks(deviceId); |
| 99 | } |
| 100 | |
| 101 | @Override |
| 102 | public Set<Link> getLinks(ConnectPoint connectPoint) { |
| 103 | checkNotNull(connectPoint, CONNECT_POINT_NULL); |
| 104 | return Sets.union(store.getEgressLinks(connectPoint), |
| 105 | store.getIngressLinks(connectPoint)); |
| 106 | } |
| 107 | |
| 108 | @Override |
| 109 | public Set<Link> getEgressLinks(ConnectPoint connectPoint) { |
| 110 | checkNotNull(connectPoint, CONNECT_POINT_NULL); |
| 111 | return store.getEgressLinks(connectPoint); |
| 112 | } |
| 113 | |
| 114 | @Override |
tom | d176fc4 | 2014-09-08 00:12:30 -0700 | [diff] [blame] | 115 | public Set<Link> getIngressLinks(ConnectPoint connectPoint) { |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 116 | checkNotNull(connectPoint, CONNECT_POINT_NULL); |
| 117 | return store.getIngressLinks(connectPoint); |
| 118 | } |
| 119 | |
| 120 | @Override |
tom | d176fc4 | 2014-09-08 00:12:30 -0700 | [diff] [blame] | 121 | public Link getLink(ConnectPoint src, ConnectPoint dst) { |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 122 | checkNotNull(src, CONNECT_POINT_NULL); |
| 123 | checkNotNull(dst, CONNECT_POINT_NULL); |
tom | d176fc4 | 2014-09-08 00:12:30 -0700 | [diff] [blame] | 124 | return store.getLink(src, dst); |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 125 | } |
| 126 | |
| 127 | @Override |
| 128 | public void removeLinks(ConnectPoint connectPoint) { |
| 129 | removeLinks(getLinks(connectPoint)); |
| 130 | } |
| 131 | |
| 132 | @Override |
| 133 | public void removeLinks(DeviceId deviceId) { |
| 134 | removeLinks(getDeviceLinks(deviceId)); |
| 135 | } |
| 136 | |
| 137 | @Override |
| 138 | public void addListener(LinkListener listener) { |
| 139 | listenerRegistry.addListener(listener); |
| 140 | } |
| 141 | |
| 142 | @Override |
| 143 | public void removeListener(LinkListener listener) { |
| 144 | listenerRegistry.removeListener(listener); |
| 145 | } |
| 146 | |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 147 | // Personalized link provider service issued to the supplied provider. |
| 148 | private class InternalLinkProviderService extends AbstractProviderService<LinkProvider> |
| 149 | implements LinkProviderService { |
| 150 | |
tom | cfde062 | 2014-09-09 11:02:42 -0700 | [diff] [blame^] | 151 | InternalLinkProviderService(LinkProvider provider) { |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 152 | super(provider); |
| 153 | } |
| 154 | |
| 155 | @Override |
| 156 | public void linkDetected(LinkDescription linkDescription) { |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 157 | checkNotNull(linkDescription, LINK_DESC_NULL); |
| 158 | checkValidity(); |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 159 | log.info("Link {} detected", linkDescription); |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 160 | LinkEvent event = store.createOrUpdateLink(provider().id(), |
| 161 | linkDescription); |
| 162 | post(event); |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 163 | } |
| 164 | |
| 165 | @Override |
| 166 | public void linkVanished(LinkDescription linkDescription) { |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 167 | checkNotNull(linkDescription, LINK_DESC_NULL); |
| 168 | checkValidity(); |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 169 | log.info("Link {} vanished", linkDescription); |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 170 | LinkEvent event = store.removeLink(linkDescription.src(), |
| 171 | linkDescription.dst()); |
| 172 | post(event); |
| 173 | } |
| 174 | |
| 175 | @Override |
| 176 | public void linksVanished(ConnectPoint connectPoint) { |
| 177 | checkNotNull(connectPoint, "Connect point cannot be null"); |
| 178 | checkValidity(); |
| 179 | log.info("Link for connection point {} vanished", connectPoint); |
| 180 | removeLinks(getLinks(connectPoint)); |
| 181 | } |
| 182 | |
| 183 | @Override |
| 184 | public void linksVanished(DeviceId deviceId) { |
| 185 | checkNotNull(deviceId, DEVICE_ID_NULL); |
| 186 | checkValidity(); |
| 187 | log.info("Link for device {} vanished", deviceId); |
| 188 | removeLinks(getDeviceLinks(deviceId)); |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 189 | } |
| 190 | } |
tom | eadbb46 | 2014-09-07 16:10:19 -0700 | [diff] [blame] | 191 | |
| 192 | // Removes all links in the specified set and emits appropriate events. |
| 193 | private void removeLinks(Set<Link> links) { |
| 194 | for (Link link : links) { |
| 195 | LinkEvent event = store.removeLink(link.src(), link.dst()); |
| 196 | post(event); |
| 197 | } |
| 198 | } |
| 199 | |
| 200 | // Posts the specified event to the local event dispatcher. |
| 201 | private void post(LinkEvent event) { |
| 202 | if (event != null && eventDispatcher != null) { |
| 203 | eventDispatcher.post(event); |
| 204 | } |
| 205 | } |
| 206 | |
tom | db0d03f | 2014-08-27 16:34:15 -0700 | [diff] [blame] | 207 | } |