123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111 |
- package opts
- import (
- "fmt"
- "strconv"
- "strings"
- "github.com/docker/docker/api/types/blkiodev"
- "github.com/docker/go-units"
- )
- // ValidatorThrottleFctType defines a validator function that returns a validated struct and/or an error.
- type ValidatorThrottleFctType func(val string) (*blkiodev.ThrottleDevice, error)
- // ValidateThrottleBpsDevice validates that the specified string has a valid device-rate format.
- func ValidateThrottleBpsDevice(val string) (*blkiodev.ThrottleDevice, error) {
- split := strings.SplitN(val, ":", 2)
- if len(split) != 2 {
- return nil, fmt.Errorf("bad format: %s", val)
- }
- if !strings.HasPrefix(split[0], "/dev/") {
- return nil, fmt.Errorf("bad format for device path: %s", val)
- }
- rate, err := units.RAMInBytes(split[1])
- if err != nil {
- return nil, fmt.Errorf("invalid rate for device: %s. The correct format is <device-path>:<number>[<unit>]. Number must be a positive integer. Unit is optional and can be kb, mb, or gb", val)
- }
- if rate < 0 {
- return nil, fmt.Errorf("invalid rate for device: %s. The correct format is <device-path>:<number>[<unit>]. Number must be a positive integer. Unit is optional and can be kb, mb, or gb", val)
- }
- return &blkiodev.ThrottleDevice{
- Path: split[0],
- Rate: uint64(rate),
- }, nil
- }
- // ValidateThrottleIOpsDevice validates that the specified string has a valid device-rate format.
- func ValidateThrottleIOpsDevice(val string) (*blkiodev.ThrottleDevice, error) {
- split := strings.SplitN(val, ":", 2)
- if len(split) != 2 {
- return nil, fmt.Errorf("bad format: %s", val)
- }
- if !strings.HasPrefix(split[0], "/dev/") {
- return nil, fmt.Errorf("bad format for device path: %s", val)
- }
- rate, err := strconv.ParseUint(split[1], 10, 64)
- if err != nil {
- return nil, fmt.Errorf("invalid rate for device: %s. The correct format is <device-path>:<number>. Number must be a positive integer", val)
- }
- if rate < 0 {
- return nil, fmt.Errorf("invalid rate for device: %s. The correct format is <device-path>:<number>. Number must be a positive integer", val)
- }
- return &blkiodev.ThrottleDevice{
- Path: split[0],
- Rate: uint64(rate),
- }, nil
- }
- // ThrottledeviceOpt defines a map of ThrottleDevices
- type ThrottledeviceOpt struct {
- values []*blkiodev.ThrottleDevice
- validator ValidatorThrottleFctType
- }
- // NewThrottledeviceOpt creates a new ThrottledeviceOpt
- func NewThrottledeviceOpt(validator ValidatorThrottleFctType) ThrottledeviceOpt {
- values := []*blkiodev.ThrottleDevice{}
- return ThrottledeviceOpt{
- values: values,
- validator: validator,
- }
- }
- // Set validates a ThrottleDevice and sets its name as a key in ThrottledeviceOpt
- func (opt *ThrottledeviceOpt) Set(val string) error {
- var value *blkiodev.ThrottleDevice
- if opt.validator != nil {
- v, err := opt.validator(val)
- if err != nil {
- return err
- }
- value = v
- }
- (opt.values) = append((opt.values), value)
- return nil
- }
- // String returns ThrottledeviceOpt values as a string.
- func (opt *ThrottledeviceOpt) String() string {
- var out []string
- for _, v := range opt.values {
- out = append(out, v.String())
- }
- return fmt.Sprintf("%v", out)
- }
- // GetList returns a slice of pointers to ThrottleDevices.
- func (opt *ThrottledeviceOpt) GetList() []*blkiodev.ThrottleDevice {
- var throttledevice []*blkiodev.ThrottleDevice
- throttledevice = append(throttledevice, opt.values...)
- return throttledevice
- }
- // Type returns the option type
- func (opt *ThrottledeviceOpt) Type() string {
- return "list"
- }
|