// 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 . 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 }