dev
This commit is contained in:
commit
27da26336f
|
|
@ -0,0 +1,153 @@
|
||||||
|
package cmd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"os/exec"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"git.apinb.com/bsm-sdk/engine/env"
|
||||||
|
"git.apinb.com/bsm-sdk/engine/utils"
|
||||||
|
"git.apinb.com/bsm-sdk/engine/vars"
|
||||||
|
"github.com/modood/table"
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ActionsReleases struct {
|
||||||
|
gorm.Model
|
||||||
|
Identity string
|
||||||
|
Origin string
|
||||||
|
ServiceKey string
|
||||||
|
Version string
|
||||||
|
FullName string
|
||||||
|
OssUrl string
|
||||||
|
EtcPath string
|
||||||
|
FilePath string
|
||||||
|
ImagePath string
|
||||||
|
}
|
||||||
|
|
||||||
|
type PsTable struct {
|
||||||
|
ServiceKey string
|
||||||
|
Origin string
|
||||||
|
Status string
|
||||||
|
Performance string
|
||||||
|
CurrentVersion string
|
||||||
|
RegisterVersion string
|
||||||
|
UpdatedAt string
|
||||||
|
}
|
||||||
|
|
||||||
|
var psCmd = &cobra.Command{
|
||||||
|
Use: "ps",
|
||||||
|
Short: "list restiry and local installed microservice.",
|
||||||
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
|
psExecute()
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
func psExecute() {
|
||||||
|
pt := make([]PsTable, 0)
|
||||||
|
body, err := utils.HttpGet(registryUrl)
|
||||||
|
checkError(err)
|
||||||
|
|
||||||
|
var ms []ActionsReleases
|
||||||
|
err = json.Unmarshal(body, &ms)
|
||||||
|
checkError(err)
|
||||||
|
|
||||||
|
for _, v := range ms {
|
||||||
|
srvLine := getSrvStatus(v.ServiceKey)
|
||||||
|
srvLine.ServiceKey = v.ServiceKey
|
||||||
|
srvLine.Origin = v.Origin
|
||||||
|
srvLine.RegisterVersion = v.Version
|
||||||
|
srvLine.UpdatedAt = v.UpdatedAt.Format(vars.YYYY_MM_DD_HH_MM_SS)
|
||||||
|
pt = append(pt, srvLine)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// Output to stdout
|
||||||
|
table.Output(pt)
|
||||||
|
}
|
||||||
|
|
||||||
|
func getSrvStatus(srv string) PsTable {
|
||||||
|
isRun := checkProcessRunning(srv)
|
||||||
|
var status string = " - "
|
||||||
|
if isRun {
|
||||||
|
status = "Running"
|
||||||
|
}
|
||||||
|
|
||||||
|
return PsTable{
|
||||||
|
Status: status,
|
||||||
|
CurrentVersion: getCurrentVersion(srv),
|
||||||
|
Performance: getProcessInfo(srv),
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func getPerformance(srv string) string {
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func getCurrentVersion(srv string) string {
|
||||||
|
var data map[string]string
|
||||||
|
|
||||||
|
cmd := exec.Command(env.MeshEnv.Prefix+srv, "--json") // 替换为适合操作系统的命令
|
||||||
|
output, err := cmd.Output()
|
||||||
|
if err != nil {
|
||||||
|
return " - "
|
||||||
|
}
|
||||||
|
|
||||||
|
err = json.Unmarshal(output, &data)
|
||||||
|
if err != nil {
|
||||||
|
return " - "
|
||||||
|
}
|
||||||
|
|
||||||
|
if val, ok := data["version"]; ok {
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
|
||||||
|
return " - "
|
||||||
|
}
|
||||||
|
|
||||||
|
func getProcessInfo(processName string) string {
|
||||||
|
var info string = " - "
|
||||||
|
cmd := exec.Command("ps", "-eo", "pid,%cpu,%mem") // 使用ps命令查询进程信息
|
||||||
|
output, err := cmd.Output() // 获取命令输出结果
|
||||||
|
if err != nil {
|
||||||
|
return info
|
||||||
|
}
|
||||||
|
|
||||||
|
// 将输出按换行符分隔成多行字符串数组
|
||||||
|
lines := strings.Split(string(output), "\n")[1:]
|
||||||
|
|
||||||
|
for _, line := range lines {
|
||||||
|
fields := strings.Fields(line) // 将每行按空格分隔为字段数组
|
||||||
|
if len(fields) >= 3 && fields[2] == processName {
|
||||||
|
pid := fields[0] // PID（进程标识）
|
||||||
|
cpuUsage := fields[1] // CPU使用情况
|
||||||
|
memoryUsage := fields[2] // 内存使用情况
|
||||||
|
info = fmt.Sprintf("PID:%s / CPU:%s / MEM:%s", pid, cpuUsage, memoryUsage)
|
||||||
|
return info
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return info
|
||||||
|
}
|
||||||
|
|
||||||
|
func checkProcessRunning(processName string) bool {
|
||||||
|
cmd := exec.Command("ps", "-ef") // 替换为适合操作系统的命令
|
||||||
|
output, err := cmd.Output()
|
||||||
|
if err != nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// 将输出按行分割成切片
|
||||||
|
lines := strings.Split(string(output), "\n")
|
||||||
|
|
||||||
|
for _, line := range lines {
|
||||||
|
if strings.Contains(line, processName) {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,48 @@
|
||||||
|
package cmd
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"git.apinb.com/bsm-sdk/engine/env"
|
||||||
|
"github.com/spf13/cobra"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
registryUrl string
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
// get os env.
|
||||||
|
|
||||||
|
psCmd.Flags().StringVarP(®istryUrl, "registry", "r", "", "registry url.")
|
||||||
|
|
||||||
|
if registryUrl == "" {
|
||||||
|
registryUrl = env.GetEnvDefault("BlocksMesh_Registry", "http://registry.apinb.com")
|
||||||
|
}
|
||||||
|
|
||||||
|
rootCmd.AddCommand(psCmd)
|
||||||
|
}
|
||||||
|
|
||||||
|
var rootCmd = &cobra.Command{
|
||||||
|
Use: "bsm",
|
||||||
|
Short: "bsm is a Blocks Service Mesh cli.",
|
||||||
|
Run: func(cmd *cobra.Command, args []string) {
|
||||||
|
fmt.Println("usage: bsm <command>")
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
func checkError(err error) {
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func Execute() {
|
||||||
|
if err := rootCmd.Execute(); err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1 @@
|
||||||
|
package cmd
|
||||||
|
|
@ -0,0 +1,20 @@
|
||||||
|
module git.apinb.com/bsm-tools/bsm
|
||||||
|
|
||||||
|
go 1.22.0
|
||||||
|
|
||||||
|
require git.apinb.com/bsm-sdk/engine v1.0.9
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/inconshreveable/mousetrap v1.1.0 // indirect
|
||||||
|
github.com/spf13/pflag v1.0.5 // indirect
|
||||||
|
)
|
||||||
|
|
||||||
|
require (
|
||||||
|
github.com/google/uuid v1.6.0 // indirect
|
||||||
|
github.com/jaevor/go-nanoid v1.3.0 // indirect
|
||||||
|
github.com/jinzhu/inflection v1.0.0 // indirect
|
||||||
|
github.com/jinzhu/now v1.1.5 // indirect
|
||||||
|
github.com/modood/table v0.0.0-20220527013332-8d47e76dad33
|
||||||
|
github.com/spf13/cobra v1.8.0
|
||||||
|
gorm.io/gorm v1.25.7 // indirect
|
||||||
|
)
|
||||||
|
|
@ -0,0 +1,24 @@
|
||||||
|
git.apinb.com/bsm-sdk/engine v1.0.9 h1:3zPYp8wtBVi7Aeg3H3/q7ZSVDRsGiXxQXwNzyMxfhao=
|
||||||
|
git.apinb.com/bsm-sdk/engine v1.0.9/go.mod h1:zCJfxj6RHHVHQmDn+Z2qa5s1qU7a3rTL9AZqEja9vIc=
|
||||||
|
github.com/cpuguy83/go-md2man/v2 v2.0.3/go.mod h1:tgQtvFlXSQOSOSIRvRPT7W67SCa46tRHOmNcaadrF8o=
|
||||||
|
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||||
|
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||||
|
github.com/inconshreveable/mousetrap v1.1.0 h1:wN+x4NVGpMsO7ErUn/mUI3vEoE6Jt13X2s0bqwp9tc8=
|
||||||
|
github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLfsEA9PFc4w1p2J65bw=
|
||||||
|
github.com/jaevor/go-nanoid v1.3.0 h1:nD+iepesZS6pr3uOVf20vR9GdGgJW1HPaR46gtrxzkg=
|
||||||
|
github.com/jaevor/go-nanoid v1.3.0/go.mod h1:SI+jFaPuddYkqkVQoNGHs81navCtH388TcrH0RqFKgY=
|
||||||
|
github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=
|
||||||
|
github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc=
|
||||||
|
github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ=
|
||||||
|
github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=
|
||||||
|
github.com/modood/table v0.0.0-20220527013332-8d47e76dad33 h1:T5IbS9C1G2zeHb6eBy6OfIvj5tfQB23kGFpewCJuGDg=
|
||||||
|
github.com/modood/table v0.0.0-20220527013332-8d47e76dad33/go.mod h1:41qyXVI5QH9/ObyPj27CGCVau5v/njfc3Gjj7yzr0HQ=
|
||||||
|
github.com/russross/blackfriday/v2 v2.1.0/go.mod h1:+Rmxgy9KzJVeS9/2gXHxylqXiyQDYRxCVz55jmeOWTM=
|
||||||
|
github.com/spf13/cobra v1.8.0 h1:7aJaZx1B85qltLMc546zn58BxxfZdR/W22ej9CFoEf0=
|
||||||
|
github.com/spf13/cobra v1.8.0/go.mod h1:WXLWApfZ71AjXPya3WOlMsY9yMs7YeiHhFVlvLyhcho=
|
||||||
|
github.com/spf13/pflag v1.0.5 h1:iy+VFUOCP1a+8yFto/drg2CJ5u0yRoB7fZw3DKv/JXA=
|
||||||
|
github.com/spf13/pflag v1.0.5/go.mod h1:McXfInJRrz4CZXVZOBLb0bTZqETkiAhM9Iw0y3An2Bg=
|
||||||
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||||
|
gorm.io/gorm v1.25.7 h1:VsD6acwRjz2zFxGO50gPO6AkNs7KKnvfzUjHQhZDz/A=
|
||||||
|
gorm.io/gorm v1.25.7/go.mod h1:hbnx/Oo0ChWMn1BIhpy1oYozzpM15i4YPuHDmfYtwg8=
|
||||||
Loading…
Reference in New Issue