diff --git a/example_test.go b/example_test.go index 26f24e7..2e9b875 100644 --- a/example_test.go +++ b/example_test.go @@ -95,17 +95,33 @@ func Example_mappings() { // output: map[john:123 mary:456] } +type commaSeparated struct { + M map[string]string +} + +func (c *commaSeparated) UnmarshalText(b []byte) error { + c.M = make(map[string]string) + for _, part := range strings.Split(string(b), ",") { + pos := strings.Index(part, "=") + if pos == -1 { + return fmt.Errorf("error parsing %q, expected format key=value", part) + } + c.M[part[:pos]] = part[pos+1:] + } + return nil +} + // This example demonstrates arguments with keys and values separated by commas -func Example_mappingsWithCommas() { +func Example_mappingWithCommas() { // The args you would pass in on the command line - os.Args = split("./example --userids john=123 mary=456") + os.Args = split("./example --values one=two,three=four") var args struct { - UserIDs map[string]int + Values commaSeparated } MustParse(&args) - fmt.Println(args.UserIDs) - // output: map[john:123 mary:456] + fmt.Println(args.Values.M) + // output: map[one:two three:four] } // This eample demonstrates multiple value arguments that can be mixed with