mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-22 16:36:27 +01:00
9f94c295ab
The ioutil.{Read|Write}File is deprecated since Go1.16 -
see https://tip.golang.org/doc/go1.16#ioutil
VictoriaMetrics needs at least Go1.18, so it is safe to remove ioutil usage
from source code.
This is a follow-up for 02ca2342ab
60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
package cgroup
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
func getStatGeneric(statName, sysfsPrefix, cgroupPath, cgroupGrepLine string) (int64, error) {
|
|
data, err := getFileContents(statName, sysfsPrefix, cgroupPath, cgroupGrepLine)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
data = strings.TrimSpace(data)
|
|
n, err := strconv.ParseInt(data, 10, 64)
|
|
if err != nil {
|
|
return 0, fmt.Errorf("cannot parse %q: %w", cgroupPath, err)
|
|
}
|
|
return n, nil
|
|
}
|
|
|
|
func getFileContents(statName, sysfsPrefix, cgroupPath, cgroupGrepLine string) (string, error) {
|
|
filepath := path.Join(sysfsPrefix, statName)
|
|
data, err := os.ReadFile(filepath)
|
|
if err == nil {
|
|
return string(data), nil
|
|
}
|
|
cgroupData, err := os.ReadFile(cgroupPath)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
subPath, err := grepFirstMatch(string(cgroupData), cgroupGrepLine, 2, ":")
|
|
if err != nil {
|
|
return "", fmt.Errorf("cannot find cgroup path for %q in %q: %w", cgroupGrepLine, cgroupPath, err)
|
|
}
|
|
filepath = path.Join(sysfsPrefix, subPath, statName)
|
|
data, err = os.ReadFile(filepath)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(data), nil
|
|
}
|
|
|
|
// grepFirstMatch searches match line at data and returns item from it by index with given delimiter.
|
|
func grepFirstMatch(data string, match string, index int, delimiter string) (string, error) {
|
|
lines := strings.Split(string(data), "\n")
|
|
for _, s := range lines {
|
|
if !strings.Contains(s, match) {
|
|
continue
|
|
}
|
|
parts := strings.Split(s, delimiter)
|
|
if index < len(parts) {
|
|
return strings.TrimSpace(parts[index]), nil
|
|
}
|
|
}
|
|
return "", fmt.Errorf("cannot find %q in %q", match, data)
|
|
}
|