jager/gormt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

generate func at early days

Browse Source
This commit is contained in:
谢小军
2020-01-04 22:06:43 +08:00
parent f1dd6a7632
commit 9a0ac52273
6 changed files with 128 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
data/view/genfunc/def.go Normal file
View File

@@ -0,0 +1,49 @@
package genfunc
const (
genBase = `
package {{.PackageName}}
import (
"context"
"github.com/jinzhu/gorm"
)
// prepare for outher
type _BaseMgr struct {
*gorm.DB
ctx *context.Context
isRelated bool
}
// SetCtx set context
func (obj *_BaseMgr) SetCtx(c *context.Context) {
obj.ctx = c
}
// GetDB get gorm.DB info
func (obj *_BaseMgr) GetDB() *gorm.DB {
return obj.DB
}
// IsRelated Query foreign key Association.是否查询外键关联(gorm.Related)
func (obj *_BaseMgr) IsRelated(b bool) {
obj.isRelated = b
}
type options struct {
query map[string]interface{}
}
// Option overrides behavior of Connect.
type Option interface {
apply(*options)
}
type optionFunc func(*options)
func (f optionFunc) apply(o *options) {
f(o)
}
`
)

25
data/view/genfunc/gen.logic.go
View File

@@ -33,10 +33,15 @@ func ExampleMgr(db *gorm.DB) *_ExampleMgr {
return &_ExampleMgr{_BaseMgr: &_BaseMgr{DB: db}}
}
// GetTableName get sql table name.获取数据库名字
func (obj *_ExampleMgr) GetTableName() string {
return "example"
}
// GetFromID 通过id获取内容
func (obj *_ExampleMgr) GetFromID(id int) (results []*Example, err error) {
err = obj.DB.Table("example").Where("id = ?", id).Find(&results).Error
if err == nil {
err = obj.DB.Table(obj.GetTableName()).Where("id = ?", id).Find(&results).Error
if err == nil && obj.isRelated {
for i := 0; i < len(results); i++ {
var userList []User
err = obj.DB.Where("job = ?", results[i].UserID).Find(&userList).Error
@@ -51,8 +56,8 @@ func (obj *_ExampleMgr) GetFromID(id int) (results []*Example, err error) {
// GetByPrimaryKey 唯一主键查找
func (obj *_ExampleMgr) GetByPrimaryKey(id int64) (result Example, err error) {
err = obj.DB.Table("example").Where("id = ?", id).Find(&result).Error
if err == nil {
err = obj.DB.Table(obj.GetTableName()).Where("id = ?", id).Find(&result).Error
if err == nil && obj.isRelated {
var info []User
err = obj.DB.Where("job = ?", result.UserID).Find(&info).Error
if err != nil {
@@ -65,8 +70,8 @@ func (obj *_ExampleMgr) GetByPrimaryKey(id int64) (result Example, err error) {
// GetByPrimaryKey 批量唯一主键查找
func (obj *_ExampleMgr) GetByPrimaryKeys(ids []int64) (results []*Example, err error) {
err = obj.DB.Table("example").Where("id IN (?)", ids).Find(&results).Error
if err == nil {
err = obj.DB.Table(obj.GetTableName()).Where("id IN (?)", ids).Find(&results).Error
if err == nil && obj.isRelated {
for i := 0; i < len(results); i++ {
var userList []User
err = obj.DB.Where("job = ?", results[i].UserID).Find(&userList).Error
@@ -90,8 +95,8 @@ func (obj *_ExampleMgr) GetByOption(opts ...Option) (result Example, err error)
o.apply(&options)
}
err = obj.DB.Table("example").Where(options.query).Find(&result).Error
if err == nil {
err = obj.DB.Table(obj.GetTableName()).Where(options.query).Find(&result).Error
if err == nil && obj.isRelated {
var info []User
err = obj.DB.Where("job = ?", result.UserID).Find(&info).Error
if err != nil {
@@ -111,8 +116,8 @@ func (obj *_ExampleMgr) GetByOptions(opts ...Option) (results []*Example, err er
o.apply(&options)
}
err = obj.DB.Table("example").Where(options.query).Find(&results).Error
if err == nil {
err = obj.DB.Table(obj.GetTableName()).Where(options.query).Find(&results).Error
if err == nil && obj.isRelated {
for i := 0; i < len(results); i++ {
var userList []User
err = obj.DB.Where("job = ?", results[i].UserID).Find(&userList).Error

10
data/view/genfunc/genfunc.go
View File

@@ -1 +1,11 @@
package genfunc
// GetGenBaseTemp get gen base template str
func GetGenBaseTemp() string {
return genBase
}
// GetGenLogicTemp get gen logic template str
func GetGenLogicTemp() string {
return ""
}

18
data/view/genstruct/common.go
View File

@@ -76,17 +76,19 @@ func (s *GenStruct) SetStructName(name string) {
// SetNotes set the notes.设置注释
func (s *GenStruct) SetNotes(notes string) {
if len(notes) > 0 {
notes = s.Name + " " + notes
if len(notes) == 0 {
notes = "[...]" // default of struct notes(for export ).struct 默认注释(为了导出注释)
}
notes = s.Name + " " + notes
a := strings.Split(notes, "\n")
var text []string
for _, v := range a {
if len(v) > 0 {
text = append(text, "// "+v)
}
// if len(v) > 0 {
text = append(text, "// "+v)
// }
}
s.Notes = strings.Join(text, "\r\n")
}
@@ -159,6 +161,12 @@ func (p *GenPackage) Generate() string {
}
// -----------end
// add func
for _, v := range p.FuncStrList {
pa.Add(v)
}
// -----------end
// output.输出
strOut := ""
for _, v := range pa.Generates() {

7
data/view/genstruct/def.go
View File

@@ -18,7 +18,8 @@ type GenStruct struct {
// GenPackage package of IPackage.包体
type GenPackage struct {
Name string // name.名字
Imports map[string]string // Inclusion term.元素组合
Structs []GenStruct // struct list .结构体组合
Name string // name.名字
Imports map[string]string // Inclusion term.元素组合
Structs []GenStruct // struct list .结构体组合
FuncStrList []string // func of template on string. 函数的最终定义
}

38
data/view/model/model.go
View File

@@ -1,11 +1,15 @@
package model
import (
"bytes"
"fmt"
"strings"
"text/template"
"github.com/xxjwxc/public/mybigcamel"
"github.com/xxjwxc/gormt/data/config"
"github.com/xxjwxc/gormt/data/view/genfunc"
"github.com/xxjwxc/gormt/data/view/genstruct"
)
@@ -27,6 +31,7 @@ func Generate(info DBInfo) (out []GenOutInfo) {
// ------end
// gen function
out = append(out, m.generateFunc()...)
// -------------- end
return
}
@@ -180,6 +185,37 @@ func (m *_Model) getColumusKeyMulti(tableName, col string) (isMulti bool, isFind
}
// ///////////////////////// func
func (m *_Model) generateFunc() {
func (m *_Model) generateFunc() (genOut []GenOutInfo) {
// getn base
tmpl, err := template.New("gen_base").Parse(genfunc.GetGenBaseTemp())
if err != nil {
panic(err)
}
var buf bytes.Buffer
tmpl.Execute(&buf, m.info)
genOut = append(genOut, GenOutInfo{
FileName: "gen.base.go",
FileCtx: buf.String(),
})
//tools.WriteFile(outDir+"gen_router.go", []string{buf.String()}, true)
// -------end------
for _, tab := range m.info.TabList {
var pkg genstruct.GenPackage
pkg.SetPackage(m.info.PackageName) //package name
// tmpl, err := template.New("gen_logic").Funcs(template.FuncMap{"GetStringList": GetStringList}).Parse(genfunc.GetGenBaseTemp())
// if err != nil {
// panic(err)
// }
// var buf bytes.Buffer
// tmpl.Execute(&buf, m.info)
genOut = append(genOut, GenOutInfo{
FileName: fmt.Sprintf("gen.%v.go", tab.Name),
FileCtx: pkg.Generate(),
})
}
return
}