package opts import ( "fmt" "strconv" "strings" "github.com/docker/docker/api/types/blkiodev" ) // ValidatorWeightFctType defines a validator function that returns a validated struct and/or an error. type ValidatorWeightFctType func(val string) (*blkiodev.WeightDevice, error) // ValidateWeightDevice validates that the specified string has a valid device-weight format. func ValidateWeightDevice(val string) (*blkiodev.WeightDevice, error) { k, v, ok := strings.Cut(val, ":") if !ok || k == "" { return nil, fmt.Errorf("bad format: %s", val) } // TODO(thaJeztah): should we really validate this on the client? if !strings.HasPrefix(k, "/dev/") { return nil, fmt.Errorf("bad format for device path: %s", val) } weight, err := strconv.ParseUint(v, 10, 16) if err != nil { return nil, fmt.Errorf("invalid weight for device: %s", val) } if weight > 0 && (weight < 10 || weight > 1000) { return nil, fmt.Errorf("invalid weight for device: %s", val) } return &blkiodev.WeightDevice{ Path: k, Weight: uint16(weight), }, nil } // WeightdeviceOpt defines a map of WeightDevices type WeightdeviceOpt struct { values []*blkiodev.WeightDevice validator ValidatorWeightFctType } // NewWeightdeviceOpt creates a new WeightdeviceOpt func NewWeightdeviceOpt(validator ValidatorWeightFctType) WeightdeviceOpt { return WeightdeviceOpt{ values: []*blkiodev.WeightDevice{}, validator: validator, } } // Set validates a WeightDevice and sets its name as a key in WeightdeviceOpt func (opt *WeightdeviceOpt) Set(val string) error { var value *blkiodev.WeightDevice 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 WeightdeviceOpt values as a string. func (opt *WeightdeviceOpt) 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 WeightDevices. func (opt *WeightdeviceOpt) GetList() []*blkiodev.WeightDevice { return opt.values } // Type returns the option type func (opt *WeightdeviceOpt) Type() string { return "list" }