mirror of
https://gitee.com/rainbond/Rainbond.git
synced 2024-12-03 20:27:46 +08:00
128 lines
3.6 KiB
Go
128 lines
3.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"
|
|
"strings"
|
|
|
|
"github.com/apcera/termtables"
|
|
"github.com/goodrain/rainbond/grctl/clients"
|
|
"github.com/goodrain/rainbond/node/api/model"
|
|
"github.com/urfave/cli"
|
|
"github.com/goodrain/rainbond/node/nodem/service"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
//NewCmdConfigs 全局配置相关命令
|
|
func NewCmdConfigs() cli.Command {
|
|
c := cli.Command{
|
|
Name: "conf",
|
|
Usage: "集群和服务配置相关工具",
|
|
Subcommands: []cli.Command{
|
|
cli.Command{
|
|
Name: "gen",
|
|
Usage: "generate systemd configs by service list yaml file.",
|
|
Action: func(c *cli.Context) error {
|
|
yamlFile := c.Args().First()
|
|
services, err := service.LoadServicesFromLocal(yamlFile)
|
|
if err != nil {
|
|
println("Can not parse the yaml file: ", err.Error())
|
|
return err
|
|
}
|
|
|
|
dirName := filepath.Dir(yamlFile)
|
|
ext := filepath.Ext(yamlFile)
|
|
baseName := filepath.Base(yamlFile)
|
|
baseName = baseName[:len(baseName)-len(ext)]
|
|
dirName = fmt.Sprintf("%s/%s", dirName, baseName)
|
|
os.Mkdir(dirName, 0755)
|
|
|
|
for _, s := range services {
|
|
content := service.ToConfig(s)
|
|
configFile := fmt.Sprintf("%s/%s.service", dirName, s.Name)
|
|
err := ioutil.WriteFile(configFile, []byte(content), 0644)
|
|
if err != nil {
|
|
println("Can not write service to file: ", err.Error())
|
|
}
|
|
}
|
|
|
|
println("Successful to generate service in: ", dirName)
|
|
|
|
return nil
|
|
},
|
|
},
|
|
cli.Command{
|
|
Name: "get",
|
|
Usage: "get all datacenter configs",
|
|
Action: func(c *cli.Context) error {
|
|
configs, err := clients.RegionClient.Configs().Get()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
taskTable := termtables.CreateTable()
|
|
taskTable.AddHeaders("Name", "CNName", "ValueType", "Value")
|
|
for _, config := range configs.Configs {
|
|
taskTable.AddRow(config.Name, config.CNName, config.ValueType, config.Value)
|
|
}
|
|
fmt.Println(taskTable.Render())
|
|
return nil
|
|
},
|
|
},
|
|
cli.Command{
|
|
Name: "put",
|
|
Usage: "put database configs",
|
|
Action: func(c *cli.Context) error {
|
|
key := c.Args().Get(0)
|
|
value := c.Args().Get(1)
|
|
configs, err := clients.RegionClient.Configs().Get()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
gc := configs.Get(key)
|
|
if gc == nil {
|
|
gcnew := model.ConfigUnit{
|
|
Name: key,
|
|
}
|
|
configs.Add(gcnew)
|
|
gc = configs.Get(key)
|
|
}
|
|
if strings.Contains(value, ",") {
|
|
vas := strings.Split(value, ",")
|
|
gc.ValueType = "array"
|
|
gc.Value = vas
|
|
} else {
|
|
gc.ValueType = "string"
|
|
gc.Value = value
|
|
}
|
|
err = clients.RegionClient.Configs().Put(configs)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
fmt.Printf("configs %s put success \n", key)
|
|
return nil
|
|
},
|
|
},
|
|
},
|
|
}
|
|
return c
|
|
}
|