tom | 6d2a43e | 2014-09-08 01:50:20 -0700 | [diff] [blame] | 1 | package org.onlab.onos.cli.net; |
| 2 | |
tom | 32085cf | 2014-10-16 00:04:33 -0700 | [diff] [blame^] | 3 | import com.fasterxml.jackson.databind.JsonNode; |
| 4 | import com.fasterxml.jackson.databind.ObjectMapper; |
| 5 | import com.fasterxml.jackson.databind.node.ArrayNode; |
| 6 | import com.fasterxml.jackson.databind.node.ObjectNode; |
tom | 6d2a43e | 2014-09-08 01:50:20 -0700 | [diff] [blame] | 7 | import org.apache.karaf.shell.commands.Command; |
| 8 | import org.onlab.onos.cli.AbstractShellCommand; |
tom | 91c7bd0 | 2014-09-25 22:50:44 -0700 | [diff] [blame] | 9 | import org.onlab.onos.cli.Comparators; |
tom | 6d2a43e | 2014-09-08 01:50:20 -0700 | [diff] [blame] | 10 | import org.onlab.onos.net.Device; |
| 11 | import org.onlab.onos.net.device.DeviceService; |
| 12 | |
tom | d79f7ec | 2014-09-08 13:29:18 -0700 | [diff] [blame] | 13 | import java.util.Collections; |
tom | d79f7ec | 2014-09-08 13:29:18 -0700 | [diff] [blame] | 14 | import java.util.List; |
| 15 | |
| 16 | import static com.google.common.collect.Lists.newArrayList; |
| 17 | |
tom | 6d2a43e | 2014-09-08 01:50:20 -0700 | [diff] [blame] | 18 | /** |
| 19 | * Lists all infrastructure devices. |
| 20 | */ |
| 21 | @Command(scope = "onos", name = "devices", |
| 22 | description = "Lists all infrastructure devices") |
| 23 | public class DevicesListCommand extends AbstractShellCommand { |
| 24 | |
| 25 | private static final String FMT = |
tom | c290a12 | 2014-09-08 14:27:13 -0700 | [diff] [blame] | 26 | "id=%s, available=%s, role=%s, type=%s, mfr=%s, hw=%s, sw=%s, serial=%s"; |
tom | 6d2a43e | 2014-09-08 01:50:20 -0700 | [diff] [blame] | 27 | |
| 28 | @Override |
tom | 0872a17 | 2014-09-23 11:24:26 -0700 | [diff] [blame] | 29 | protected void execute() { |
tom | caf3bf7 | 2014-09-23 13:20:53 -0700 | [diff] [blame] | 30 | DeviceService service = get(DeviceService.class); |
tom | 32085cf | 2014-10-16 00:04:33 -0700 | [diff] [blame^] | 31 | if (outputJson()) { |
| 32 | print("%s", json(service, getSortedDevices(service))); |
| 33 | } else { |
| 34 | for (Device device : getSortedDevices(service)) { |
| 35 | printDevice(service, device); |
| 36 | } |
tom | 6d2a43e | 2014-09-08 01:50:20 -0700 | [diff] [blame] | 37 | } |
tom | 6d2a43e | 2014-09-08 01:50:20 -0700 | [diff] [blame] | 38 | } |
tom | ff7eb7c | 2014-09-08 12:49:03 -0700 | [diff] [blame] | 39 | |
| 40 | /** |
tom | 32085cf | 2014-10-16 00:04:33 -0700 | [diff] [blame^] | 41 | * Returns JSON node representing the specified devices. |
| 42 | * |
| 43 | * @param service device service |
| 44 | * @param devices collection of devices |
| 45 | * @return JSON node |
| 46 | */ |
| 47 | public static JsonNode json(DeviceService service, Iterable<Device> devices) { |
| 48 | ObjectMapper mapper = new ObjectMapper(); |
| 49 | ArrayNode result = mapper.createArrayNode(); |
| 50 | for (Device device : devices) { |
| 51 | result.add(json(service, mapper, device)); |
| 52 | } |
| 53 | return result; |
| 54 | } |
| 55 | |
| 56 | /** |
| 57 | * Returns JSON node representing the specified device. |
| 58 | * |
| 59 | * @param service device service |
| 60 | * @param mapper object mapper |
| 61 | * @param device infrastructure device |
| 62 | * @return JSON node |
| 63 | */ |
| 64 | public static ObjectNode json(DeviceService service, ObjectMapper mapper, |
| 65 | Device device) { |
| 66 | ObjectNode result = mapper.createObjectNode(); |
| 67 | if (device != null) { |
| 68 | result.put("id", device.id().toString()) |
| 69 | .put("available", service.isAvailable(device.id())) |
| 70 | .put("role", service.getRole(device.id()).toString()) |
| 71 | .put("mfr", device.manufacturer()) |
| 72 | .put("hw", device.hwVersion()) |
| 73 | .put("sw", device.swVersion()) |
| 74 | .put("serial", device.serialNumber()); |
| 75 | } |
| 76 | return result; |
| 77 | } |
| 78 | |
| 79 | /** |
tom | d79f7ec | 2014-09-08 13:29:18 -0700 | [diff] [blame] | 80 | * Returns the list of devices sorted using the device ID URIs. |
| 81 | * |
| 82 | * @param service device service |
| 83 | * @return sorted device list |
| 84 | */ |
tom | 53945d5 | 2014-10-07 11:01:36 -0700 | [diff] [blame] | 85 | protected static List<Device> getSortedDevices(DeviceService service) { |
tom | d79f7ec | 2014-09-08 13:29:18 -0700 | [diff] [blame] | 86 | List<Device> devices = newArrayList(service.getDevices()); |
tom | 1380eee | 2014-09-24 09:22:02 -0700 | [diff] [blame] | 87 | Collections.sort(devices, Comparators.ELEMENT_COMPARATOR); |
tom | d79f7ec | 2014-09-08 13:29:18 -0700 | [diff] [blame] | 88 | return devices; |
| 89 | } |
| 90 | |
| 91 | /** |
tom | ff7eb7c | 2014-09-08 12:49:03 -0700 | [diff] [blame] | 92 | * Prints information about the specified device. |
| 93 | * |
tom | c290a12 | 2014-09-08 14:27:13 -0700 | [diff] [blame] | 94 | * @param service device service |
| 95 | * @param device infrastructure device |
tom | ff7eb7c | 2014-09-08 12:49:03 -0700 | [diff] [blame] | 96 | */ |
tom | c290a12 | 2014-09-08 14:27:13 -0700 | [diff] [blame] | 97 | protected void printDevice(DeviceService service, Device device) { |
tom | 9eb57fb | 2014-09-11 19:42:38 -0700 | [diff] [blame] | 98 | if (device != null) { |
| 99 | print(FMT, device.id(), service.isAvailable(device.id()), |
| 100 | service.getRole(device.id()), device.type(), |
| 101 | device.manufacturer(), device.hwVersion(), device.swVersion(), |
| 102 | device.serialNumber()); |
| 103 | } |
tom | ff7eb7c | 2014-09-08 12:49:03 -0700 | [diff] [blame] | 104 | } |
| 105 | |
tom | 6d2a43e | 2014-09-08 01:50:20 -0700 | [diff] [blame] | 106 | } |