HIGUCHI Yuta | 2d01158 | 2013-06-15 01:47:11 -0700 | [diff] [blame] | 1 | package net.onrc.onos.ofcontroller.core.internal; |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 2 | |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 3 | import java.util.ArrayList; |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 4 | import java.util.Arrays; |
Jonathan Hart | 18ad950 | 2013-12-15 18:28:00 -0800 | [diff] [blame] | 5 | import java.util.Iterator; |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 6 | import java.util.List; |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 7 | |
| 8 | import net.floodlightcontroller.devicemanager.IDevice; |
| 9 | import net.floodlightcontroller.devicemanager.SwitchPort; |
| 10 | import net.onrc.onos.graph.GraphDBOperation; |
| 11 | import net.onrc.onos.ofcontroller.core.IDeviceStorage; |
| 12 | import net.onrc.onos.ofcontroller.core.INetMapTopologyObjects.IDeviceObject; |
| 13 | import net.onrc.onos.ofcontroller.core.INetMapTopologyObjects.IIpv4Address; |
| 14 | import net.onrc.onos.ofcontroller.core.INetMapTopologyObjects.IPortObject; |
Jonathan Hart | d857ad6 | 2013-12-14 18:08:17 -0800 | [diff] [blame] | 15 | import net.onrc.onos.ofcontroller.core.INetMapTopologyObjects.ISwitchObject; |
| 16 | import net.onrc.onos.ofcontroller.devicemanager.OnosDevice; |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 17 | |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 18 | import org.openflow.util.HexString; |
| 19 | import org.slf4j.Logger; |
| 20 | import org.slf4j.LoggerFactory; |
| 21 | |
| 22 | import com.google.common.collect.Lists; |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 23 | import com.google.common.net.InetAddresses; |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 24 | import com.thinkaurelius.titan.core.TitanException; |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 25 | |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 26 | /** |
| 27 | * This is the class for storing the information of devices into CassandraDB |
| 28 | * @author Pankaj |
| 29 | */ |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 30 | public class DeviceStorageImpl implements IDeviceStorage { |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 31 | protected final static Logger log = LoggerFactory.getLogger(DeviceStorageImpl.class); |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 32 | |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 33 | private GraphDBOperation ope; |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 34 | |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 35 | /*** |
| 36 | * Initialize function. Before you use this class, please call this method |
| 37 | * @param conf configuration file for Cassandra DB |
| 38 | */ |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 39 | @Override |
| 40 | public void init(String conf) { |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 41 | try { |
Toshio Koide | bfe9b92 | 2013-06-18 10:56:05 -0700 | [diff] [blame] | 42 | ope = new GraphDBOperation(conf); |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 43 | } catch (TitanException e) { |
| 44 | log.error("Couldn't open graph operation", e); |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 45 | } |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 46 | } |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 47 | |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 48 | /*** |
| 49 | * Finalize/close function. After you use this class, please call this method. |
| 50 | * It will close the DB connection. |
| 51 | */ |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 52 | @Override |
| 53 | public void close() { |
Toshio Koide | bfe9b92 | 2013-06-18 10:56:05 -0700 | [diff] [blame] | 54 | ope.close(); |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 55 | } |
| 56 | |
| 57 | /*** |
| 58 | * Finalize/close function. After you use this class, please call this method. |
Pavlin Radoslavov | ef0cb00 | 2013-06-21 14:55:23 -0700 | [diff] [blame] | 59 | * It will close the DB connection. This is for Java garbage collection. |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 60 | */ |
| 61 | @Override |
| 62 | public void finalize() { |
| 63 | close(); |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 64 | } |
| 65 | |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 66 | /*** |
| 67 | * This function is for adding the device into the DB. |
| 68 | * @param device The device you want to add into the DB. |
| 69 | * @return IDeviceObject which was added in the DB. |
| 70 | */ |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 71 | @Override |
| 72 | public IDeviceObject addDevice(IDevice device) { |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 73 | IDeviceObject obj = null; |
Jonathan Hart | ba9ced9 | 2013-11-24 16:52:13 -0800 | [diff] [blame] | 74 | for (int i = 0; i < 6; i++) { |
| 75 | try { |
| 76 | if (i > 0) { |
| 77 | log.debug("Retrying add device: i is {}", i); |
| 78 | } |
| 79 | if ((obj = ope.searchDevice(device.getMACAddressString())) != null) { |
| 80 | log.debug("Adding device {}: found existing device", device.getMACAddressString()); |
| 81 | } else { |
| 82 | obj = ope.newDevice(); |
| 83 | log.debug("Adding device {}: creating new device", device.getMACAddressString()); |
| 84 | } |
| 85 | |
Naoki Shiota | e2f4da7 | 2013-12-09 16:34:17 -0800 | [diff] [blame] | 86 | if (obj == null) { |
| 87 | return null; |
| 88 | } |
| 89 | |
Jonathan Hart | ba9ced9 | 2013-11-24 16:52:13 -0800 | [diff] [blame] | 90 | changeDeviceAttachments(device, obj); |
| 91 | |
| 92 | changeDeviceIpv4Addresses(device, obj); |
| 93 | |
| 94 | obj.setMACAddress(device.getMACAddressString()); |
| 95 | obj.setType("device"); |
| 96 | obj.setState("ACTIVE"); |
| 97 | ope.commit(); |
| 98 | |
| 99 | break; |
| 100 | //log.debug("Adding device {}",device.getMACAddressString()); |
| 101 | } catch (TitanException e) { |
| 102 | ope.rollback(); |
| 103 | log.error("Adding device {} failed", device.getMACAddressString(), e); |
| 104 | obj = null; |
| 105 | } |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 106 | } |
| 107 | |
| 108 | return obj; |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 109 | } |
| 110 | |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 111 | /*** |
| 112 | * This function is for updating the Device properties. |
| 113 | * @param device The device you want to add into the DB. |
| 114 | * @return IDeviceObject which was added in the DB. |
| 115 | */ |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 116 | @Override |
| 117 | public IDeviceObject updateDevice(IDevice device) { |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 118 | return addDevice(device); |
| 119 | } |
| 120 | |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 121 | /*** |
| 122 | * This function is for removing the Device from the DB. |
| 123 | * @param device The device you want to delete from the DB. |
| 124 | */ |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 125 | @Override |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 126 | public void removeDevice(IDevice device) { |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 127 | IDeviceObject dev; |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 128 | |
| 129 | if ((dev = ope.searchDevice(device.getMACAddressString())) != null) { |
| 130 | removeDevice(dev); |
| 131 | } |
| 132 | } |
| 133 | |
| 134 | public void removeDevice(IDeviceObject deviceObject) { |
| 135 | String deviceMac = deviceObject.getMACAddress(); |
| 136 | |
Jonathan Hart | 4cfd193 | 2013-11-19 16:42:25 -0800 | [diff] [blame] | 137 | removeDeviceImpl(deviceObject); |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 138 | |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 139 | try { |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 140 | ope.commit(); |
Jonathan Hart | 4cfd193 | 2013-11-19 16:42:25 -0800 | [diff] [blame] | 141 | log.debug("DeviceStorage:removeDevice mac:{} done", deviceMac); |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 142 | } catch (TitanException e) { |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 143 | ope.rollback(); |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 144 | log.error("DeviceStorage:removeDevice mac:{} failed", deviceMac); |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 145 | } |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 146 | } |
Jonathan Hart | 4cfd193 | 2013-11-19 16:42:25 -0800 | [diff] [blame] | 147 | |
| 148 | public void removeDeviceImpl(IDeviceObject deviceObject) { |
| 149 | for (IIpv4Address ipv4AddressVertex : deviceObject.getIpv4Addresses()) { |
| 150 | ope.removeIpv4Address(ipv4AddressVertex); |
| 151 | } |
| 152 | |
| 153 | ope.removeDevice(deviceObject); |
| 154 | } |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 155 | |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 156 | /*** |
| 157 | * This function is for getting the Device from the DB by Mac address of the device. |
| 158 | * @param mac The device mac address you want to get from the DB. |
| 159 | * @return IDeviceObject you want to get. |
| 160 | */ |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 161 | @Override |
| 162 | public IDeviceObject getDeviceByMac(String mac) { |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 163 | return ope.searchDevice(mac); |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 164 | } |
| 165 | |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 166 | /*** |
| 167 | * This function is for getting the Device from the DB by IP address of the device. |
| 168 | * @param ip The device ip address you want to get from the DB. |
| 169 | * @return IDeviceObject you want to get. |
| 170 | */ |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 171 | @Override |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 172 | public IDeviceObject getDeviceByIP(int ipv4Address) { |
| 173 | try { |
| 174 | IIpv4Address ipv4AddressVertex = ope.searchIpv4Address(ipv4Address); |
| 175 | if (ipv4AddressVertex != null) { |
| 176 | return ipv4AddressVertex.getDevice(); |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 177 | } |
| 178 | return null; |
| 179 | } |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 180 | catch (TitanException e) { |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 181 | log.error("DeviceStorage:getDeviceByIP:{} failed"); |
| 182 | return null; |
| 183 | } |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 184 | } |
| 185 | |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 186 | /*** |
| 187 | * This function is for changing the Device attachment point. |
| 188 | * @param device The device you want change the attachment point |
| 189 | */ |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 190 | @Override |
| 191 | public void changeDeviceAttachments(IDevice device) { |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 192 | IDeviceObject obj = null; |
| 193 | try { |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 194 | if ((obj = ope.searchDevice(device.getMACAddressString())) != null) { |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 195 | log.debug("Changing device ports {}: found existing device", device.getMACAddressString()); |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 196 | changeDeviceAttachments(device, obj); |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 197 | ope.commit(); |
| 198 | } else { |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 199 | log.debug("failed to search device...now adding {}", device.getMACAddressString()); |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 200 | addDevice(device); |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 201 | } |
| 202 | } catch (TitanException e) { |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 203 | ope.rollback(); |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 204 | log.error(":addDevice mac:{} failed", device.getMACAddressString()); |
| 205 | } |
| 206 | } |
| 207 | |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 208 | /*** |
| 209 | * This function is for changing the Device attachment point. |
| 210 | * @param device The new device you want change the attachment point |
| 211 | * @param obj The old device IDeviceObject that is going to change the attachment point. |
| 212 | */ |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 213 | public void changeDeviceAttachments(IDevice device, IDeviceObject obj) { |
| 214 | SwitchPort[] attachmentPoints = device.getAttachmentPoints(); |
| 215 | List<IPortObject> attachedPorts = Lists.newArrayList(obj.getAttachedPorts()); |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 216 | |
| 217 | for (SwitchPort ap : attachmentPoints) { |
| 218 | //Check if there is the port |
| 219 | IPortObject port = ope.searchPort(HexString.toHexString(ap.getSwitchDPID()), |
| 220 | (short) ap.getPort()); |
| 221 | log.debug("New Switch Port is {},{}", |
| 222 | HexString.toHexString(ap.getSwitchDPID()), (short) ap.getPort()); |
| 223 | |
| 224 | if (port != null){ |
| 225 | if (attachedPorts.contains(port)) { |
| 226 | log.debug("This is the port you already attached {}: do nothing", device.getMACAddressString()); |
| 227 | //This port will be remained, so remove from the removed port lists. |
| 228 | attachedPorts.remove(port); |
| 229 | } else { |
| 230 | log.debug("Adding device {}: attaching to port", device.getMACAddressString()); |
| 231 | port.setDevice(obj); |
| 232 | } |
| 233 | |
Yuta HIGUCHI | 5302ddf | 2014-01-06 12:53:35 -0800 | [diff] [blame] | 234 | log.debug("port number is {}", port.getNumber()); |
| 235 | log.debug("port desc is {}", port.getDesc()); |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 236 | } |
| 237 | } |
| 238 | |
| 239 | for (IPortObject port: attachedPorts) { |
| 240 | log.debug("Detaching the device {}: detaching from port", device.getMACAddressString()); |
| 241 | port.removeDevice(obj); |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 242 | |
| 243 | if (!obj.getAttachedPorts().iterator().hasNext()) { |
| 244 | // XXX If there are no more ports attached to the device, |
| 245 | // delete it. Otherwise we have a situation where the |
| 246 | // device remains forever with an IP address attached. |
| 247 | // When we implement device probing we should get rid of this. |
| 248 | removeDevice(obj); |
| 249 | } |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 250 | } |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 251 | } |
| 252 | |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 253 | /*** |
| 254 | * This function is for changing the Device IPv4 address. |
| 255 | * @param device The new device you want change the ipaddress |
| 256 | */ |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 257 | @Override |
| 258 | public void changeDeviceIPv4Address(IDevice device) { |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 259 | log.debug("Changing IP address for {} to {}", device.getMACAddressString(), |
| 260 | device.getIPv4Addresses()); |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 261 | IDeviceObject obj; |
| 262 | try { |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 263 | if ((obj = ope.searchDevice(device.getMACAddressString())) != null) { |
Jonathan Hart | c8ae36c | 2013-11-03 19:06:40 -0800 | [diff] [blame] | 264 | changeDeviceIpv4Addresses(device, obj); |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 265 | |
| 266 | ope.commit(); |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 267 | } else { |
| 268 | log.error(":changeDeviceIPv4Address mac:{} failed", device.getMACAddressString()); |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 269 | } |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 270 | } catch (TitanException e) { |
Teru | 87cd2da | 2013-06-15 20:33:08 -0700 | [diff] [blame] | 271 | ope.rollback(); |
Jonathan Hart | d6ed62b | 2013-11-01 13:18:25 -0700 | [diff] [blame] | 272 | log.error(":changeDeviceIPv4Address mac:{} failed due to exception {}", device.getMACAddressString(), e); |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 273 | } |
| 274 | } |
Jonathan Hart | c8ae36c | 2013-11-03 19:06:40 -0800 | [diff] [blame] | 275 | |
| 276 | private void changeDeviceIpv4Addresses(IDevice device, IDeviceObject deviceObject) { |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 277 | List<String> dbIpv4Addresses = new ArrayList<String>(); |
Jonathan Hart | ba9ced9 | 2013-11-24 16:52:13 -0800 | [diff] [blame] | 278 | List<Integer> intDbIpv4Addresses = new ArrayList<Integer>(); |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 279 | for (IIpv4Address ipv4Vertex : deviceObject.getIpv4Addresses()) { |
| 280 | dbIpv4Addresses.add(InetAddresses.fromInteger(ipv4Vertex.getIpv4Address()).getHostAddress()); |
Jonathan Hart | ba9ced9 | 2013-11-24 16:52:13 -0800 | [diff] [blame] | 281 | intDbIpv4Addresses.add(ipv4Vertex.getIpv4Address()); |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 282 | } |
| 283 | |
| 284 | List<String> memIpv4Addresses = new ArrayList<String>(); |
| 285 | for (int addr : device.getIPv4Addresses()) { |
| 286 | memIpv4Addresses.add(InetAddresses.fromInteger(addr).getHostAddress()); |
| 287 | } |
| 288 | |
| 289 | log.debug("Device IP addresses {}, database IP addresses {}", |
| 290 | memIpv4Addresses, dbIpv4Addresses); |
| 291 | |
Jonathan Hart | c8ae36c | 2013-11-03 19:06:40 -0800 | [diff] [blame] | 292 | for (int ipv4Address : device.getIPv4Addresses()) { |
Jonathan Hart | ba9ced9 | 2013-11-24 16:52:13 -0800 | [diff] [blame] | 293 | //if (deviceObject.getIpv4Address(ipv4Address) == null) { |
| 294 | if (!intDbIpv4Addresses.contains(ipv4Address)) { |
Jonathan Hart | c8ae36c | 2013-11-03 19:06:40 -0800 | [diff] [blame] | 295 | IIpv4Address dbIpv4Address = ope.ensureIpv4Address(ipv4Address); |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 296 | |
Jonathan Hart | ba9ced9 | 2013-11-24 16:52:13 -0800 | [diff] [blame] | 297 | /* |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 298 | IDeviceObject oldDevice = dbIpv4Address.getDevice(); |
| 299 | if (oldDevice != null) { |
| 300 | oldDevice.removeIpv4Address(dbIpv4Address); |
| 301 | } |
Jonathan Hart | ba9ced9 | 2013-11-24 16:52:13 -0800 | [diff] [blame] | 302 | */ |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 303 | |
| 304 | log.debug("Adding IP address {}", |
| 305 | InetAddresses.fromInteger(ipv4Address).getHostAddress()); |
Jonathan Hart | c8ae36c | 2013-11-03 19:06:40 -0800 | [diff] [blame] | 306 | deviceObject.addIpv4Address(dbIpv4Address); |
| 307 | } |
| 308 | } |
| 309 | |
| 310 | List<Integer> deviceIpv4Addresses = Arrays.asList(device.getIPv4Addresses()); |
| 311 | for (IIpv4Address dbIpv4Address : deviceObject.getIpv4Addresses()) { |
| 312 | if (!deviceIpv4Addresses.contains(dbIpv4Address.getIpv4Address())) { |
Jonathan Hart | 1a6f1d6 | 2013-11-14 11:33:46 -0800 | [diff] [blame] | 313 | log.debug("Removing IP address {}", |
| 314 | InetAddresses.fromInteger(dbIpv4Address.getIpv4Address()) |
| 315 | .getHostAddress()); |
Jonathan Hart | c8ae36c | 2013-11-03 19:06:40 -0800 | [diff] [blame] | 316 | deviceObject.removeIpv4Address(dbIpv4Address); |
| 317 | } |
| 318 | } |
| 319 | } |
Jonathan Hart | d857ad6 | 2013-12-14 18:08:17 -0800 | [diff] [blame] | 320 | |
| 321 | /** |
| 322 | * Takes an {@link OnosDevice} and adds it into the database. There is no |
| 323 | * checking of the database data and removing old data - an |
| 324 | * {@link OnosDevice} basically corresponds to a packet we've just seen, |
| 325 | * and we need to add that information into the database. |
| 326 | */ |
| 327 | @Override |
| 328 | public void addOnosDevice(OnosDevice onosDevice) { |
Jonathan Hart | d857ad6 | 2013-12-14 18:08:17 -0800 | [diff] [blame] | 329 | String macAddress = HexString.toHexString(onosDevice.getMacAddress().toBytes()); |
| 330 | |
pingping-lin | 0092603 | 2013-12-18 12:13:08 +0800 | [diff] [blame] | 331 | //if the switch port we try to attach a new device already has a link, then stop adding device |
| 332 | IPortObject portObject1 = ope.searchPort(HexString.toHexString( |
| 333 | onosDevice.getSwitchDPID()), onosDevice.getSwitchPort()); |
| 334 | |
Pavlin Radoslavov | 7807328 | 2013-12-19 17:01:23 -0800 | [diff] [blame] | 335 | if ((portObject1 != null) && portObject1.getLinkedPorts().iterator().hasNext()) { |
Yuta HIGUCHI | 5302ddf | 2014-01-06 12:53:35 -0800 | [diff] [blame] | 336 | if (log.isDebugEnabled()) { |
| 337 | log.debug("stop adding OnosDevice: {} due to there is a link to: {}", |
| 338 | onosDevice, portObject1.getLinkedPorts().iterator().next().getPortId()); |
| 339 | } |
pingping-lin | 0092603 | 2013-12-18 12:13:08 +0800 | [diff] [blame] | 340 | return; |
| 341 | } |
Jonathan Hart | 18ad950 | 2013-12-15 18:28:00 -0800 | [diff] [blame] | 342 | |
pingping-lin | 0092603 | 2013-12-18 12:13:08 +0800 | [diff] [blame] | 343 | log.debug("addOnosDevice: {}", onosDevice); |
| 344 | |
Jonathan Hart | d857ad6 | 2013-12-14 18:08:17 -0800 | [diff] [blame] | 345 | try { |
| 346 | IDeviceObject device = ope.searchDevice(macAddress); |
| 347 | |
| 348 | if (device == null) { |
| 349 | device = ope.newDevice(); |
| 350 | device.setType("device"); |
| 351 | device.setState("ACTIVE"); |
| 352 | device.setMACAddress(macAddress); |
| 353 | } |
| 354 | |
| 355 | // Check if the device has the IP address, add it if it doesn't |
Jonathan Hart | d857ad6 | 2013-12-14 18:08:17 -0800 | [diff] [blame] | 356 | if (onosDevice.getIpv4Address() != null) { |
| 357 | boolean hasIpAddress = false; |
| 358 | for (IIpv4Address ipv4Address : device.getIpv4Addresses()) { |
| 359 | if (ipv4Address.getIpv4Address() == onosDevice.getIpv4Address().intValue()) { |
| 360 | hasIpAddress = true; |
| 361 | break; |
| 362 | } |
| 363 | } |
| 364 | |
| 365 | if (!hasIpAddress) { |
| 366 | IIpv4Address ipv4Address = ope.ensureIpv4Address(onosDevice.getIpv4Address().intValue()); |
| 367 | IDeviceObject oldDevice = ipv4Address.getDevice(); |
| 368 | if (oldDevice != null && oldDevice.getMACAddress() != macAddress) { |
| 369 | oldDevice.removeIpv4Address(ipv4Address); |
| 370 | } |
| 371 | device.addIpv4Address(ipv4Address); |
| 372 | } |
| 373 | } |
| 374 | |
| 375 | // Check if the device has the attachment point, add it if not |
Jonathan Hart | 18ad950 | 2013-12-15 18:28:00 -0800 | [diff] [blame] | 376 | // TODO single attachment point for now, extend to multiple later |
Jonathan Hart | d857ad6 | 2013-12-14 18:08:17 -0800 | [diff] [blame] | 377 | String switchDpid = HexString.toHexString(onosDevice.getSwitchDPID()); |
| 378 | boolean hasAttachmentPoint = false; |
Jonathan Hart | 18ad950 | 2013-12-15 18:28:00 -0800 | [diff] [blame] | 379 | Iterator<IPortObject> it = device.getAttachedPorts().iterator(); |
| 380 | if (it.hasNext()) { |
| 381 | IPortObject existingPort = it.next(); |
| 382 | if (existingPort != null) { |
| 383 | ISwitchObject existingSwitch = existingPort.getSwitch(); |
| 384 | if (!existingSwitch.getDPID().equals(switchDpid) || |
| 385 | existingPort.getNumber() != onosDevice.getSwitchPort()) { |
| 386 | existingPort.removeDevice(device); |
| 387 | } |
| 388 | else { |
| 389 | hasAttachmentPoint = true; |
| 390 | } |
| 391 | } |
| 392 | } |
| 393 | |
| 394 | /* |
Jonathan Hart | d857ad6 | 2013-12-14 18:08:17 -0800 | [diff] [blame] | 395 | for (IPortObject portObject : device.getAttachedPorts()) { |
| 396 | ISwitchObject switchObject = portObject.getSwitch(); |
| 397 | if (switchObject.getDPID().equals(switchDpid) |
| 398 | && portObject.getNumber() == onosDevice.getSwitchPort()) { |
| 399 | hasAttachmentPoint = true; |
| 400 | break; |
| 401 | } |
| 402 | } |
Jonathan Hart | 18ad950 | 2013-12-15 18:28:00 -0800 | [diff] [blame] | 403 | */ |
Jonathan Hart | d857ad6 | 2013-12-14 18:08:17 -0800 | [diff] [blame] | 404 | |
| 405 | if (!hasAttachmentPoint) { |
| 406 | IPortObject portObject = ope.searchPort(switchDpid, onosDevice.getSwitchPort()); |
Jonathan Hart | 18ad950 | 2013-12-15 18:28:00 -0800 | [diff] [blame] | 407 | if (portObject != null) { |
| 408 | portObject.setDevice(device); |
| 409 | } |
Jonathan Hart | d857ad6 | 2013-12-14 18:08:17 -0800 | [diff] [blame] | 410 | } |
| 411 | |
| 412 | ope.commit(); |
| 413 | } |
| 414 | catch (TitanException e) { |
| 415 | log.error("addOnosDevice {} failed:", macAddress, e); |
| 416 | ope.rollback(); |
| 417 | } |
| 418 | } |
Pankaj Berde | da80957 | 2013-02-22 15:31:20 -0800 | [diff] [blame] | 419 | |
Pankaj Berde | b603134 | 2013-02-19 18:51:51 -0800 | [diff] [blame] | 420 | } |