-
Notifications
You must be signed in to change notification settings - Fork 6
/
rlimit.go
32 lines (24 loc) · 662 Bytes
/
rlimit.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
package main
import (
"flag"
"log"
"syscall"
)
var maxConnections = flag.Int("max_connections", getRlimitMax(syscall.RLIMIT_NOFILE), "The maximum number of incoming connections allowed.")
func getRlimitMax(resource int) int {
var rlimit syscall.Rlimit
if err := syscall.Getrlimit(resource, &rlimit); err != nil {
return 0
}
return int(rlimit.Max)
}
func setRlimit(resource, value int) {
rlimit := &syscall.Rlimit{Cur: uint64(value), Max: uint64(value)}
err := syscall.Setrlimit(resource, rlimit)
if err != nil {
log.Fatalln("Error Setting Rlimit ", err)
}
}
func setRlimitFromFlags() {
setRlimit(syscall.RLIMIT_NOFILE, *maxConnections)
}