mirror of
https://github.com/beego/bee.git
synced 2024-11-24 13:30:53 +00:00
commit
1daa1d1bae
50
g_models.go
50
g_models.go
@ -440,10 +440,9 @@ func writeModelFiles(tables []*Table, mPath string) {
|
||||
fpath := path.Join(mPath, filename+".go")
|
||||
f, err := os.OpenFile(fpath, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666)
|
||||
if err != nil {
|
||||
ColorLog("[ERRO] %v\n", err)
|
||||
os.Exit(2)
|
||||
ColorLog("[WARN] %v\n", err)
|
||||
continue
|
||||
}
|
||||
defer f.Close()
|
||||
template := ""
|
||||
if tb.Pk == "" {
|
||||
template = STRUCT_MODEL_TPL
|
||||
@ -456,6 +455,7 @@ func writeModelFiles(tables []*Table, mPath string) {
|
||||
ColorLog("[ERRO] Could not write model file to %s\n", fpath)
|
||||
os.Exit(2)
|
||||
}
|
||||
f.Close()
|
||||
ColorLog("[INFO] model => %s\n", fpath)
|
||||
formatAndFixImports(fpath)
|
||||
}
|
||||
@ -471,15 +471,15 @@ func writeControllerFiles(tables []*Table, cPath string) {
|
||||
fpath := path.Join(cPath, filename+".go")
|
||||
f, err := os.OpenFile(fpath, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666)
|
||||
if err != nil {
|
||||
ColorLog("[ERRO] %v\n", err)
|
||||
os.Exit(2)
|
||||
ColorLog("[WARN] %v\n", err)
|
||||
continue
|
||||
}
|
||||
defer f.Close()
|
||||
fileStr := strings.Replace(CTRL_TPL, "{{ctrlName}}", camelCase(tb.Name), -1)
|
||||
if _, err := f.WriteString(fileStr); err != nil {
|
||||
ColorLog("[ERRO] Could not write controller file to %s\n", fpath)
|
||||
os.Exit(2)
|
||||
}
|
||||
f.Close()
|
||||
ColorLog("[INFO] controller => %s\n", fpath)
|
||||
formatAndFixImports(fpath)
|
||||
}
|
||||
@ -504,14 +504,14 @@ func writeRouterFile(tables []*Table, rPath string) {
|
||||
routerStr = strings.Replace(routerStr, "{{projectName}}", projectName, 1)
|
||||
f, err := os.OpenFile(fpath, os.O_CREATE|os.O_EXCL|os.O_RDWR, 0666)
|
||||
if err != nil {
|
||||
ColorLog("[ERRO] %v\n", err)
|
||||
os.Exit(2)
|
||||
ColorLog("[WARN] %v\n", err)
|
||||
return
|
||||
}
|
||||
defer f.Close()
|
||||
if _, err := f.WriteString(routerStr); err != nil {
|
||||
ColorLog("[ERRO] Could not write router file to %s\n", fpath)
|
||||
os.Exit(2)
|
||||
}
|
||||
f.Close()
|
||||
ColorLog("[INFO] router => %s\n", fpath)
|
||||
formatAndFixImports(fpath)
|
||||
}
|
||||
@ -624,6 +624,18 @@ func Get{{modelName}}ById(id int) (v *{{modelName}}, err error) {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// GetAll{{modelName}} retrieves all {{modelName}} matches certain condition. Returns empty list if
|
||||
// no records exist
|
||||
func GetAll{{modelName}}() (l []{{modelName}}, err error) {
|
||||
o := orm.NewOrm()
|
||||
t := new({{modelName}})
|
||||
qs := o.QueryTable(t)
|
||||
if _, err := qs.All(&l); err == nil {
|
||||
return l, nil
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// Update{{modelName}} updates {{modelName}} by Id and returns error if
|
||||
// the record to be updated doesn't exist
|
||||
func Update{{modelName}}ById(m *{{modelName}}) (err error) {
|
||||
@ -696,7 +708,23 @@ func (this *{{ctrlName}}Controller) GetOne() {
|
||||
this.ServeJson()
|
||||
}
|
||||
|
||||
// @Title update
|
||||
// @Title Get All
|
||||
// @Description get {{ctrlName}}
|
||||
// @Param id path string true "get all records matches certain condition"
|
||||
// @Success 200 {object} models.{{ctrlName}}
|
||||
// @Failure 403
|
||||
// @router / [get]
|
||||
func (this *{{ctrlName}}Controller) GetAll() {
|
||||
l, err := models.GetAll{{ctrlName}}()
|
||||
if err != nil {
|
||||
this.Data["json"] = err.Error()
|
||||
} else {
|
||||
this.Data["json"] = l
|
||||
}
|
||||
this.ServeJson()
|
||||
}
|
||||
|
||||
// @Title Update
|
||||
// @Description update the {{ctrlName}}
|
||||
// @Param id path string true "The id you want to update"
|
||||
// @Param body body models.{{ctrlName}} true "body for {{ctrlName}} content"
|
||||
@ -716,7 +744,7 @@ func (this *{{ctrlName}}Controller) Put() {
|
||||
this.ServeJson()
|
||||
}
|
||||
|
||||
// @Title delete
|
||||
// @Title Delete
|
||||
// @Description delete the {{ctrlName}}
|
||||
// @Param id path string true "The id you want to delete"
|
||||
// @Success 200 {string} delete success!
|
||||
|
Loading…
Reference in New Issue
Block a user