Add parsing for max and min length for response body

parvez3019 · Nov 4, 2023 · 996128f · 996128f
1 parent a508fe1
commit 996128f
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 0 deletions.
diff --git a/main b/main
diff --git a/openApi3Schema/oas.go b/openApi3Schema/oas.go
@@ -140,6 +140,8 @@ type SchemaObject struct {
 	Deprecated         bool                   `json:"deprecated,omitempty"`
 	Ref                string                 `json:"$ref,omitempty"` // Ref is used when SchemaObject is as a ReferenceObject
 	Enum               interface{}            `json:"enum,omitempty"`
+	MaxLength          uint                   `json:"maxLength,omitempty"`
+	MinLength          uint                   `json:"minLength,omitempty"`
 
 	// Title
 	// MultipleOf

diff --git a/parser/schema/custom_type_parser.go b/parser/schema/custom_type_parser.go
@@ -318,6 +318,14 @@ astFieldsLoop:
 			if enumValues := astFieldTag.Get("enum"); enumValues != "" {
 				fieldSchema.Enum = parseEnumValues(enumValues)
 			}
+
+			if maxLength := astFieldTag.Get("maxLength"); maxLength != "" {
+				fieldSchema.MaxLength = parseMaxLength(maxLength)
+			}
+
+			if minLength := astFieldTag.Get("minLength"); minLength != "" {
+				fieldSchema.MinLength = parseMinLength(minLength)
+			}
 		}
 
 		structSchema.Properties.Set(name, fieldSchema)
@@ -452,6 +460,24 @@ func parseEnumValues(enumString string) interface{} {
 	return result
 }
 
+func parseMaxLength(maxLengthString string) uint {
+	value, err := strconv.ParseUint(maxLengthString, 10, 64)
+	if err != nil {
+		value = 0
+	}
+
+	return uint(value)
+}
+
+func parseMinLength(minLengthString string) uint {
+	value, err := strconv.ParseUint(minLengthString, 10, 64)
+	if err != nil {
+		value = 0
+	}
+
+	return uint(value)
+}
+
 type DECL struct {
 	Type struct {
 		Name string