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

给自定义SQL增加了根据条件判断是否生成的功能

Browse Source
This commit is contained in:
jiang4869
2022-01-06 23:38:48 +08:00
parent f79e1becc7
commit deb37930bd
3 changed files with 73 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
data/view/genfunc/def.go
View File

@@ -117,31 +117,45 @@ func CloseRelated() {
// 自定义sql查询
type Condition struct {
list []*conditionInfo
type Query struct {
list []*queryInfo
}
// And a condition by and .and 一个条件
func (c *Condition) And(column string, cases string, value interface{}) {
c.list = append(c.list, &conditionInfo{
andor: "and",
column: column, // 列名
case_: cases, // 条件(and,or,in,>=,<=)
value: value,
})
func (c *Query) AndOnCondition(condition bool,column string, cases string, value interface{}) {
if condition {
c.list = append(c.list, &queryInfo{
andor: "and",
column: column, // 列名
case_: cases, // 条件(and,or,in,>=,<=)
value: value,
})
}
}
// Or a condition by or .or 一个条件
func (c *Condition) Or(column string, cases string, value interface{}) {
c.list = append(c.list, &conditionInfo{
andor: "or",
column: column, // 列名
case_: cases, // 条件(and,or,in,>=,<=)
value: value,
})
// And a Condition by and .and 一个条件
func (c *Query) And(column string, cases string, value interface{}) {
c.AndOnCondition(true,column,cases,value)
}
func (c *Condition) Get() (where string, out []interface{}) {
func (c *Query) OrOnCondition(condition bool,column string, cases string, value interface{}) {
if condition {
c.list = append(c.list, &queryInfo{
andor: "or",
column: column, // 列名
case_: cases, // 条件(and,or,in,>=,<=)
value: value,
})
}
}
// Or a Condition by or .or 一个条件
func (c *Query) Or(column string, cases string, value interface{}) {
c.OrOnCondition(true,column,cases,value)
}
func (c *Query) Get() (where string, out []interface{}) {
firstAnd := -1
for i := 0; i < len(c.list); i++ { // 查找第一个and
if c.list[i].andor == "and" {
@@ -168,7 +182,7 @@ func (c *Condition) Get() (where string, out []interface{}) {
return
}
type conditionInfo struct {
type queryInfo struct {
andor string
column string // 列名
case_ string // 条件(in,>=,<=)

13
data/view/genfunc/genfunc_test.go
View File

@@ -140,12 +140,13 @@ func TestFuncFetchBy(t *testing.T) {
// TestCondition 测试sql构建
func TestCondition(t *testing.T) {
condition := model.Condition{}
condition.And(model.AccountColumns.AccountID, ">=", "1")
condition.And(model.AccountColumns.UserID, "in", []string{"1", "2", "3"})
condition.Or(model.AccountColumns.Type, "in", []string{"1", "2", "3"})
query := model.Query{}
query.And(model.AccountColumns.AccountID, ">=", "1")
query.And(model.AccountColumns.UserID, "in", []string{"1", "2", "3"})
query.AndOnCondition(false, model.AccountColumns.AccountID, "in", []string{"5"})
query.Or(model.AccountColumns.Type, "in", []string{"1", "2", "3"})
where, obj := condition.Get()
where, obj := query.Get()
fmt.Println(where)
fmt.Println(obj...)
@@ -155,6 +156,6 @@ func TestCondition(t *testing.T) {
sqldb.Close()
}()
accountMgr := model.AccountMgr(db.Where(condition.Get()))
accountMgr := model.AccountMgr(db.Where(where, obj))
accountMgr.Gets()
}

52
data/view/genfunc/model/gen.base.go
View File

@@ -93,31 +93,43 @@ func CloseRelated() {
}
// 自定义sql查询
type Condition struct {
list []*conditionInfo
type Query struct {
list []*queryInfo
}
// And a condition by and .and 一个条件
func (c *Condition) And(column string, cases string, value interface{}) {
c.list = append(c.list, &conditionInfo{
andor: "and",
column: column, // 列名
case_: cases, // 条件(and,or,in,>=,<=)
value: value,
})
func (c *Query) AndOnCondition(condition bool, column string, cases string, value interface{}) {
if condition {
c.list = append(c.list, &queryInfo{
andor: "and",
column: column, // 列名
case_: cases, // 条件(and,or,in,>=,<=)
value: value,
})
}
}
// Or a condition by or .or 一个条件
func (c *Condition) Or(column string, cases string, value interface{}) {
c.list = append(c.list, &conditionInfo{
andor: "or",
column: column, // 列名
case_: cases, // 条件(and,or,in,>=,<=)
value: value,
})
// And a Condition by and .and 一个条件
func (c *Query) And(column string, cases string, value interface{}) {
c.AndOnCondition(true, column, cases, value)
}
func (c *Condition) Get() (where string, out []interface{}) {
func (c *Query) OrOnCondition(condition bool, column string, cases string, value interface{}) {
if condition {
c.list = append(c.list, &queryInfo{
andor: "or",
column: column, // 列名
case_: cases, // 条件(and,or,in,>=,<=)
value: value,
})
}
}
// Or a Condition by or .or 一个条件
func (c *Query) Or(column string, cases string, value interface{}) {
c.OrOnCondition(true, column, cases, value)
}
func (c *Query) Get() (where string, out []interface{}) {
firstAnd := -1
for i := 0; i < len(c.list); i++ { // 查找第一个and
if c.list[i].andor == "and" {
@@ -144,7 +156,7 @@ func (c *Condition) Get() (where string, out []interface{}) {
return
}
type conditionInfo struct {
type queryInfo struct {
andor string
column string // 列名
case_ string // 条件(in,>=,<=)