admin-cli/executor/server_info.go

/*
 * 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 executor

import (
	"github.com/apache/incubator-pegasus/admin-cli/client"
	"github.com/apache/incubator-pegasus/admin-cli/tabular"
	"github.com/apache/incubator-pegasus/admin-cli/util"
	"github.com/apache/incubator-pegasus/go-client/session"
	"github.com/olekukonko/tablewriter"
)

// ServerInfo command
func ServerInfo(c *Client) error {

	nodes := c.Nodes.GetAllNodes(session.NodeTypeMeta)
	nodes = append(nodes, c.Nodes.GetAllNodes(session.NodeTypeReplica)...)

	results := client.BatchCallCmd(nodes, "server-info", []string{})

	type serverInfoStruct struct {
		Server  string `json:"server"`
		Node    string `json:"node"`
		Version string `json:"version"`
	}
	// always print meta-server first.
	var metaList []interface{}
	var replicaList []interface{}
	for n, result := range results {
		if n.Type == session.NodeTypeMeta {
			metaList = append(metaList, serverInfoStruct{
				Server:  string(n.Type),
				Node:    n.CombinedAddr(),
				Version: result.String(),
			})
		} else if n.Type == session.NodeTypeReplica {
			replicaList = append(replicaList, serverInfoStruct{
				Server:  string(n.Type),
				Node:    n.CombinedAddr(),
				Version: result.String(),
			})
		}
	}
	util.SortStructsByField(metaList, "Node")
	util.SortStructsByField(replicaList, "Node")
	valueList := append(metaList, replicaList...)
	tabular.New(c, valueList, func(table *tablewriter.Table) {
		table.SetAutoWrapText(false)
	}).Render()
	return nil
}