chore: simplify linux nocgroup logic (#3949)
This commit is contained in:
@@ -1,5 +1,3 @@
|
||||
//go:build !nocgroup
|
||||
|
||||
package internal
|
||||
|
||||
import (
|
||||
@@ -25,6 +23,7 @@ var (
|
||||
preTotal uint64
|
||||
limit float64
|
||||
cores uint64
|
||||
noCgroup bool
|
||||
initOnce sync.Once
|
||||
)
|
||||
|
||||
@@ -32,6 +31,7 @@ var (
|
||||
func initialize() {
|
||||
cpus, err := effectiveCpus()
|
||||
if err != nil {
|
||||
noCgroup = true
|
||||
logx.Error(err)
|
||||
return
|
||||
}
|
||||
@@ -47,12 +47,14 @@ func initialize() {
|
||||
|
||||
preSystem, err = systemCpuUsage()
|
||||
if err != nil {
|
||||
noCgroup = true
|
||||
logx.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
preTotal, err = cpuUsage()
|
||||
if err != nil {
|
||||
noCgroup = true
|
||||
logx.Error(err)
|
||||
return
|
||||
}
|
||||
@@ -62,6 +64,10 @@ func initialize() {
|
||||
func RefreshCpu() uint64 {
|
||||
initOnce.Do(initialize)
|
||||
|
||||
if noCgroup {
|
||||
return 0
|
||||
}
|
||||
|
||||
total, err := cpuUsage()
|
||||
if err != nil {
|
||||
return 0
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
//go:build !linux || nocgroup
|
||||
//go:build !linux
|
||||
|
||||
package internal
|
||||
|
||||
|
||||
@@ -141,7 +141,6 @@ func (s statement) QueryRowsPartialCtx(ctx context.Context, v any, args ...any)
|
||||
func (s statement) queryRows(ctx context.Context, scanFn func(any, rowsScanner) error,
|
||||
v any, args ...any) error {
|
||||
var scanFailed bool
|
||||
|
||||
err := s.brk.DoWithAcceptable(func() error {
|
||||
return queryStmt(ctx, s.stmt, func(rows *sql.Rows) error {
|
||||
err := scanFn(v, rows)
|
||||
|
||||
Reference in New Issue
Block a user