github.com/kunnos/engine@v1.13.1/runconfig/opts/weightdevice.go (about)

     1  package opts
     2  
     3  import (
     4  	"fmt"
     5  	"strconv"
     6  	"strings"
     7  
     8  	"github.com/docker/docker/api/types/blkiodev"
     9  )
    10  
    11  // ValidatorWeightFctType defines a validator function that returns a validated struct and/or an error.
    12  type ValidatorWeightFctType func(val string) (*blkiodev.WeightDevice, error)
    13  
    14  // ValidateWeightDevice validates that the specified string has a valid device-weight format.
    15  func ValidateWeightDevice(val string) (*blkiodev.WeightDevice, error) {
    16  	split := strings.SplitN(val, ":", 2)
    17  	if len(split) != 2 {
    18  		return nil, fmt.Errorf("bad format: %s", val)
    19  	}
    20  	if !strings.HasPrefix(split[0], "/dev/") {
    21  		return nil, fmt.Errorf("bad format for device path: %s", val)
    22  	}
    23  	weight, err := strconv.ParseUint(split[1], 10, 0)
    24  	if err != nil {
    25  		return nil, fmt.Errorf("invalid weight for device: %s", val)
    26  	}
    27  	if weight > 0 && (weight < 10 || weight > 1000) {
    28  		return nil, fmt.Errorf("invalid weight for device: %s", val)
    29  	}
    30  
    31  	return &blkiodev.WeightDevice{
    32  		Path:   split[0],
    33  		Weight: uint16(weight),
    34  	}, nil
    35  }
    36  
    37  // WeightdeviceOpt defines a map of WeightDevices
    38  type WeightdeviceOpt struct {
    39  	values    []*blkiodev.WeightDevice
    40  	validator ValidatorWeightFctType
    41  }
    42  
    43  // NewWeightdeviceOpt creates a new WeightdeviceOpt
    44  func NewWeightdeviceOpt(validator ValidatorWeightFctType) WeightdeviceOpt {
    45  	values := []*blkiodev.WeightDevice{}
    46  	return WeightdeviceOpt{
    47  		values:    values,
    48  		validator: validator,
    49  	}
    50  }
    51  
    52  // Set validates a WeightDevice and sets its name as a key in WeightdeviceOpt
    53  func (opt *WeightdeviceOpt) Set(val string) error {
    54  	var value *blkiodev.WeightDevice
    55  	if opt.validator != nil {
    56  		v, err := opt.validator(val)
    57  		if err != nil {
    58  			return err
    59  		}
    60  		value = v
    61  	}
    62  	(opt.values) = append((opt.values), value)
    63  	return nil
    64  }
    65  
    66  // String returns WeightdeviceOpt values as a string.
    67  func (opt *WeightdeviceOpt) String() string {
    68  	var out []string
    69  	for _, v := range opt.values {
    70  		out = append(out, v.String())
    71  	}
    72  
    73  	return fmt.Sprintf("%v", out)
    74  }
    75  
    76  // GetList returns a slice of pointers to WeightDevices.
    77  func (opt *WeightdeviceOpt) GetList() []*blkiodev.WeightDevice {
    78  	var weightdevice []*blkiodev.WeightDevice
    79  	for _, v := range opt.values {
    80  		weightdevice = append(weightdevice, v)
    81  	}
    82  
    83  	return weightdevice
    84  }
    85  
    86  // Type returns the option type
    87  func (opt *WeightdeviceOpt) Type() string {
    88  	return "weighted-device"
    89  }