Change not found error to be typed with cgroup subsystem

Signed-off-by: Michael Crosby <michael@docker.com>
This commit is contained in:
Michael Crosby 2014-08-20 10:32:01 -07:00
parent 4aa9963faf
commit 3cdf12b041
8 changed files with 41 additions and 15 deletions

View File

@ -1,15 +1,11 @@
package cgroups package cgroups
import ( import (
"errors" "fmt"
"github.com/docker/libcontainer/devices" "github.com/docker/libcontainer/devices"
) )
var (
ErrNotFound = errors.New("mountpoint not found")
)
type FreezerState string type FreezerState string
const ( const (
@ -18,6 +14,29 @@ const (
Thawed FreezerState = "THAWED" Thawed FreezerState = "THAWED"
) )
type NotFoundError struct {
Subsystem string
}
func (e *NotFoundError) Error() string {
return fmt.Sprintf("mountpoint for %s not found", e.Subsystem)
}
func NewNotFoundError(sub string) error {
return &NotFoundError{
Subsystem: sub,
}
}
func IsNotFound(err error) bool {
if err == nil {
return false
}
_, ok := err.(*NotFoundError)
return ok
}
type Cgroup struct { type Cgroup struct {
Name string `json:"name,omitempty"` Name string `json:"name,omitempty"`
Parent string `json:"parent,omitempty"` // name of parent cgroup or slice Parent string `json:"parent,omitempty"` // name of parent cgroup or slice

View File

@ -76,7 +76,7 @@ func GetStats(c *cgroups.Cgroup) (*cgroups.Stats, error) {
path, err := d.path(sysname) path, err := d.path(sysname)
if err != nil { if err != nil {
// Don't fail if a cgroup hierarchy was not found, just skip this subsystem // Don't fail if a cgroup hierarchy was not found, just skip this subsystem
if err == cgroups.ErrNotFound { if cgroups.IsNotFound(err) {
continue continue
} }
@ -172,7 +172,7 @@ func (raw *data) path(subsystem string) (string, error) {
if _, err := os.Stat(path); err != nil { if _, err := os.Stat(path); err != nil {
if os.IsNotExist(err) { if os.IsNotExist(err) {
return "", cgroups.ErrNotFound return "", cgroups.NewNotFoundError(subsystem)
} }
return "", err return "", err

View File

@ -16,9 +16,10 @@ type BlkioGroup struct {
func (s *BlkioGroup) Set(d *data) error { func (s *BlkioGroup) Set(d *data) error {
// we just want to join this group even though we don't set anything // we just want to join this group even though we don't set anything
if _, err := d.join("blkio"); err != nil && err != cgroups.ErrNotFound { if _, err := d.join("blkio"); err != nil && !cgroups.IsNotFound(err) {
return err return err
} }
return nil return nil
} }

View File

@ -27,9 +27,10 @@ type CpuacctGroup struct {
func (s *CpuacctGroup) Set(d *data) error { func (s *CpuacctGroup) Set(d *data) error {
// we just want to join this group even though we don't set anything // we just want to join this group even though we don't set anything
if _, err := d.join("cpuacct"); err != nil && err != cgroups.ErrNotFound { if _, err := d.join("cpuacct"); err != nil && !cgroups.IsNotFound(err) {
return err return err
} }
return nil return nil
} }

View File

@ -33,7 +33,7 @@ func (s *FreezerGroup) Set(d *data) error {
time.Sleep(1 * time.Millisecond) time.Sleep(1 * time.Millisecond)
} }
default: default:
if _, err := d.join("freezer"); err != nil && err != cgroups.ErrNotFound { if _, err := d.join("freezer"); err != nil && !cgroups.IsNotFound(err) {
return err return err
} }
} }

View File

@ -9,7 +9,7 @@ type PerfEventGroup struct {
func (s *PerfEventGroup) Set(d *data) error { func (s *PerfEventGroup) Set(d *data) error {
// we just want to join this group even though we don't set anything // we just want to join this group even though we don't set anything
if _, err := d.join("perf_event"); err != nil && err != cgroups.ErrNotFound { if _, err := d.join("perf_event"); err != nil && !cgroups.IsNotFound(err) {
return err return err
} }
return nil return nil

View File

@ -166,7 +166,7 @@ func (c *systemdCgroup) Paths() (map[string]string, error) {
subsystemPath, err := getSubsystemPath(c.cgroup, sysname) subsystemPath, err := getSubsystemPath(c.cgroup, sysname)
if err != nil { if err != nil {
// Don't fail if a cgroup hierarchy was not found, just skip this subsystem // Don't fail if a cgroup hierarchy was not found, just skip this subsystem
if err == cgroups.ErrNotFound { if cgroups.IsNotFound(err) {
continue continue
} }
@ -274,7 +274,7 @@ func GetStats(c *cgroups.Cgroup) (*cgroups.Stats, error) {
subsystemPath, err := getSubsystemPath(c, sysname) subsystemPath, err := getSubsystemPath(c, sysname)
if err != nil { if err != nil {
// Don't fail if a cgroup hierarchy was not found, just skip this subsystem // Don't fail if a cgroup hierarchy was not found, just skip this subsystem
if err == cgroups.ErrNotFound { if cgroups.IsNotFound(err) {
continue continue
} }

View File

@ -29,7 +29,8 @@ func FindCgroupMountpoint(subsystem string) (string, error) {
} }
} }
} }
return "", ErrNotFound
return "", NewNotFoundError(subsystem)
} }
type Mount struct { type Mount struct {
@ -153,19 +154,23 @@ func ReadProcsFile(dir string) ([]int, error) {
func parseCgroupFile(subsystem string, r io.Reader) (string, error) { func parseCgroupFile(subsystem string, r io.Reader) (string, error) {
s := bufio.NewScanner(r) s := bufio.NewScanner(r)
for s.Scan() { for s.Scan() {
if err := s.Err(); err != nil { if err := s.Err(); err != nil {
return "", err return "", err
} }
text := s.Text() text := s.Text()
parts := strings.Split(text, ":") parts := strings.Split(text, ":")
for _, subs := range strings.Split(parts[1], ",") { for _, subs := range strings.Split(parts[1], ",") {
if subs == subsystem { if subs == subsystem {
return parts[2], nil return parts[2], nil
} }
} }
} }
return "", ErrNotFound
return "", NewNotFoundError(subsystem)
} }
func pathExists(path string) bool { func pathExists(path string) bool {