Remove package pkg/ulimit, use go-units instead.
Signed-off-by: Daniel Nephin <dnephin@docker.com>
This commit is contained in:
parent
c2c7305180
commit
d2d85c6d52
2 changed files with 0 additions and 166 deletions
111
ulimit/ulimit.go
111
ulimit/ulimit.go
|
@ -1,111 +0,0 @@
|
||||||
// Package ulimit provides structure and helper function to parse and represent
|
|
||||||
// resource limits (Rlimit and Ulimit, its human friendly version).
|
|
||||||
package ulimit
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Ulimit is a human friendly version of Rlimit.
|
|
||||||
type Ulimit struct {
|
|
||||||
Name string
|
|
||||||
Hard int64
|
|
||||||
Soft int64
|
|
||||||
}
|
|
||||||
|
|
||||||
// Rlimit specifies the resource limits, such as max open files.
|
|
||||||
type Rlimit struct {
|
|
||||||
Type int `json:"type,omitempty"`
|
|
||||||
Hard uint64 `json:"hard,omitempty"`
|
|
||||||
Soft uint64 `json:"soft,omitempty"`
|
|
||||||
}
|
|
||||||
|
|
||||||
const (
|
|
||||||
// magic numbers for making the syscall
|
|
||||||
// some of these are defined in the syscall package, but not all.
|
|
||||||
// Also since Windows client doesn't get access to the syscall package, need to
|
|
||||||
// define these here
|
|
||||||
rlimitAs = 9
|
|
||||||
rlimitCore = 4
|
|
||||||
rlimitCPU = 0
|
|
||||||
rlimitData = 2
|
|
||||||
rlimitFsize = 1
|
|
||||||
rlimitLocks = 10
|
|
||||||
rlimitMemlock = 8
|
|
||||||
rlimitMsgqueue = 12
|
|
||||||
rlimitNice = 13
|
|
||||||
rlimitNofile = 7
|
|
||||||
rlimitNproc = 6
|
|
||||||
rlimitRss = 5
|
|
||||||
rlimitRtprio = 14
|
|
||||||
rlimitRttime = 15
|
|
||||||
rlimitSigpending = 11
|
|
||||||
rlimitStack = 3
|
|
||||||
)
|
|
||||||
|
|
||||||
var ulimitNameMapping = map[string]int{
|
|
||||||
//"as": rlimitAs, // Disabled since this doesn't seem usable with the way Docker inits a container.
|
|
||||||
"core": rlimitCore,
|
|
||||||
"cpu": rlimitCPU,
|
|
||||||
"data": rlimitData,
|
|
||||||
"fsize": rlimitFsize,
|
|
||||||
"locks": rlimitLocks,
|
|
||||||
"memlock": rlimitMemlock,
|
|
||||||
"msgqueue": rlimitMsgqueue,
|
|
||||||
"nice": rlimitNice,
|
|
||||||
"nofile": rlimitNofile,
|
|
||||||
"nproc": rlimitNproc,
|
|
||||||
"rss": rlimitRss,
|
|
||||||
"rtprio": rlimitRtprio,
|
|
||||||
"rttime": rlimitRttime,
|
|
||||||
"sigpending": rlimitSigpending,
|
|
||||||
"stack": rlimitStack,
|
|
||||||
}
|
|
||||||
|
|
||||||
// Parse parses and returns a Ulimit from the specified string.
|
|
||||||
func Parse(val string) (*Ulimit, error) {
|
|
||||||
parts := strings.SplitN(val, "=", 2)
|
|
||||||
if len(parts) != 2 {
|
|
||||||
return nil, fmt.Errorf("invalid ulimit argument: %s", val)
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, exists := ulimitNameMapping[parts[0]]; !exists {
|
|
||||||
return nil, fmt.Errorf("invalid ulimit type: %s", parts[0])
|
|
||||||
}
|
|
||||||
|
|
||||||
limitVals := strings.SplitN(parts[1], ":", 2)
|
|
||||||
if len(limitVals) > 2 {
|
|
||||||
return nil, fmt.Errorf("too many limit value arguments - %s, can only have up to two, `soft[:hard]`", parts[1])
|
|
||||||
}
|
|
||||||
|
|
||||||
soft, err := strconv.ParseInt(limitVals[0], 10, 64)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
hard := soft // in case no hard was set
|
|
||||||
if len(limitVals) == 2 {
|
|
||||||
hard, err = strconv.ParseInt(limitVals[1], 10, 64)
|
|
||||||
}
|
|
||||||
if soft > hard {
|
|
||||||
return nil, fmt.Errorf("ulimit soft limit must be less than or equal to hard limit: %d > %d", soft, hard)
|
|
||||||
}
|
|
||||||
|
|
||||||
return &Ulimit{Name: parts[0], Soft: soft, Hard: hard}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// GetRlimit returns the RLimit corresponding to Ulimit.
|
|
||||||
func (u *Ulimit) GetRlimit() (*Rlimit, error) {
|
|
||||||
t, exists := ulimitNameMapping[u.Name]
|
|
||||||
if !exists {
|
|
||||||
return nil, fmt.Errorf("invalid ulimit name %s", u.Name)
|
|
||||||
}
|
|
||||||
|
|
||||||
return &Rlimit{Type: t, Soft: uint64(u.Soft), Hard: uint64(u.Hard)}, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (u *Ulimit) String() string {
|
|
||||||
return fmt.Sprintf("%s=%d:%d", u.Name, u.Soft, u.Hard)
|
|
||||||
}
|
|
|
@ -1,55 +0,0 @@
|
||||||
package ulimit
|
|
||||||
|
|
||||||
import "testing"
|
|
||||||
|
|
||||||
func TestParseValid(t *testing.T) {
|
|
||||||
u1 := &Ulimit{"nofile", 1024, 512}
|
|
||||||
if u2, _ := Parse("nofile=512:1024"); *u1 != *u2 {
|
|
||||||
t.Fatalf("expected %q, but got %q", u1, u2)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestParseInvalidLimitType(t *testing.T) {
|
|
||||||
if _, err := Parse("notarealtype=1024:1024"); err == nil {
|
|
||||||
t.Fatalf("expected error on invalid ulimit type")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestParseBadFormat(t *testing.T) {
|
|
||||||
if _, err := Parse("nofile:1024:1024"); err == nil {
|
|
||||||
t.Fatal("expected error on bad syntax")
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := Parse("nofile"); err == nil {
|
|
||||||
t.Fatal("expected error on bad syntax")
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := Parse("nofile="); err == nil {
|
|
||||||
t.Fatal("expected error on bad syntax")
|
|
||||||
}
|
|
||||||
if _, err := Parse("nofile=:"); err == nil {
|
|
||||||
t.Fatal("expected error on bad syntax")
|
|
||||||
}
|
|
||||||
if _, err := Parse("nofile=:1024"); err == nil {
|
|
||||||
t.Fatal("expected error on bad syntax")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestParseHardLessThanSoft(t *testing.T) {
|
|
||||||
if _, err := Parse("nofile:1024:1"); err == nil {
|
|
||||||
t.Fatal("expected error on hard limit less than soft limit")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestParseInvalidValueType(t *testing.T) {
|
|
||||||
if _, err := Parse("nofile:asdf"); err == nil {
|
|
||||||
t.Fatal("expected error on bad value type")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestStringOutput(t *testing.T) {
|
|
||||||
u := &Ulimit{"nofile", 1024, 512}
|
|
||||||
if s := u.String(); s != "nofile=512:1024" {
|
|
||||||
t.Fatal("expected String to return nofile=512:1024, but got", s)
|
|
||||||
}
|
|
||||||
}
|
|
Loading…
Add table
Reference in a new issue