92 lines
2.7 KiB
Go
92 lines
2.7 KiB
Go
|
package models
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"encoding/json"
|
|||
|
|
"fmt"
|
|||
|
|
"io/ioutil"
|
|||
|
|
"regexp"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"github.com/bitly/go-simplejson"
|
|||
|
|
"github.com/go-admin-team/go-admin-core/sdk"
|
|||
|
|
"github.com/go-admin-team/go-admin-core/sdk/runtime"
|
|||
|
|
"github.com/go-admin-team/go-admin-core/storage"
|
|||
|
|
|
|||
|
|
"go-admin/common/models"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type SysApi struct {
|
|||
|
|
Id int `json:"id" gorm:"primaryKey;autoIncrement;comment:主键编码"`
|
|||
|
|
Handle string `json:"handle" gorm:"size:128;comment:handle"`
|
|||
|
|
Title string `json:"title" gorm:"size:128;comment:标题"`
|
|||
|
|
Path string `json:"path" gorm:"size:128;comment:地址"`
|
|||
|
|
Action string `json:"action" gorm:"size:16;comment:请求类型"`
|
|||
|
|
Type string `json:"type" gorm:"size:16;comment:接口类型"`
|
|||
|
|
models.ModelTime
|
|||
|
|
models.ControlBy
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (*SysApi) TableName() string {
|
|||
|
|
return "sys_api"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (e *SysApi) Generate() models.ActiveRecord {
|
|||
|
|
o := *e
|
|||
|
|
return &o
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func (e *SysApi) GetId() interface{} {
|
|||
|
|
return e.Id
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func SaveSysApi(message storage.Messager) (err error) {
|
|||
|
|
var rb []byte
|
|||
|
|
rb, err = json.Marshal(message.GetValues())
|
|||
|
|
if err != nil {
|
|||
|
|
err = fmt.Errorf("json Marshal error, %v", err.Error())
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var l runtime.Routers
|
|||
|
|
err = json.Unmarshal(rb, &l)
|
|||
|
|
if err != nil {
|
|||
|
|
err = fmt.Errorf("json Unmarshal error, %s", err.Error())
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
dbList := sdk.Runtime.GetDb()
|
|||
|
|
for _, d := range dbList {
|
|||
|
|
for _, v := range l.List {
|
|||
|
|
if v.HttpMethod != "HEAD" ||
|
|||
|
|
strings.Contains(v.RelativePath, "/swagger/") ||
|
|||
|
|
strings.Contains(v.RelativePath, "/static/") ||
|
|||
|
|
strings.Contains(v.RelativePath, "/form-generator/") ||
|
|||
|
|
strings.Contains(v.RelativePath, "/sys/tables") {
|
|||
|
|
|
|||
|
|
// 根据接口方法注释里的@Summary填充接口名称，适用于代码生成器
|
|||
|
|
// 可在此处增加配置路径前缀的if判断，只对代码生成的自建应用进行定向的接口名称填充
|
|||
|
|
jsonFile, _ := ioutil.ReadFile("docs/swagger.json")
|
|||
|
|
jsonData, _ := simplejson.NewFromReader(bytes.NewReader(jsonFile))
|
|||
|
|
urlPath := v.RelativePath
|
|||
|
|
idPatten := "(.*)/:(\\w+)" // 正则替换，把:id换成{id}
|
|||
|
|
reg, _ := regexp.Compile(idPatten)
|
|||
|
|
if reg.MatchString(urlPath) {
|
|||
|
|
urlPath = reg.ReplaceAllString(v.RelativePath, "${1}/{${2}}") // 把:id换成{id}
|
|||
|
|
}
|
|||
|
|
apiTitle, _ := jsonData.Get("paths").Get(urlPath).Get(strings.ToLower(v.HttpMethod)).Get("summary").String()
|
|||
|
|
|
|||
|
|
err := d.Debug().Where(SysApi{Path: v.RelativePath, Action: v.HttpMethod}).
|
|||
|
|
Attrs(SysApi{Handle: v.Handler, Title: apiTitle}).
|
|||
|
|
FirstOrCreate(&SysApi{}).
|
|||
|
|
//Update("handle", v.Handler).
|
|||
|
|
Error
|
|||
|
|
if err != nil {
|
|||
|
|
err := fmt.Errorf("Models SaveSysApi error: %s \r\n ", err.Error())
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|