diff --git a/util/tool.go b/util/tool.go index 53108c1..4419a44 100644 --- a/util/tool.go +++ b/util/tool.go @@ -50,15 +50,41 @@ func GetParameterDefaultLocation(requestMethod string) string { // // Date : 12:23 2024/4/22 func GetSwaggerType(inputType string) string { - convertTable := map[string]string{ - consts.DataTypeString.String(): "string", - consts.DataTypeInt.String(): "integer", - consts.DataTypeUint.String(): "integer", - consts.DataTypeFloat.String(): "number", - consts.DataTypeBool.String(): "boolean", + for _, itemType := range consts.DataTypeBaseInt { + if itemType.String() == inputType { + return consts.SwaggerDataTypeInteger + } } - if _, exist := convertTable[inputType]; exist { - return convertTable[inputType] + for _, itemType := range consts.DataTypeBaseUint { + if itemType.String() == inputType { + return consts.SwaggerDataTypeInteger + } + } + for _, itemType := range consts.DataTypeBaseFloat { + if itemType.String() == inputType { + return consts.SwaggerDataTypeDouble + } + } + for _, itemType := range consts.DataTypeBaseString { + if itemType.String() == inputType { + return consts.SwaggerDataTypeString + } + } + for _, itemType := range consts.DataTypeBaseString { + if itemType.String() == inputType { + return consts.SwaggerDataTypeString + } + } + for _, itemType := range consts.DataTypeBaseBool { + if itemType.String() == inputType { + return consts.SwaggerDataTypeBoolean + } + } + if strings.HasPrefix(inputType, "[]") { + return consts.SwaggerDataTypeArray + } + if strings.HasPrefix(inputType, "map") { + return consts.SwaggerDataTypeObject } return inputType }