mirror of
https://github.com/ceph/ceph-csi.git
synced 2025-05-24 00:06:41 +00:00
deploy: add -automaxprocs
to reduce CPU and memory resources
With the new `-automaxprocs` commandline parameter, Ceph-CSI will adjust the GOMAXPROCS environment variable for the Golang runtime. The values are based on the CPU quota that is given to the process. This can reduce the number of threads that the Golang runtime spawns, which affects the require amount of memory as well. Updates: #5228 Signed-off-by: Niels de Vos <ndevos@ibm.com>
This commit is contained in:
parent
0cb6771115
commit
4db8b6222c
@ -35,6 +35,7 @@ import (
|
|||||||
"github.com/ceph/ceph-csi/internal/util"
|
"github.com/ceph/ceph-csi/internal/util"
|
||||||
"github.com/ceph/ceph-csi/internal/util/log"
|
"github.com/ceph/ceph-csi/internal/util/log"
|
||||||
|
|
||||||
|
"go.uber.org/automaxprocs/maxprocs"
|
||||||
"k8s.io/klog/v2"
|
"k8s.io/klog/v2"
|
||||||
ctrlLog "sigs.k8s.io/controller-runtime/pkg/log"
|
ctrlLog "sigs.k8s.io/controller-runtime/pkg/log"
|
||||||
)
|
)
|
||||||
@ -77,6 +78,8 @@ func init() {
|
|||||||
flag.StringVar(&conf.InstanceID, "instanceid", "default", "Unique ID distinguishing this instance of Ceph-CSI"+
|
flag.StringVar(&conf.InstanceID, "instanceid", "default", "Unique ID distinguishing this instance of Ceph-CSI"+
|
||||||
" among other instances, when sharing Ceph clusters across CSI instances for provisioning")
|
" among other instances, when sharing Ceph clusters across CSI instances for provisioning")
|
||||||
flag.IntVar(&conf.PidLimit, "pidlimit", 0, "the PID limit to configure through cgroups")
|
flag.IntVar(&conf.PidLimit, "pidlimit", 0, "the PID limit to configure through cgroups")
|
||||||
|
flag.BoolVar(&conf.AutoMaxProcs, "automaxprocs", false,
|
||||||
|
"automatically set GOMAXPROCS to match Linux container CPU quota")
|
||||||
flag.BoolVar(&conf.IsControllerServer, "controllerserver", false, "start cephcsi controller server")
|
flag.BoolVar(&conf.IsControllerServer, "controllerserver", false, "start cephcsi controller server")
|
||||||
flag.BoolVar(&conf.IsNodeServer, "nodeserver", false, "start cephcsi node server")
|
flag.BoolVar(&conf.IsNodeServer, "nodeserver", false, "start cephcsi node server")
|
||||||
flag.StringVar(
|
flag.StringVar(
|
||||||
@ -215,6 +218,15 @@ func main() {
|
|||||||
logAndExit(err.Error())
|
logAndExit(err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// maxprocs detects CPU quota based on cgroups, do this before setting PID-limit
|
||||||
|
if conf.AutoMaxProcs {
|
||||||
|
_, mpErr := maxprocs.Set(maxprocs.Logger(log.DefaultLog))
|
||||||
|
if mpErr != nil {
|
||||||
|
log.FatalLogMsg("failed to configure automaxprocs: %v", mpErr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// set the PID limit (for native Ceph threads) after conf.AutoMaxProcs
|
||||||
setPIDLimit(&conf)
|
setPIDLimit(&conf)
|
||||||
|
|
||||||
if conf.EnableProfiling || conf.Vtype == livenessType {
|
if conf.EnableProfiling || conf.Vtype == livenessType {
|
||||||
|
1
go.mod
1
go.mod
@ -29,6 +29,7 @@ require (
|
|||||||
github.com/pkg/xattr v0.4.10
|
github.com/pkg/xattr v0.4.10
|
||||||
github.com/prometheus/client_golang v1.22.0
|
github.com/prometheus/client_golang v1.22.0
|
||||||
github.com/stretchr/testify v1.10.0
|
github.com/stretchr/testify v1.10.0
|
||||||
|
go.uber.org/automaxprocs v1.6.0
|
||||||
golang.org/x/crypto v0.38.0
|
golang.org/x/crypto v0.38.0
|
||||||
golang.org/x/net v0.40.0
|
golang.org/x/net v0.40.0
|
||||||
golang.org/x/sys v0.33.0
|
golang.org/x/sys v0.33.0
|
||||||
|
4
go.sum
4
go.sum
@ -588,6 +588,8 @@ github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2/go.mod h1:iKH
|
|||||||
github.com/portworx/dcos-secrets v0.0.0-20180616013705-8e8ec3f66611/go.mod h1:4hklRW/4DQpLqkcXcjtNprbH2tz/sJaNtqinfPWl/LA=
|
github.com/portworx/dcos-secrets v0.0.0-20180616013705-8e8ec3f66611/go.mod h1:4hklRW/4DQpLqkcXcjtNprbH2tz/sJaNtqinfPWl/LA=
|
||||||
github.com/portworx/kvdb v0.0.0-20200929023115-b312c7519467/go.mod h1:Q8YyrNDvPp3DVF96BDcQuaC7fAYUCuUX+l58S7OnD2M=
|
github.com/portworx/kvdb v0.0.0-20200929023115-b312c7519467/go.mod h1:Q8YyrNDvPp3DVF96BDcQuaC7fAYUCuUX+l58S7OnD2M=
|
||||||
github.com/posener/complete v1.1.1/go.mod h1:em0nMJCgc9GFtwrmVmEMR/ZL6WyhyjMBndrE9hABlRI=
|
github.com/posener/complete v1.1.1/go.mod h1:em0nMJCgc9GFtwrmVmEMR/ZL6WyhyjMBndrE9hABlRI=
|
||||||
|
github.com/prashantv/gostub v1.1.0 h1:BTyx3RfQjRHnUWaGF9oQos79AlQ5k8WNktv7VGvVH4g=
|
||||||
|
github.com/prashantv/gostub v1.1.0/go.mod h1:A5zLQHz7ieHGG7is6LLXLz7I8+3LZzsrV0P1IAHhP5U=
|
||||||
github.com/prometheus/client_golang v1.22.0 h1:rb93p9lokFEsctTys46VnV1kLCDpVZ0a/Y92Vm0Zc6Q=
|
github.com/prometheus/client_golang v1.22.0 h1:rb93p9lokFEsctTys46VnV1kLCDpVZ0a/Y92Vm0Zc6Q=
|
||||||
github.com/prometheus/client_golang v1.22.0/go.mod h1:R7ljNsLXhuQXYZYtw6GAE9AZg8Y7vEW5scdCXrWRXC0=
|
github.com/prometheus/client_golang v1.22.0/go.mod h1:R7ljNsLXhuQXYZYtw6GAE9AZg8Y7vEW5scdCXrWRXC0=
|
||||||
github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=
|
github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=
|
||||||
@ -672,6 +674,8 @@ go.opentelemetry.io/otel/trace v1.34.0/go.mod h1:Svm7lSjQD7kG7KJ/MUHPVXSDGz2OX4h
|
|||||||
go.opentelemetry.io/proto/otlp v0.7.0/go.mod h1:PqfVotwruBrMGOCsRd/89rSnXhoiJIqeYNgFYFoEGnI=
|
go.opentelemetry.io/proto/otlp v0.7.0/go.mod h1:PqfVotwruBrMGOCsRd/89rSnXhoiJIqeYNgFYFoEGnI=
|
||||||
go.uber.org/atomic v1.7.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
|
go.uber.org/atomic v1.7.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
|
||||||
go.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
|
go.uber.org/atomic v1.9.0/go.mod h1:fEN4uk6kAWBTFdckzkM89CLk9XfWZrxpCo0nPH17wJc=
|
||||||
|
go.uber.org/automaxprocs v1.6.0 h1:O3y2/QNTOdbF+e/dpXNNW7Rx2hZ4sTIPyybbxyNqTUs=
|
||||||
|
go.uber.org/automaxprocs v1.6.0/go.mod h1:ifeIMSnPZuznNm6jmdzmU3/bfk01Fe2fotchwEFJ8r8=
|
||||||
go.uber.org/goleak v1.1.10/go.mod h1:8a7PlsEVH3e/a/GLqe5IIrQx6GzcnRmZEufDUTk4A7A=
|
go.uber.org/goleak v1.1.10/go.mod h1:8a7PlsEVH3e/a/GLqe5IIrQx6GzcnRmZEufDUTk4A7A=
|
||||||
go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=
|
go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=
|
||||||
go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=
|
go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=
|
||||||
|
@ -133,6 +133,8 @@ type Config struct {
|
|||||||
// are considered slow.
|
// are considered slow.
|
||||||
LogSlowOpInterval time.Duration
|
LogSlowOpInterval time.Duration
|
||||||
|
|
||||||
|
AutoMaxProcs bool // configure GOMAXPROCS with automaxprocs
|
||||||
|
|
||||||
EnableProfiling bool // flag to enable profiling
|
EnableProfiling bool // flag to enable profiling
|
||||||
IsControllerServer bool // if set to true start provisioner server
|
IsControllerServer bool // if set to true start provisioner server
|
||||||
IsNodeServer bool // if set to true start node server
|
IsNodeServer bool // if set to true start node server
|
||||||
|
19
vendor/go.uber.org/automaxprocs/LICENSE
generated
vendored
Normal file
19
vendor/go.uber.org/automaxprocs/LICENSE
generated
vendored
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
Copyright (c) 2017 Uber Technologies, Inc.
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
of this software and associated documentation files (the "Software"), to deal
|
||||||
|
in the Software without restriction, including without limitation the rights
|
||||||
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
copies of the Software, and to permit persons to whom the Software is
|
||||||
|
furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in
|
||||||
|
all copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
THE SOFTWARE.
|
79
vendor/go.uber.org/automaxprocs/internal/cgroups/cgroup.go
generated
vendored
Normal file
79
vendor/go.uber.org/automaxprocs/internal/cgroups/cgroup.go
generated
vendored
Normal file
@ -0,0 +1,79 @@
|
|||||||
|
// Copyright (c) 2017 Uber Technologies, Inc.
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in
|
||||||
|
// all copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
// THE SOFTWARE.
|
||||||
|
|
||||||
|
//go:build linux
|
||||||
|
// +build linux
|
||||||
|
|
||||||
|
package cgroups
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strconv"
|
||||||
|
)
|
||||||
|
|
||||||
|
// CGroup represents the data structure for a Linux control group.
|
||||||
|
type CGroup struct {
|
||||||
|
path string
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewCGroup returns a new *CGroup from a given path.
|
||||||
|
func NewCGroup(path string) *CGroup {
|
||||||
|
return &CGroup{path: path}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Path returns the path of the CGroup*.
|
||||||
|
func (cg *CGroup) Path() string {
|
||||||
|
return cg.path
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParamPath returns the path of the given cgroup param under itself.
|
||||||
|
func (cg *CGroup) ParamPath(param string) string {
|
||||||
|
return filepath.Join(cg.path, param)
|
||||||
|
}
|
||||||
|
|
||||||
|
// readFirstLine reads the first line from a cgroup param file.
|
||||||
|
func (cg *CGroup) readFirstLine(param string) (string, error) {
|
||||||
|
paramFile, err := os.Open(cg.ParamPath(param))
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
defer paramFile.Close()
|
||||||
|
|
||||||
|
scanner := bufio.NewScanner(paramFile)
|
||||||
|
if scanner.Scan() {
|
||||||
|
return scanner.Text(), nil
|
||||||
|
}
|
||||||
|
if err := scanner.Err(); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return "", io.ErrUnexpectedEOF
|
||||||
|
}
|
||||||
|
|
||||||
|
// readInt parses the first line from a cgroup param file as int.
|
||||||
|
func (cg *CGroup) readInt(param string) (int, error) {
|
||||||
|
text, err := cg.readFirstLine(param)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
return strconv.Atoi(text)
|
||||||
|
}
|
118
vendor/go.uber.org/automaxprocs/internal/cgroups/cgroups.go
generated
vendored
Normal file
118
vendor/go.uber.org/automaxprocs/internal/cgroups/cgroups.go
generated
vendored
Normal file
@ -0,0 +1,118 @@
|
|||||||
|
// Copyright (c) 2017 Uber Technologies, Inc.
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in
|
||||||
|
// all copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
// THE SOFTWARE.
|
||||||
|
|
||||||
|
//go:build linux
|
||||||
|
// +build linux
|
||||||
|
|
||||||
|
package cgroups
|
||||||
|
|
||||||
|
const (
|
||||||
|
// _cgroupFSType is the Linux CGroup file system type used in
|
||||||
|
// `/proc/$PID/mountinfo`.
|
||||||
|
_cgroupFSType = "cgroup"
|
||||||
|
// _cgroupSubsysCPU is the CPU CGroup subsystem.
|
||||||
|
_cgroupSubsysCPU = "cpu"
|
||||||
|
// _cgroupSubsysCPUAcct is the CPU accounting CGroup subsystem.
|
||||||
|
_cgroupSubsysCPUAcct = "cpuacct"
|
||||||
|
// _cgroupSubsysCPUSet is the CPUSet CGroup subsystem.
|
||||||
|
_cgroupSubsysCPUSet = "cpuset"
|
||||||
|
// _cgroupSubsysMemory is the Memory CGroup subsystem.
|
||||||
|
_cgroupSubsysMemory = "memory"
|
||||||
|
|
||||||
|
// _cgroupCPUCFSQuotaUsParam is the file name for the CGroup CFS quota
|
||||||
|
// parameter.
|
||||||
|
_cgroupCPUCFSQuotaUsParam = "cpu.cfs_quota_us"
|
||||||
|
// _cgroupCPUCFSPeriodUsParam is the file name for the CGroup CFS period
|
||||||
|
// parameter.
|
||||||
|
_cgroupCPUCFSPeriodUsParam = "cpu.cfs_period_us"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
_procPathCGroup = "/proc/self/cgroup"
|
||||||
|
_procPathMountInfo = "/proc/self/mountinfo"
|
||||||
|
)
|
||||||
|
|
||||||
|
// CGroups is a map that associates each CGroup with its subsystem name.
|
||||||
|
type CGroups map[string]*CGroup
|
||||||
|
|
||||||
|
// NewCGroups returns a new *CGroups from given `mountinfo` and `cgroup` files
|
||||||
|
// under for some process under `/proc` file system (see also proc(5) for more
|
||||||
|
// information).
|
||||||
|
func NewCGroups(procPathMountInfo, procPathCGroup string) (CGroups, error) {
|
||||||
|
cgroupSubsystems, err := parseCGroupSubsystems(procPathCGroup)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
cgroups := make(CGroups)
|
||||||
|
newMountPoint := func(mp *MountPoint) error {
|
||||||
|
if mp.FSType != _cgroupFSType {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, opt := range mp.SuperOptions {
|
||||||
|
subsys, exists := cgroupSubsystems[opt]
|
||||||
|
if !exists {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
cgroupPath, err := mp.Translate(subsys.Name)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
cgroups[opt] = NewCGroup(cgroupPath)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := parseMountInfo(procPathMountInfo, newMountPoint); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return cgroups, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewCGroupsForCurrentProcess returns a new *CGroups instance for the current
|
||||||
|
// process.
|
||||||
|
func NewCGroupsForCurrentProcess() (CGroups, error) {
|
||||||
|
return NewCGroups(_procPathMountInfo, _procPathCGroup)
|
||||||
|
}
|
||||||
|
|
||||||
|
// CPUQuota returns the CPU quota applied with the CPU cgroup controller.
|
||||||
|
// It is a result of `cpu.cfs_quota_us / cpu.cfs_period_us`. If the value of
|
||||||
|
// `cpu.cfs_quota_us` was not set (-1), the method returns `(-1, nil)`.
|
||||||
|
func (cg CGroups) CPUQuota() (float64, bool, error) {
|
||||||
|
cpuCGroup, exists := cg[_cgroupSubsysCPU]
|
||||||
|
if !exists {
|
||||||
|
return -1, false, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
cfsQuotaUs, err := cpuCGroup.readInt(_cgroupCPUCFSQuotaUsParam)
|
||||||
|
if defined := cfsQuotaUs > 0; err != nil || !defined {
|
||||||
|
return -1, defined, err
|
||||||
|
}
|
||||||
|
|
||||||
|
cfsPeriodUs, err := cpuCGroup.readInt(_cgroupCPUCFSPeriodUsParam)
|
||||||
|
if defined := cfsPeriodUs > 0; err != nil || !defined {
|
||||||
|
return -1, defined, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return float64(cfsQuotaUs) / float64(cfsPeriodUs), true, nil
|
||||||
|
}
|
176
vendor/go.uber.org/automaxprocs/internal/cgroups/cgroups2.go
generated
vendored
Normal file
176
vendor/go.uber.org/automaxprocs/internal/cgroups/cgroups2.go
generated
vendored
Normal file
@ -0,0 +1,176 @@
|
|||||||
|
// Copyright (c) 2022 Uber Technologies, Inc.
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in
|
||||||
|
// all copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
// THE SOFTWARE.
|
||||||
|
|
||||||
|
//go:build linux
|
||||||
|
// +build linux
|
||||||
|
|
||||||
|
package cgroups
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"path"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// _cgroupv2CPUMax is the file name for the CGroup-V2 CPU max and period
|
||||||
|
// parameter.
|
||||||
|
_cgroupv2CPUMax = "cpu.max"
|
||||||
|
// _cgroupFSType is the Linux CGroup-V2 file system type used in
|
||||||
|
// `/proc/$PID/mountinfo`.
|
||||||
|
_cgroupv2FSType = "cgroup2"
|
||||||
|
|
||||||
|
_cgroupv2MountPoint = "/sys/fs/cgroup"
|
||||||
|
|
||||||
|
_cgroupV2CPUMaxDefaultPeriod = 100000
|
||||||
|
_cgroupV2CPUMaxQuotaMax = "max"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
_cgroupv2CPUMaxQuotaIndex = iota
|
||||||
|
_cgroupv2CPUMaxPeriodIndex
|
||||||
|
)
|
||||||
|
|
||||||
|
// ErrNotV2 indicates that the system is not using cgroups2.
|
||||||
|
var ErrNotV2 = errors.New("not using cgroups2")
|
||||||
|
|
||||||
|
// CGroups2 provides access to cgroups data for systems using cgroups2.
|
||||||
|
type CGroups2 struct {
|
||||||
|
mountPoint string
|
||||||
|
groupPath string
|
||||||
|
cpuMaxFile string
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewCGroups2ForCurrentProcess builds a CGroups2 for the current process.
|
||||||
|
//
|
||||||
|
// This returns ErrNotV2 if the system is not using cgroups2.
|
||||||
|
func NewCGroups2ForCurrentProcess() (*CGroups2, error) {
|
||||||
|
return newCGroups2From(_procPathMountInfo, _procPathCGroup)
|
||||||
|
}
|
||||||
|
|
||||||
|
func newCGroups2From(mountInfoPath, procPathCGroup string) (*CGroups2, error) {
|
||||||
|
isV2, err := isCGroupV2(mountInfoPath)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if !isV2 {
|
||||||
|
return nil, ErrNotV2
|
||||||
|
}
|
||||||
|
|
||||||
|
subsystems, err := parseCGroupSubsystems(procPathCGroup)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Find v2 subsystem by looking for the `0` id
|
||||||
|
var v2subsys *CGroupSubsys
|
||||||
|
for _, subsys := range subsystems {
|
||||||
|
if subsys.ID == 0 {
|
||||||
|
v2subsys = subsys
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if v2subsys == nil {
|
||||||
|
return nil, ErrNotV2
|
||||||
|
}
|
||||||
|
|
||||||
|
return &CGroups2{
|
||||||
|
mountPoint: _cgroupv2MountPoint,
|
||||||
|
groupPath: v2subsys.Name,
|
||||||
|
cpuMaxFile: _cgroupv2CPUMax,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func isCGroupV2(procPathMountInfo string) (bool, error) {
|
||||||
|
var (
|
||||||
|
isV2 bool
|
||||||
|
newMountPoint = func(mp *MountPoint) error {
|
||||||
|
isV2 = isV2 || (mp.FSType == _cgroupv2FSType && mp.MountPoint == _cgroupv2MountPoint)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
if err := parseMountInfo(procPathMountInfo, newMountPoint); err != nil {
|
||||||
|
return false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return isV2, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// CPUQuota returns the CPU quota applied with the CPU cgroup2 controller.
|
||||||
|
// It is a result of reading cpu quota and period from cpu.max file.
|
||||||
|
// It will return `cpu.max / cpu.period`. If cpu.max is set to max, it returns
|
||||||
|
// (-1, false, nil)
|
||||||
|
func (cg *CGroups2) CPUQuota() (float64, bool, error) {
|
||||||
|
cpuMaxParams, err := os.Open(path.Join(cg.mountPoint, cg.groupPath, cg.cpuMaxFile))
|
||||||
|
if err != nil {
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
return -1, false, nil
|
||||||
|
}
|
||||||
|
return -1, false, err
|
||||||
|
}
|
||||||
|
defer cpuMaxParams.Close()
|
||||||
|
|
||||||
|
scanner := bufio.NewScanner(cpuMaxParams)
|
||||||
|
if scanner.Scan() {
|
||||||
|
fields := strings.Fields(scanner.Text())
|
||||||
|
if len(fields) == 0 || len(fields) > 2 {
|
||||||
|
return -1, false, fmt.Errorf("invalid format")
|
||||||
|
}
|
||||||
|
|
||||||
|
if fields[_cgroupv2CPUMaxQuotaIndex] == _cgroupV2CPUMaxQuotaMax {
|
||||||
|
return -1, false, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
max, err := strconv.Atoi(fields[_cgroupv2CPUMaxQuotaIndex])
|
||||||
|
if err != nil {
|
||||||
|
return -1, false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var period int
|
||||||
|
if len(fields) == 1 {
|
||||||
|
period = _cgroupV2CPUMaxDefaultPeriod
|
||||||
|
} else {
|
||||||
|
period, err = strconv.Atoi(fields[_cgroupv2CPUMaxPeriodIndex])
|
||||||
|
if err != nil {
|
||||||
|
return -1, false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if period == 0 {
|
||||||
|
return -1, false, errors.New("zero value for period is not allowed")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return float64(max) / float64(period), true, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := scanner.Err(); err != nil {
|
||||||
|
return -1, false, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0, false, io.ErrUnexpectedEOF
|
||||||
|
}
|
23
vendor/go.uber.org/automaxprocs/internal/cgroups/doc.go
generated
vendored
Normal file
23
vendor/go.uber.org/automaxprocs/internal/cgroups/doc.go
generated
vendored
Normal file
@ -0,0 +1,23 @@
|
|||||||
|
// Copyright (c) 2017 Uber Technologies, Inc.
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in
|
||||||
|
// all copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
// THE SOFTWARE.
|
||||||
|
|
||||||
|
// Package cgroups provides utilities to access Linux control group (CGroups)
|
||||||
|
// parameters (CPU quota, for example) for a given process.
|
||||||
|
package cgroups
|
52
vendor/go.uber.org/automaxprocs/internal/cgroups/errors.go
generated
vendored
Normal file
52
vendor/go.uber.org/automaxprocs/internal/cgroups/errors.go
generated
vendored
Normal file
@ -0,0 +1,52 @@
|
|||||||
|
// Copyright (c) 2017 Uber Technologies, Inc.
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in
|
||||||
|
// all copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
// THE SOFTWARE.
|
||||||
|
|
||||||
|
//go:build linux
|
||||||
|
// +build linux
|
||||||
|
|
||||||
|
package cgroups
|
||||||
|
|
||||||
|
import "fmt"
|
||||||
|
|
||||||
|
type cgroupSubsysFormatInvalidError struct {
|
||||||
|
line string
|
||||||
|
}
|
||||||
|
|
||||||
|
type mountPointFormatInvalidError struct {
|
||||||
|
line string
|
||||||
|
}
|
||||||
|
|
||||||
|
type pathNotExposedFromMountPointError struct {
|
||||||
|
mountPoint string
|
||||||
|
root string
|
||||||
|
path string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (err cgroupSubsysFormatInvalidError) Error() string {
|
||||||
|
return fmt.Sprintf("invalid format for CGroupSubsys: %q", err.line)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (err mountPointFormatInvalidError) Error() string {
|
||||||
|
return fmt.Sprintf("invalid format for MountPoint: %q", err.line)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (err pathNotExposedFromMountPointError) Error() string {
|
||||||
|
return fmt.Sprintf("path %q is not a descendant of mount point root %q and cannot be exposed from %q", err.path, err.root, err.mountPoint)
|
||||||
|
}
|
171
vendor/go.uber.org/automaxprocs/internal/cgroups/mountpoint.go
generated
vendored
Normal file
171
vendor/go.uber.org/automaxprocs/internal/cgroups/mountpoint.go
generated
vendored
Normal file
@ -0,0 +1,171 @@
|
|||||||
|
// Copyright (c) 2017 Uber Technologies, Inc.
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in
|
||||||
|
// all copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
// THE SOFTWARE.
|
||||||
|
|
||||||
|
//go:build linux
|
||||||
|
// +build linux
|
||||||
|
|
||||||
|
package cgroups
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
_mountInfoSep = " "
|
||||||
|
_mountInfoOptsSep = ","
|
||||||
|
_mountInfoOptionalFieldsSep = "-"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
_miFieldIDMountID = iota
|
||||||
|
_miFieldIDParentID
|
||||||
|
_miFieldIDDeviceID
|
||||||
|
_miFieldIDRoot
|
||||||
|
_miFieldIDMountPoint
|
||||||
|
_miFieldIDOptions
|
||||||
|
_miFieldIDOptionalFields
|
||||||
|
|
||||||
|
_miFieldCountFirstHalf
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
_miFieldOffsetFSType = iota
|
||||||
|
_miFieldOffsetMountSource
|
||||||
|
_miFieldOffsetSuperOptions
|
||||||
|
|
||||||
|
_miFieldCountSecondHalf
|
||||||
|
)
|
||||||
|
|
||||||
|
const _miFieldCountMin = _miFieldCountFirstHalf + _miFieldCountSecondHalf
|
||||||
|
|
||||||
|
// MountPoint is the data structure for the mount points in
|
||||||
|
// `/proc/$PID/mountinfo`. See also proc(5) for more information.
|
||||||
|
type MountPoint struct {
|
||||||
|
MountID int
|
||||||
|
ParentID int
|
||||||
|
DeviceID string
|
||||||
|
Root string
|
||||||
|
MountPoint string
|
||||||
|
Options []string
|
||||||
|
OptionalFields []string
|
||||||
|
FSType string
|
||||||
|
MountSource string
|
||||||
|
SuperOptions []string
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewMountPointFromLine parses a line read from `/proc/$PID/mountinfo` and
|
||||||
|
// returns a new *MountPoint.
|
||||||
|
func NewMountPointFromLine(line string) (*MountPoint, error) {
|
||||||
|
fields := strings.Split(line, _mountInfoSep)
|
||||||
|
|
||||||
|
if len(fields) < _miFieldCountMin {
|
||||||
|
return nil, mountPointFormatInvalidError{line}
|
||||||
|
}
|
||||||
|
|
||||||
|
mountID, err := strconv.Atoi(fields[_miFieldIDMountID])
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
parentID, err := strconv.Atoi(fields[_miFieldIDParentID])
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, field := range fields[_miFieldIDOptionalFields:] {
|
||||||
|
if field == _mountInfoOptionalFieldsSep {
|
||||||
|
// End of optional fields.
|
||||||
|
fsTypeStart := _miFieldIDOptionalFields + i + 1
|
||||||
|
|
||||||
|
// Now we know where the optional fields end, split the line again with a
|
||||||
|
// limit to avoid issues with spaces in super options as present on WSL.
|
||||||
|
fields = strings.SplitN(line, _mountInfoSep, fsTypeStart+_miFieldCountSecondHalf)
|
||||||
|
if len(fields) != fsTypeStart+_miFieldCountSecondHalf {
|
||||||
|
return nil, mountPointFormatInvalidError{line}
|
||||||
|
}
|
||||||
|
|
||||||
|
miFieldIDFSType := _miFieldOffsetFSType + fsTypeStart
|
||||||
|
miFieldIDMountSource := _miFieldOffsetMountSource + fsTypeStart
|
||||||
|
miFieldIDSuperOptions := _miFieldOffsetSuperOptions + fsTypeStart
|
||||||
|
|
||||||
|
return &MountPoint{
|
||||||
|
MountID: mountID,
|
||||||
|
ParentID: parentID,
|
||||||
|
DeviceID: fields[_miFieldIDDeviceID],
|
||||||
|
Root: fields[_miFieldIDRoot],
|
||||||
|
MountPoint: fields[_miFieldIDMountPoint],
|
||||||
|
Options: strings.Split(fields[_miFieldIDOptions], _mountInfoOptsSep),
|
||||||
|
OptionalFields: fields[_miFieldIDOptionalFields:(fsTypeStart - 1)],
|
||||||
|
FSType: fields[miFieldIDFSType],
|
||||||
|
MountSource: fields[miFieldIDMountSource],
|
||||||
|
SuperOptions: strings.Split(fields[miFieldIDSuperOptions], _mountInfoOptsSep),
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, mountPointFormatInvalidError{line}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Translate converts an absolute path inside the *MountPoint's file system to
|
||||||
|
// the host file system path in the mount namespace the *MountPoint belongs to.
|
||||||
|
func (mp *MountPoint) Translate(absPath string) (string, error) {
|
||||||
|
relPath, err := filepath.Rel(mp.Root, absPath)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
if relPath == ".." || strings.HasPrefix(relPath, "../") {
|
||||||
|
return "", pathNotExposedFromMountPointError{
|
||||||
|
mountPoint: mp.MountPoint,
|
||||||
|
root: mp.Root,
|
||||||
|
path: absPath,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return filepath.Join(mp.MountPoint, relPath), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// parseMountInfo parses procPathMountInfo (usually at `/proc/$PID/mountinfo`)
|
||||||
|
// and yields parsed *MountPoint into newMountPoint.
|
||||||
|
func parseMountInfo(procPathMountInfo string, newMountPoint func(*MountPoint) error) error {
|
||||||
|
mountInfoFile, err := os.Open(procPathMountInfo)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer mountInfoFile.Close()
|
||||||
|
|
||||||
|
scanner := bufio.NewScanner(mountInfoFile)
|
||||||
|
|
||||||
|
for scanner.Scan() {
|
||||||
|
mountPoint, err := NewMountPointFromLine(scanner.Text())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if err := newMountPoint(mountPoint); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return scanner.Err()
|
||||||
|
}
|
103
vendor/go.uber.org/automaxprocs/internal/cgroups/subsys.go
generated
vendored
Normal file
103
vendor/go.uber.org/automaxprocs/internal/cgroups/subsys.go
generated
vendored
Normal file
@ -0,0 +1,103 @@
|
|||||||
|
// Copyright (c) 2017 Uber Technologies, Inc.
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in
|
||||||
|
// all copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
// THE SOFTWARE.
|
||||||
|
|
||||||
|
//go:build linux
|
||||||
|
// +build linux
|
||||||
|
|
||||||
|
package cgroups
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"os"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
_cgroupSep = ":"
|
||||||
|
_cgroupSubsysSep = ","
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
_csFieldIDID = iota
|
||||||
|
_csFieldIDSubsystems
|
||||||
|
_csFieldIDName
|
||||||
|
_csFieldCount
|
||||||
|
)
|
||||||
|
|
||||||
|
// CGroupSubsys represents the data structure for entities in
|
||||||
|
// `/proc/$PID/cgroup`. See also proc(5) for more information.
|
||||||
|
type CGroupSubsys struct {
|
||||||
|
ID int
|
||||||
|
Subsystems []string
|
||||||
|
Name string
|
||||||
|
}
|
||||||
|
|
||||||
|
// NewCGroupSubsysFromLine returns a new *CGroupSubsys by parsing a string in
|
||||||
|
// the format of `/proc/$PID/cgroup`
|
||||||
|
func NewCGroupSubsysFromLine(line string) (*CGroupSubsys, error) {
|
||||||
|
fields := strings.SplitN(line, _cgroupSep, _csFieldCount)
|
||||||
|
|
||||||
|
if len(fields) != _csFieldCount {
|
||||||
|
return nil, cgroupSubsysFormatInvalidError{line}
|
||||||
|
}
|
||||||
|
|
||||||
|
id, err := strconv.Atoi(fields[_csFieldIDID])
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
cgroup := &CGroupSubsys{
|
||||||
|
ID: id,
|
||||||
|
Subsystems: strings.Split(fields[_csFieldIDSubsystems], _cgroupSubsysSep),
|
||||||
|
Name: fields[_csFieldIDName],
|
||||||
|
}
|
||||||
|
|
||||||
|
return cgroup, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// parseCGroupSubsystems parses procPathCGroup (usually at `/proc/$PID/cgroup`)
|
||||||
|
// and returns a new map[string]*CGroupSubsys.
|
||||||
|
func parseCGroupSubsystems(procPathCGroup string) (map[string]*CGroupSubsys, error) {
|
||||||
|
cgroupFile, err := os.Open(procPathCGroup)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer cgroupFile.Close()
|
||||||
|
|
||||||
|
scanner := bufio.NewScanner(cgroupFile)
|
||||||
|
subsystems := make(map[string]*CGroupSubsys)
|
||||||
|
|
||||||
|
for scanner.Scan() {
|
||||||
|
cgroup, err := NewCGroupSubsysFromLine(scanner.Text())
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
for _, subsys := range cgroup.Subsystems {
|
||||||
|
subsystems[subsys] = cgroup
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := scanner.Err(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return subsystems, nil
|
||||||
|
}
|
75
vendor/go.uber.org/automaxprocs/internal/runtime/cpu_quota_linux.go
generated
vendored
Normal file
75
vendor/go.uber.org/automaxprocs/internal/runtime/cpu_quota_linux.go
generated
vendored
Normal file
@ -0,0 +1,75 @@
|
|||||||
|
// Copyright (c) 2017 Uber Technologies, Inc.
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in
|
||||||
|
// all copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
// THE SOFTWARE.
|
||||||
|
|
||||||
|
//go:build linux
|
||||||
|
// +build linux
|
||||||
|
|
||||||
|
package runtime
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
|
||||||
|
cg "go.uber.org/automaxprocs/internal/cgroups"
|
||||||
|
)
|
||||||
|
|
||||||
|
// CPUQuotaToGOMAXPROCS converts the CPU quota applied to the calling process
|
||||||
|
// to a valid GOMAXPROCS value. The quota is converted from float to int using round.
|
||||||
|
// If round == nil, DefaultRoundFunc is used.
|
||||||
|
func CPUQuotaToGOMAXPROCS(minValue int, round func(v float64) int) (int, CPUQuotaStatus, error) {
|
||||||
|
if round == nil {
|
||||||
|
round = DefaultRoundFunc
|
||||||
|
}
|
||||||
|
cgroups, err := _newQueryer()
|
||||||
|
if err != nil {
|
||||||
|
return -1, CPUQuotaUndefined, err
|
||||||
|
}
|
||||||
|
|
||||||
|
quota, defined, err := cgroups.CPUQuota()
|
||||||
|
if !defined || err != nil {
|
||||||
|
return -1, CPUQuotaUndefined, err
|
||||||
|
}
|
||||||
|
|
||||||
|
maxProcs := round(quota)
|
||||||
|
if minValue > 0 && maxProcs < minValue {
|
||||||
|
return minValue, CPUQuotaMinUsed, nil
|
||||||
|
}
|
||||||
|
return maxProcs, CPUQuotaUsed, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type queryer interface {
|
||||||
|
CPUQuota() (float64, bool, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
_newCgroups2 = cg.NewCGroups2ForCurrentProcess
|
||||||
|
_newCgroups = cg.NewCGroupsForCurrentProcess
|
||||||
|
_newQueryer = newQueryer
|
||||||
|
)
|
||||||
|
|
||||||
|
func newQueryer() (queryer, error) {
|
||||||
|
cgroups, err := _newCgroups2()
|
||||||
|
if err == nil {
|
||||||
|
return cgroups, nil
|
||||||
|
}
|
||||||
|
if errors.Is(err, cg.ErrNotV2) {
|
||||||
|
return _newCgroups()
|
||||||
|
}
|
||||||
|
return nil, err
|
||||||
|
}
|
31
vendor/go.uber.org/automaxprocs/internal/runtime/cpu_quota_unsupported.go
generated
vendored
Normal file
31
vendor/go.uber.org/automaxprocs/internal/runtime/cpu_quota_unsupported.go
generated
vendored
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
// Copyright (c) 2017 Uber Technologies, Inc.
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in
|
||||||
|
// all copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
// THE SOFTWARE.
|
||||||
|
|
||||||
|
//go:build !linux
|
||||||
|
// +build !linux
|
||||||
|
|
||||||
|
package runtime
|
||||||
|
|
||||||
|
// CPUQuotaToGOMAXPROCS converts the CPU quota applied to the calling process
|
||||||
|
// to a valid GOMAXPROCS value. This is Linux-specific and not supported in the
|
||||||
|
// current OS.
|
||||||
|
func CPUQuotaToGOMAXPROCS(_ int, _ func(v float64) int) (int, CPUQuotaStatus, error) {
|
||||||
|
return -1, CPUQuotaUndefined, nil
|
||||||
|
}
|
40
vendor/go.uber.org/automaxprocs/internal/runtime/runtime.go
generated
vendored
Normal file
40
vendor/go.uber.org/automaxprocs/internal/runtime/runtime.go
generated
vendored
Normal file
@ -0,0 +1,40 @@
|
|||||||
|
// Copyright (c) 2017 Uber Technologies, Inc.
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in
|
||||||
|
// all copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
// THE SOFTWARE.
|
||||||
|
|
||||||
|
package runtime
|
||||||
|
|
||||||
|
import "math"
|
||||||
|
|
||||||
|
// CPUQuotaStatus presents the status of how CPU quota is used
|
||||||
|
type CPUQuotaStatus int
|
||||||
|
|
||||||
|
const (
|
||||||
|
// CPUQuotaUndefined is returned when CPU quota is undefined
|
||||||
|
CPUQuotaUndefined CPUQuotaStatus = iota
|
||||||
|
// CPUQuotaUsed is returned when a valid CPU quota can be used
|
||||||
|
CPUQuotaUsed
|
||||||
|
// CPUQuotaMinUsed is returned when CPU quota is smaller than the min value
|
||||||
|
CPUQuotaMinUsed
|
||||||
|
)
|
||||||
|
|
||||||
|
// DefaultRoundFunc is the default function to convert CPU quota from float to int. It rounds the value down (floor).
|
||||||
|
func DefaultRoundFunc(v float64) int {
|
||||||
|
return int(math.Floor(v))
|
||||||
|
}
|
139
vendor/go.uber.org/automaxprocs/maxprocs/maxprocs.go
generated
vendored
Normal file
139
vendor/go.uber.org/automaxprocs/maxprocs/maxprocs.go
generated
vendored
Normal file
@ -0,0 +1,139 @@
|
|||||||
|
// Copyright (c) 2017 Uber Technologies, Inc.
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in
|
||||||
|
// all copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
// THE SOFTWARE.
|
||||||
|
|
||||||
|
// Package maxprocs lets Go programs easily configure runtime.GOMAXPROCS to
|
||||||
|
// match the configured Linux CPU quota. Unlike the top-level automaxprocs
|
||||||
|
// package, it lets the caller configure logging and handle errors.
|
||||||
|
package maxprocs // import "go.uber.org/automaxprocs/maxprocs"
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"runtime"
|
||||||
|
|
||||||
|
iruntime "go.uber.org/automaxprocs/internal/runtime"
|
||||||
|
)
|
||||||
|
|
||||||
|
const _maxProcsKey = "GOMAXPROCS"
|
||||||
|
|
||||||
|
func currentMaxProcs() int {
|
||||||
|
return runtime.GOMAXPROCS(0)
|
||||||
|
}
|
||||||
|
|
||||||
|
type config struct {
|
||||||
|
printf func(string, ...interface{})
|
||||||
|
procs func(int, func(v float64) int) (int, iruntime.CPUQuotaStatus, error)
|
||||||
|
minGOMAXPROCS int
|
||||||
|
roundQuotaFunc func(v float64) int
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *config) log(fmt string, args ...interface{}) {
|
||||||
|
if c.printf != nil {
|
||||||
|
c.printf(fmt, args...)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// An Option alters the behavior of Set.
|
||||||
|
type Option interface {
|
||||||
|
apply(*config)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Logger uses the supplied printf implementation for log output. By default,
|
||||||
|
// Set doesn't log anything.
|
||||||
|
func Logger(printf func(string, ...interface{})) Option {
|
||||||
|
return optionFunc(func(cfg *config) {
|
||||||
|
cfg.printf = printf
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// Min sets the minimum GOMAXPROCS value that will be used.
|
||||||
|
// Any value below 1 is ignored.
|
||||||
|
func Min(n int) Option {
|
||||||
|
return optionFunc(func(cfg *config) {
|
||||||
|
if n >= 1 {
|
||||||
|
cfg.minGOMAXPROCS = n
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// RoundQuotaFunc sets the function that will be used to covert the CPU quota from float to int.
|
||||||
|
func RoundQuotaFunc(rf func(v float64) int) Option {
|
||||||
|
return optionFunc(func(cfg *config) {
|
||||||
|
cfg.roundQuotaFunc = rf
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
type optionFunc func(*config)
|
||||||
|
|
||||||
|
func (of optionFunc) apply(cfg *config) { of(cfg) }
|
||||||
|
|
||||||
|
// Set GOMAXPROCS to match the Linux container CPU quota (if any), returning
|
||||||
|
// any error encountered and an undo function.
|
||||||
|
//
|
||||||
|
// Set is a no-op on non-Linux systems and in Linux environments without a
|
||||||
|
// configured CPU quota.
|
||||||
|
func Set(opts ...Option) (func(), error) {
|
||||||
|
cfg := &config{
|
||||||
|
procs: iruntime.CPUQuotaToGOMAXPROCS,
|
||||||
|
roundQuotaFunc: iruntime.DefaultRoundFunc,
|
||||||
|
minGOMAXPROCS: 1,
|
||||||
|
}
|
||||||
|
for _, o := range opts {
|
||||||
|
o.apply(cfg)
|
||||||
|
}
|
||||||
|
|
||||||
|
undoNoop := func() {
|
||||||
|
cfg.log("maxprocs: No GOMAXPROCS change to reset")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Honor the GOMAXPROCS environment variable if present. Otherwise, amend
|
||||||
|
// `runtime.GOMAXPROCS()` with the current process' CPU quota if the OS is
|
||||||
|
// Linux, and guarantee a minimum value of 1. The minimum guaranteed value
|
||||||
|
// can be overridden using `maxprocs.Min()`.
|
||||||
|
if max, exists := os.LookupEnv(_maxProcsKey); exists {
|
||||||
|
cfg.log("maxprocs: Honoring GOMAXPROCS=%q as set in environment", max)
|
||||||
|
return undoNoop, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
maxProcs, status, err := cfg.procs(cfg.minGOMAXPROCS, cfg.roundQuotaFunc)
|
||||||
|
if err != nil {
|
||||||
|
return undoNoop, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if status == iruntime.CPUQuotaUndefined {
|
||||||
|
cfg.log("maxprocs: Leaving GOMAXPROCS=%v: CPU quota undefined", currentMaxProcs())
|
||||||
|
return undoNoop, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
prev := currentMaxProcs()
|
||||||
|
undo := func() {
|
||||||
|
cfg.log("maxprocs: Resetting GOMAXPROCS to %v", prev)
|
||||||
|
runtime.GOMAXPROCS(prev)
|
||||||
|
}
|
||||||
|
|
||||||
|
switch status {
|
||||||
|
case iruntime.CPUQuotaMinUsed:
|
||||||
|
cfg.log("maxprocs: Updating GOMAXPROCS=%v: using minimum allowed GOMAXPROCS", maxProcs)
|
||||||
|
case iruntime.CPUQuotaUsed:
|
||||||
|
cfg.log("maxprocs: Updating GOMAXPROCS=%v: determined from CPU quota", maxProcs)
|
||||||
|
}
|
||||||
|
|
||||||
|
runtime.GOMAXPROCS(maxProcs)
|
||||||
|
return undo, nil
|
||||||
|
}
|
24
vendor/go.uber.org/automaxprocs/maxprocs/version.go
generated
vendored
Normal file
24
vendor/go.uber.org/automaxprocs/maxprocs/version.go
generated
vendored
Normal file
@ -0,0 +1,24 @@
|
|||||||
|
// Copyright (c) 2017 Uber Technologies, Inc.
|
||||||
|
//
|
||||||
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
// of this software and associated documentation files (the "Software"), to deal
|
||||||
|
// in the Software without restriction, including without limitation the rights
|
||||||
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
// copies of the Software, and to permit persons to whom the Software is
|
||||||
|
// furnished to do so, subject to the following conditions:
|
||||||
|
//
|
||||||
|
// The above copyright notice and this permission notice shall be included in
|
||||||
|
// all copies or substantial portions of the Software.
|
||||||
|
//
|
||||||
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
// THE SOFTWARE.
|
||||||
|
|
||||||
|
package maxprocs
|
||||||
|
|
||||||
|
// Version is the current package version.
|
||||||
|
const Version = "1.6.0"
|
5
vendor/modules.txt
vendored
5
vendor/modules.txt
vendored
@ -518,6 +518,11 @@ go.opentelemetry.io/otel/metric/noop
|
|||||||
go.opentelemetry.io/otel/trace
|
go.opentelemetry.io/otel/trace
|
||||||
go.opentelemetry.io/otel/trace/embedded
|
go.opentelemetry.io/otel/trace/embedded
|
||||||
go.opentelemetry.io/otel/trace/noop
|
go.opentelemetry.io/otel/trace/noop
|
||||||
|
# go.uber.org/automaxprocs v1.6.0
|
||||||
|
## explicit; go 1.20
|
||||||
|
go.uber.org/automaxprocs/internal/cgroups
|
||||||
|
go.uber.org/automaxprocs/internal/runtime
|
||||||
|
go.uber.org/automaxprocs/maxprocs
|
||||||
# go.uber.org/multierr v1.11.0
|
# go.uber.org/multierr v1.11.0
|
||||||
## explicit; go 1.19
|
## explicit; go 1.19
|
||||||
go.uber.org/multierr
|
go.uber.org/multierr
|
||||||
|
Loading…
x
Reference in New Issue
Block a user