devttl/goctl-swagger
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

support api prefix (#22) patch method (#21)

Browse Source
This commit is contained in:
MaxToby
2022-02-15 21:22:54 +08:00
parent bd6092ac10
commit 0aac4b39f3
7 changed files with 68 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
generate/entities.go
View File

@ -8,26 +8,24 @@ import (
"github.com/grpc-ecosystem/grpc-gateway/protoc-gen-grpc-gateway/descriptor"
)
var (
swaggerMapTypes = map[string]reflect.Kind{
"string": reflect.String,
"int": reflect.Int,
"int32": reflect.Int,
"uint32": reflect.Int,
"uint64": reflect.Int64,
"int64": reflect.Int64,
"[]string": reflect.Slice,
"[]int": reflect.Slice,
"[]int64": reflect.Slice,
"[]int32": reflect.Slice,
"[]uint32": reflect.Slice,
"[]uint64": reflect.Slice,
"bool": reflect.Bool,
"struct": reflect.Struct,
"float32": reflect.Float32,
"float64": reflect.Float64,
}
)
var swaggerMapTypes = map[string]reflect.Kind{
"string": reflect.String,
"int": reflect.Int,
"int32": reflect.Int,
"uint32": reflect.Int,
"uint64": reflect.Int64,
"int64": reflect.Int64,
"[]string": reflect.Slice,
"[]int": reflect.Slice,
"[]int64": reflect.Slice,
"[]int32": reflect.Slice,
"[]uint32": reflect.Slice,
"[]uint64": reflect.Slice,
"bool": reflect.Bool,
"struct": reflect.Struct,
"float32": reflect.Float32,
"float64": reflect.Float64,
}
// http://swagger.io/specification/#infoObject
type swaggerInfoObject struct {
@ -116,14 +114,20 @@ type swaggerOperationObject struct {
OperationID string `json:"operationId"`
Responses swaggerResponsesObject `json:"responses"`
Parameters swaggerParametersObject `json:"parameters,omitempty"`
Tags []string `json:"tags,omitempty"`
Deprecated bool `json:"deprecated,omitempty"`
RequestBody struct {
Content swaggerContentObject `json:"content,omitempty"`
} `json:"requestBody,omitempty"`
Tags []string `json:"tags,omitempty"`
Deprecated bool `json:"deprecated,omitempty"`
Security *[]swaggerSecurityRequirementObject `json:"security,omitempty"`
ExternalDocs *swaggerExternalDocumentationObject `json:"externalDocs,omitempty"`
}
type swaggerParametersObject []swaggerParameterObject
type (
swaggerParametersObject []swaggerParameterObject
swaggerContentObject map[string]swaggerParametersObject
)
// http://swagger.io/specification/#parameterObject
type swaggerParameterObject struct {

18
generate/parser.go
View File

@ -73,7 +73,8 @@ func applyGenerate(p *plugin.Plugin, host string, basePath string) (*swaggerObje
func renderServiceRoutes(service spec.Service, groups []spec.Group, paths swaggerPathsObject, requestResponseRefs refMap) {
for _, group := range groups {
for _, route := range group.Routes {
path := route.Path
path := group.GetAnnotation("prefix") + route.Path
parameters := swaggerParametersObject{}
if countParams(path) > 0 {
@ -163,12 +164,21 @@ func renderServiceRoutes(service spec.Service, groups []spec.Group, paths swagge
Ref: reqRef,
},
}
parameters = append(parameters, swaggerParameterObject{
parameter := swaggerParameterObject{
Name: "body",
In: "body",
Required: true,
Schema: &schema,
})
}
doc := strings.Join(route.RequestType.Documents(), ",")
doc = strings.Replace(doc, "//", "", -1)
if doc != "" {
parameter.Description = doc
}
parameters = append(parameters, parameter)
}
}
}
@ -232,6 +242,8 @@ func renderServiceRoutes(service spec.Service, groups []spec.Group, paths swagge
pathItemObject.Delete = operationObject
case http.MethodPut:
pathItemObject.Put = operationObject
case http.MethodPatch:
pathItemObject.Put = operationObject
}
paths[path] = pathItemObject