mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-12-05 13:17:42 +08:00
184 lines
5.6 KiB
Go
184 lines
5.6 KiB
Go
// Copyright (C) 2014-2018 Goodrain Co., Ltd.
|
||
// RAINBOND, Application Management Platform
|
||
|
||
// This program is free software: you can redistribute it and/or modify
|
||
// it under the terms of the GNU General Public License as published by
|
||
// the Free Software Foundation, either version 3 of the License, or
|
||
// (at your option) any later version. For any non-GPL usage of Rainbond,
|
||
// one or multiple Commercial Licenses authorized by Goodrain Co., Ltd.
|
||
// must be obtained first.
|
||
|
||
// This program is distributed in the hope that it will be useful,
|
||
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
// GNU General Public License for more details.
|
||
|
||
// You should have received a copy of the GNU General Public License
|
||
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
||
package cmd
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
|
||
"github.com/Sirupsen/logrus"
|
||
"github.com/apcera/termtables"
|
||
"github.com/goodrain/rainbond/grctl/clients"
|
||
"github.com/gosuri/uitable"
|
||
"github.com/urfave/cli"
|
||
|
||
//"github.com/goodrain/rainbond/eventlog/conf"
|
||
"errors"
|
||
|
||
config "github.com/goodrain/rainbond/cmd/grctl/option"
|
||
)
|
||
|
||
//NewCmdTenant tenant cmd
|
||
func NewCmdTenant() cli.Command {
|
||
c := cli.Command{
|
||
Name: "tenant",
|
||
Usage: "grctl tenant -h",
|
||
Subcommands: []cli.Command{
|
||
cli.Command{
|
||
Name: "list",
|
||
Usage: "list all tenant info",
|
||
Action: func(c *cli.Context) error {
|
||
Common(c)
|
||
return getAllTenant(c)
|
||
},
|
||
},
|
||
cli.Command{
|
||
Name: "get",
|
||
Usage: "get all app details by specified tenant name",
|
||
Action: func(c *cli.Context) error {
|
||
Common(c)
|
||
return getTenantInfo(c)
|
||
},
|
||
},
|
||
cli.Command{
|
||
Name: "res",
|
||
Usage: "get tenant resource details by specified tenant name",
|
||
Action: func(c *cli.Context) error {
|
||
Common(c)
|
||
return findTenantResourceUsage(c)
|
||
},
|
||
},
|
||
cli.Command{
|
||
Name: "batchstop",
|
||
Usage: "batch stop app by specified tenant name",
|
||
Flags: []cli.Flag{
|
||
cli.BoolFlag{
|
||
Name: "f",
|
||
Usage: "Continuous log output",
|
||
},
|
||
cli.StringFlag{
|
||
Name: "event_log_server",
|
||
Usage: "event log server address",
|
||
},
|
||
},
|
||
Action: func(c *cli.Context) error {
|
||
Common(c)
|
||
return stopTenantService(c)
|
||
},
|
||
},
|
||
cli.Command{
|
||
Name: "setdefname",
|
||
Usage: "set default tenant name",
|
||
Action: func(c *cli.Context) error {
|
||
err := CreateTenantFile(c.Args().First())
|
||
if err != nil {
|
||
logrus.Error("set default tenantname fail", err.Error())
|
||
}
|
||
return nil
|
||
},
|
||
},
|
||
},
|
||
}
|
||
return c
|
||
}
|
||
|
||
// grctrl tenant TENANT_NAME
|
||
func getTenantInfo(c *cli.Context) error {
|
||
tenantID := c.Args().First()
|
||
if tenantID == "" {
|
||
fmt.Println("Please provide tenant name")
|
||
os.Exit(1)
|
||
}
|
||
services, err := clients.RegionClient.Tenants(tenantID).Services("").List()
|
||
handleErr(err)
|
||
if services != nil {
|
||
runtable := termtables.CreateTable()
|
||
closedtable := termtables.CreateTable()
|
||
runtable.AddHeaders("服务别名", "应用状态", "Deploy版本", "实例数量", "内存占用")
|
||
closedtable.AddHeaders("租户ID", "服务ID", "服务别名", "应用状态", "Deploy版本")
|
||
for _, service := range services {
|
||
if service.CurStatus != "closed" && service.CurStatus != "closing" && service.CurStatus != "undeploy" && service.CurStatus != "deploying" {
|
||
runtable.AddRow(service.ServiceAlias, service.CurStatus, service.DeployVersion, service.Replicas, fmt.Sprintf("%d Mb", service.ContainerMemory*service.Replicas))
|
||
} else {
|
||
closedtable.AddRow(service.TenantID, service.ServiceID, service.ServiceAlias, service.CurStatus, service.DeployVersion)
|
||
}
|
||
}
|
||
fmt.Println("运行中的应用:")
|
||
fmt.Println(runtable.Render())
|
||
fmt.Println("不在运行的应用:")
|
||
fmt.Println(closedtable.Render())
|
||
return nil
|
||
}
|
||
return nil
|
||
}
|
||
func findTenantResourceUsage(c *cli.Context) error {
|
||
tenantName := c.Args().First()
|
||
if tenantName == "" {
|
||
fmt.Println("Please provide tenant name")
|
||
os.Exit(1)
|
||
}
|
||
resources, err := clients.RegionClient.Resources().Tenants(tenantName).Get()
|
||
handleErr(err)
|
||
table := uitable.New()
|
||
table.Wrap = true // wrap columns
|
||
table.AddRow("租户名:", resources.Name)
|
||
table.AddRow("租户ID:", resources.UUID)
|
||
table.AddRow("企业ID:", resources.EID)
|
||
table.AddRow("正使用CPU资源:", fmt.Sprintf("%.2f Core", float64(resources.UsedCPU)/1000))
|
||
table.AddRow("正使用内存资源:", fmt.Sprintf("%d %s", resources.UsedMEM, "Mb"))
|
||
table.AddRow("正使用磁盘资源:", fmt.Sprintf("%.2f Mb", resources.UsedDisk/1024))
|
||
table.AddRow("总分配CPU资源:", fmt.Sprintf("%.2f Core", float64(resources.AllocatedCPU)/1000))
|
||
table.AddRow("总分配内存资源:", fmt.Sprintf("%d %s", resources.AllocatedMEM, "Mb"))
|
||
fmt.Println(table)
|
||
return nil
|
||
}
|
||
|
||
func getAllTenant(c *cli.Context) error {
|
||
tenants, err := clients.RegionClient.Tenants("").List()
|
||
handleErr(err)
|
||
tenantsTable := termtables.CreateTable()
|
||
tenantsTable.AddHeaders("TenantAlias", "TenantID", "TenantLimit")
|
||
for _, t := range tenants {
|
||
tenantsTable.AddRow(t.Name, t.UUID, fmt.Sprintf("%d GB", t.LimitMemory))
|
||
}
|
||
fmt.Print(tenantsTable.Render())
|
||
return nil
|
||
}
|
||
|
||
//CreateTenantFile Create Tenant File
|
||
func CreateTenantFile(tname string) error {
|
||
filename, err := config.GetTenantNamePath()
|
||
if err != nil {
|
||
logrus.Warn("Load config file error.")
|
||
return errors.New("Load config file error")
|
||
}
|
||
f, err := os.OpenFile(filename, os.O_WRONLY|os.O_TRUNC|os.O_CREATE, 0644)
|
||
if err != nil {
|
||
logrus.Warn("load teantnamefile file", err.Error())
|
||
f.Close()
|
||
return err
|
||
}
|
||
_, err = f.WriteString(tname)
|
||
if err != nil {
|
||
logrus.Warn("write teantnamefile file", err.Error())
|
||
}
|
||
f.Close()
|
||
return err
|
||
}
|