add func base template

This commit is contained in:
谢小军
2019-12-16 19:24:40 +08:00
parent d0a57e4c0a
commit 95ec33c371
2 changed files with 112 additions and 12 deletions

View File

@@ -2,11 +2,11 @@ package genfunc
import (
"context"
"fmt"
"github.com/jinzhu/gorm"
)
// prepare for outher
type _BaseMgr struct {
*gorm.DB
ctx *context.Context
@@ -17,18 +17,22 @@ func (obj *_BaseMgr) SetCtx(c *context.Context) {
obj.ctx = c
}
////////////////////////////////////////////logic
type _ExampleMgr struct {
*_BaseMgr
// GetDB get gorm.DB info
func (obj *_BaseMgr) GetDB() *gorm.DB {
return obj.DB
}
// ExampleMgr open func
func ExampleMgr(db *gorm.DB) *_ExampleMgr {
if db == nil {
panic(fmt.Errorf("ExampleMgr init need db"))
}
return &_ExampleMgr{_BaseMgr: &_BaseMgr{DB: db}}
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)
}

View File

@@ -0,0 +1,96 @@
package genfunc
import (
"fmt"
"github.com/jinzhu/gorm"
)
// Example demo of root
type Example struct {
ID int `json:"-"`
UserID int `json:"userId"`
UserList []User `gorm:"association_foreignkey:userId;foreignkey:job" json:"user_list"`
}
// User demo of user
type User struct {
UserID int `json:"userId"`
Job int `json:"job"`
}
////////////////////////////////////////////-----logic------
type _ExampleMgr struct {
*_BaseMgr
}
// ExampleMgr open func
func ExampleMgr(db *gorm.DB) *_ExampleMgr {
if db == nil {
panic(fmt.Errorf("ExampleMgr init need db"))
}
return &_ExampleMgr{_BaseMgr: &_BaseMgr{DB: db}}
}
// 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 {
}
return
}
// GetByPrimaryKey 唯一主键查找
func (obj *_ExampleMgr) GetByPrimaryKey(id int64) (Example, error) {
var tmp Example
err := obj.DB.Table("example").Where("id = ?", id).Find(&tmp).Error
return tmp, err
}
// GetByPrimaryKey 批量唯一主键查找
func (obj *_ExampleMgr) GetByPrimaryKeys(ids []int64) ([]*Example, error) {
var tmp []*Example
err := obj.DB.Table("example").Where("id", ids).Find(&tmp).Error
return tmp, err
}
//////////////////////////option case ////////////////////////////////////////////
// GetByPrimaryKey 功能选项模式获取
func (obj *_ExampleMgr) GetByOption(opts ...Option) (Example, error) {
options := options{
query: make(map[string]interface{}, len(opts)),
}
for _, o := range opts {
o.apply(&options)
}
var tmp Example
err := obj.DB.Table("example").Where(options.query).Find(&tmp).Error
return tmp, err
}
// GetByPrimaryKey 批量功能选项模式获取
func (obj *_ExampleMgr) GetByOptions(opts ...Option) ([]*Example, error) {
options := options{
query: make(map[string]interface{}, len(opts)),
}
for _, o := range opts {
o.apply(&options)
}
var tmp []*Example
err := obj.DB.Table("example").Where(options.query).Find(&tmp).Error
return tmp, err
}
// WithID id获取
func (obj *_ExampleMgr) WithID(id string) Option {
return optionFunc(func(o *options) {
o.query["id"] = id
})
}
///////////////////////////////////////////////////