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