2015-12-21 21:34:05 +00:00
package opts
import (
"fmt"
"strconv"
"strings"
2016-09-06 18:18:12 +00:00
"github.com/docker/docker/api/types/blkiodev"
2015-12-21 21:34:05 +00:00
"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
for _ , v := range opt . values {
throttledevice = append ( throttledevice , v )
}
return throttledevice
}
2016-05-31 23:49:32 +00:00
// Type returns the option type
func ( opt * ThrottledeviceOpt ) Type ( ) string {
return "throttled-device"
}