tom | e472987 | 2014-09-23 00:37:37 -0700 | [diff] [blame] | 1 | package org.onlab.onos.cli; |
| 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; |
tom | e472987 | 2014-09-23 00:37:37 -0700 | [diff] [blame] | 6 | import org.apache.karaf.shell.commands.Command; |
| 7 | import org.onlab.onos.cluster.ClusterService; |
| 8 | import org.onlab.onos.cluster.ControllerNode; |
| 9 | |
| 10 | import java.util.Collections; |
tom | e472987 | 2014-09-23 00:37:37 -0700 | [diff] [blame] | 11 | import java.util.List; |
| 12 | |
| 13 | import static com.google.common.collect.Lists.newArrayList; |
| 14 | |
| 15 | /** |
| 16 | * Lists all controller cluster nodes. |
| 17 | */ |
| 18 | @Command(scope = "onos", name = "nodes", |
| 19 | description = "Lists all controller cluster nodes") |
| 20 | public class NodesListCommand extends AbstractShellCommand { |
| 21 | |
| 22 | private static final String FMT = |
tom | ee49c37 | 2014-09-26 15:14:50 -0700 | [diff] [blame] | 23 | "id=%s, address=%s:%s, state=%s %s"; |
tom | e472987 | 2014-09-23 00:37:37 -0700 | [diff] [blame] | 24 | |
tom | e472987 | 2014-09-23 00:37:37 -0700 | [diff] [blame] | 25 | @Override |
tom | 0872a17 | 2014-09-23 11:24:26 -0700 | [diff] [blame] | 26 | protected void execute() { |
| 27 | ClusterService service = get(ClusterService.class); |
tom | e472987 | 2014-09-23 00:37:37 -0700 | [diff] [blame] | 28 | List<ControllerNode> nodes = newArrayList(service.getNodes()); |
tom | 1380eee | 2014-09-24 09:22:02 -0700 | [diff] [blame] | 29 | Collections.sort(nodes, Comparators.NODE_COMPARATOR); |
tom | 32085cf | 2014-10-16 00:04:33 -0700 | [diff] [blame^] | 30 | if (outputJson()) { |
| 31 | print("%s", json(service, nodes)); |
| 32 | } else { |
| 33 | ControllerNode self = service.getLocalNode(); |
| 34 | for (ControllerNode node : nodes) { |
| 35 | print(FMT, node.id(), node.ip(), node.tcpPort(), |
| 36 | service.getState(node.id()), |
| 37 | node.equals(self) ? "*" : ""); |
| 38 | } |
| 39 | } |
| 40 | } |
| 41 | |
| 42 | // Produces JSON structure. |
| 43 | private JsonNode json(ClusterService service, List<ControllerNode> nodes) { |
| 44 | ObjectMapper mapper = new ObjectMapper(); |
| 45 | ArrayNode result = mapper.createArrayNode(); |
tom | e472987 | 2014-09-23 00:37:37 -0700 | [diff] [blame] | 46 | ControllerNode self = service.getLocalNode(); |
| 47 | for (ControllerNode node : nodes) { |
tom | 32085cf | 2014-10-16 00:04:33 -0700 | [diff] [blame^] | 48 | result.add(mapper.createObjectNode() |
| 49 | .put("id", node.id().toString()) |
| 50 | .put("ip", node.ip().toString()) |
| 51 | .put("tcpPort", node.tcpPort()) |
| 52 | .put("state", service.getState(node.id()).toString()) |
| 53 | .put("self", node.equals(self))); |
tom | e472987 | 2014-09-23 00:37:37 -0700 | [diff] [blame] | 54 | } |
tom | 32085cf | 2014-10-16 00:04:33 -0700 | [diff] [blame^] | 55 | return result; |
tom | e472987 | 2014-09-23 00:37:37 -0700 | [diff] [blame] | 56 | } |
| 57 | |
| 58 | } |