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

feature: 支持指定schemes参数

Browse Source
This commit is contained in:
waouooo
2023-11-19 12:39:59 +08:00
parent c76c668b5f
commit f5343a24b7
5 changed files with 25 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
generate/generate.go
View File

@ -9,8 +9,8 @@ import (
"github.com/zeromicro/go-zero/tools/goctl/plugin"
)
func Do(filename string, host string, basePath string, in *plugin.Plugin) error {
swagger, err := applyGenerate(in, host, basePath)
func Do(filename string, host string, basePath string, schemes string, in *plugin.Plugin) error {
swagger, err := applyGenerate(in, host, basePath, schemes)
if err != nil {
fmt.Println(err)
}

16
generate/parser.go
View File

@ -4,6 +4,7 @@ import (
"bytes"
"encoding/json"
"fmt"
"log"
"net/http"
"reflect"
"regexp"
@ -53,7 +54,7 @@ func parseRangeOption(option string) (float64, float64, bool) {
return min, max, true
}
func applyGenerate(p *plugin.Plugin, host string, basePath string) (*swaggerObject, error) {
func applyGenerate(p *plugin.Plugin, host string, basePath string, schemes string) (*swaggerObject, error) {
title, _ := strconv.Unquote(p.Api.Info.Properties["title"])
version, _ := strconv.Unquote(p.Api.Info.Properties["version"])
desc, _ := strconv.Unquote(p.Api.Info.Properties["desc"])
@ -79,6 +80,19 @@ func applyGenerate(p *plugin.Plugin, host string, basePath string) (*swaggerObje
s.BasePath = basePath
}
if len(schemes) > 0 {
supportedSchemes := []string{"http", "https", "ws", "wss"}
ss := strings.Split(schemes, ",")
for i := range ss {
scheme := ss[i]
scheme = strings.TrimSpace(scheme)
if !contains(supportedSchemes, scheme) {
log.Fatalf("unsupport scheme: [%s], only support [http, https, ws, wss]", scheme)
}
ss[i] = scheme
}
s.Schemes = ss
}
s.SecurityDefinitions = swaggerSecurityDefinitionsObject{}
newSecDefValue := swaggerSecuritySchemeObject{}
newSecDefValue.Name = "Authorization"