30 lines
514 B
Go
30 lines
514 B
Go
|
package kvFlag
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Flag is a flag.Value implementation for parsing user variables
|
||
|
|
// from the command-line in the format of '-var key=value'.
|
||
|
|
type Flag map[string]string
|
||
|
|
|
||
|
|
func (v *Flag) String() string {
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
|
||
|
|
func (v *Flag) Set(raw string) error {
|
||
|
|
idx := strings.Index(raw, "=")
|
||
|
|
if idx == -1 {
|
||
|
|
return fmt.Errorf("No '=' value in arg: %s", raw)
|
||
|
|
}
|
||
|
|
|
||
|
|
if *v == nil {
|
||
|
|
*v = make(map[string]string)
|
||
|
|
}
|
||
|
|
|
||
|
|
key, value := raw[0:idx], raw[idx+1:]
|
||
|
|
(*v)[key] = value
|
||
|
|
return nil
|
||
|
|
}
|