fix: validator.v9 enum flag is "oneof"

alexjomin · Dec 9, 2019 · c8d4119 · c8d4119
1 parent 24bd020
commit c8d4119
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/docparser/parser.go b/docparser/parser.go
@@ -13,6 +13,7 @@ import (
 )
 
 var enumRegex = regexp.MustCompile(`enum=([\w ]+)`)
+var oneOfRegex = regexp.MustCompile(`oneof=([\w ]+)`) // validator.v9 enum tag is oneof
 
 func parseFile(path string) (*ast.File, error) {
 	data, err := ioutil.ReadFile(path) // just pass the file name
@@ -61,6 +62,9 @@ func parseJSONTag(field *ast.Field) (j jsonTagInfo, err error) {
 					if matches := enumRegex.FindStringSubmatch(v); len(matches) > 0 {
 						j.enum = strings.Fields(matches[1])
 					}
+					if matches := oneOfRegex.FindStringSubmatch(v); len(matches) > 0 {
+						j.enum = strings.Fields(matches[1])
+					}
 				}
 
 				j.name = jsonName