ClickHouse/programs/diagnostics/internal/platform/utils/process.go

50 lines
1.1 KiB
Go
Raw Normal View History

2022-04-27 12:22:20 +00:00
package utils
import (
"github.com/elastic/gosigar"
"strings"
)
func FindClickHouseProcesses() ([]gosigar.ProcArgs, error) {
pids := gosigar.ProcList{}
err := pids.Get()
if err != nil {
return nil, err
}
var clickhousePs []gosigar.ProcArgs
for _, pid := range pids.List {
args := gosigar.ProcArgs{}
if err := args.Get(pid); err != nil {
continue
}
if len(args.List) > 0 {
if strings.Contains(args.List[0], "clickhouse-server") {
clickhousePs = append(clickhousePs, args)
}
}
}
return clickhousePs, nil
}
func FindConfigsFromClickHouseProcesses() ([]string, error) {
clickhouseProcesses, err := FindClickHouseProcesses()
if err != nil {
return nil, err
}
var configs []string
if len(clickhouseProcesses) > 0 {
// we have candidate matches
for _, ps := range clickhouseProcesses {
for _, arg := range ps.List {
if strings.Contains(arg, "--config") {
configFile := strings.ReplaceAll(arg, "--config-file=", "")
// containers receive config with --config
configFile = strings.ReplaceAll(configFile, "--config=", "")
configs = append(configs, configFile)
}
}
}
}
return configs, err
}