| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| package org.onlab.onos.cli; |
| |
| import static com.google.common.collect.Lists.newArrayList; |
| |
| import java.util.Collections; |
| import java.util.List; |
| |
| import org.apache.karaf.shell.commands.Command; |
| import org.onlab.onos.cluster.NodeId; |
| import org.onlab.onos.cluster.RoleInfo; |
| import org.onlab.onos.mastership.MastershipService; |
| import org.onlab.onos.net.Device; |
| import org.onlab.onos.net.DeviceId; |
| import org.onlab.onos.net.device.DeviceService; |
| |
| |
| /** |
| * Lists mastership roles of nodes for each device. |
| */ |
| @Command(scope = "onos", name = "roles", |
| description = "Lists mastership roles of nodes for each device.") |
| public class RolesCommand extends AbstractShellCommand { |
| |
| private static final String FMT_HDR = "%s: master=%s, standbys=[ %s]"; |
| |
| @Override |
| protected void execute() { |
| DeviceService deviceService = get(DeviceService.class); |
| MastershipService roleService = get(MastershipService.class); |
| |
| for (Device d : getSortedDevices(deviceService)) { |
| DeviceId did = d.id(); |
| printRoles(roleService, did); |
| } |
| } |
| |
| /** |
| * Returns the list of devices sorted using the device ID URIs. |
| * |
| * @param service device service |
| * @return sorted device list |
| */ |
| protected static List<Device> getSortedDevices(DeviceService service) { |
| List<Device> devices = newArrayList(service.getDevices()); |
| Collections.sort(devices, Comparators.ELEMENT_COMPARATOR); |
| return devices; |
| } |
| |
| /** |
| * Prints the role information for a device. |
| * |
| * @param deviceId the ID of the device |
| * @param master the current master |
| */ |
| protected void printRoles(MastershipService service, DeviceId deviceId) { |
| RoleInfo nodes = service.getNodesFor(deviceId); |
| StringBuilder builder = new StringBuilder(); |
| for (NodeId nid : nodes.backups()) { |
| builder.append(nid).append(" "); |
| } |
| |
| print(FMT_HDR, deviceId, |
| nodes.master() == null ? "NONE" : nodes.master(), |
| builder.toString()); |
| } |
| } |