优化代码

This commit is contained in:
GoEdgeLab
2022-03-19 12:17:28 +08:00
parent fa044e29a1
commit f4fd243d1d
3 changed files with 15 additions and 12 deletions

View File

@@ -1,3 +1,4 @@
//go:build darwin
// +build darwin // +build darwin
package utils package utils
@@ -6,9 +7,9 @@ import (
"syscall" "syscall"
) )
// set resource limit // SetRLimit set resource limit
func SetRLimit(limit uint64) error { func SetRLimit(limit uint64) error {
rLimit := &syscall.Rlimit{} var rLimit = &syscall.Rlimit{}
err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, rLimit) err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, rLimit)
if err != nil { if err != nil {
return err return err
@@ -23,7 +24,7 @@ func SetRLimit(limit uint64) error {
return syscall.Setrlimit(syscall.RLIMIT_NOFILE, rLimit) return syscall.Setrlimit(syscall.RLIMIT_NOFILE, rLimit)
} }
// set best resource limit value // SetSuitableRLimit set best resource limit value
func SetSuitableRLimit() { func SetSuitableRLimit() error {
SetRLimit(4096 * 100) // 1M=100Files return SetRLimit(4096 * 100) // 1M=100Files
} }

View File

@@ -1,3 +1,4 @@
//go:build linux
// +build linux // +build linux
package utils package utils
@@ -6,9 +7,9 @@ import (
"syscall" "syscall"
) )
// set resource limit // SetRLimit set resource limit
func SetRLimit(limit uint64) error { func SetRLimit(limit uint64) error {
rLimit := &syscall.Rlimit{} var rLimit = &syscall.Rlimit{}
err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, rLimit) err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, rLimit)
if err != nil { if err != nil {
return err return err
@@ -23,7 +24,7 @@ func SetRLimit(limit uint64) error {
return syscall.Setrlimit(syscall.RLIMIT_NOFILE, rLimit) return syscall.Setrlimit(syscall.RLIMIT_NOFILE, rLimit)
} }
// set best resource limit value // SetSuitableRLimit set best resource limit value
func SetSuitableRLimit() { func SetSuitableRLimit() error {
SetRLimit(4096 * 100) // 1M=100Files return SetRLimit(4096 * 100) // 1M=100Files
} }

View File

@@ -1,3 +1,4 @@
//go:build !linux && !darwin
// +build !linux,!darwin // +build !linux,!darwin
package utils package utils
@@ -8,6 +9,6 @@ func SetRLimit(limit uint64) error {
} }
// set best resource limit value // set best resource limit value
func SetSuitableRLimit() { func SetSuitableRLimit() error {
return nil
} }