lbsadmin/bee
1
0
Fork 0
mirror of https://github.com/beego/bee.git synced 2025-10-31 17:33:26 +00:00
Code Issues Releases Activity

[g_docs] Redefine 'swagger.Definitions'

Browse Source 
Signed-off-by: weiyang <weiyang.ones@gmail.com>
This commit is contained in:
weiyang
2017-08-21 18:22:49 +08:00
parent ebbc704811
commit 27ea6dfce9
1 changed files with 8 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
generate/swaggergen/g_docs.go
View File

@@ -47,28 +47,21 @@ const (
aform = "multipart/form-data" aform = "multipart/form-data"
) )
// Swagger redefines definitions
type Swagger struct {
swagger.Swagger
Definitions spec.Definitions `json:"definitions,omitempty" yaml:"definitions,omitempty"`
}
var pkgCache map[string]struct{} //pkg:controller:function:comments comments: key:value var pkgCache map[string]struct{} //pkg:controller:function:comments comments: key:value
var controllerComments map[string]string var controllerComments map[string]string
var importlist map[string]string var importlist map[string]string
var controllerList map[string]map[string]*swagger.Item //controllername Paths items var controllerList map[string]map[string]*swagger.Item //controllername Paths items
var rootapi swagger.Swagger var rootapi Swagger
var tparser *tspec.Parser var tparser *tspec.Parser
var controllerPkg *ast.Package var controllerPkg *ast.Package
func convertSpecDefinitions(specDefs spec.Definitions) (defs map[string]swagger.Schema, err error) {
bytes, err := json.Marshal(specDefs)
if err != nil {
return
}
defs = make(map[string]swagger.Schema)
err = json.Unmarshal(bytes, &defs)
if err != nil {
return
}
return
}
func parseModel(pkg *ast.Package, typeStr string) (typeTitle string, err error) { func parseModel(pkg *ast.Package, typeStr string) (typeTitle string, err error) {
if pkg == nil { if pkg == nil {
panic("pkg can not be nil") panic("pkg can not be nil")
@@ -278,12 +271,7 @@ func GenerateDocs(curpath string) {
} }
} }
} }
defs, err := convertSpecDefinitions(tparser.Definitions()) rootapi.Definitions = tparser.Definitions()
if err != nil {
msg := fmt.Sprintf("failed to convert spec definitions into swagger definitions: %s", err)
panic(msg)
}
rootapi.Definitions = defs
os.Mkdir(path.Join(curpath, "swagger"), 0755) os.Mkdir(path.Join(curpath, "swagger"), 0755)
fd, err := os.Create(path.Join(curpath, "swagger", "swagger.json")) fd, err := os.Create(path.Join(curpath, "swagger", "swagger.json"))