refactor: move boomer from hrp internal to pkg

This commit is contained in:
debugtalk
2022-10-11 11:17:58 +08:00
parent c81c5841ee
commit 0a0700dda8
41 changed files with 41 additions and 38 deletions

33
hrp/pkg/boomer/ulimit.go Normal file
View File

@@ -0,0 +1,33 @@
//go:build !windows
package boomer
import (
"syscall"
"github.com/rs/zerolog/log"
)
// set resource limit
// ulimit -n 10240
func SetUlimit(limit uint64) {
var rLimit syscall.Rlimit
err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, &rLimit)
if err != nil {
log.Error().Err(err).Msg("get ulimit failed")
return
}
log.Info().Uint64("limit", rLimit.Cur).Msg("get current ulimit")
if rLimit.Cur >= limit {
return
}
rLimit.Cur = limit
rLimit.Max = limit
log.Info().Uint64("limit", rLimit.Cur).Msg("set current ulimit")
err = syscall.Setrlimit(syscall.RLIMIT_NOFILE, &rLimit)
if err != nil {
log.Error().Err(err).Msg("set ulimit failed")
return
}
}