diff --git a/data/view/genfunc/def.go b/data/view/genfunc/def.go index a6db57d..aeb9e8e 100755 --- a/data/view/genfunc/def.go +++ b/data/view/genfunc/def.go @@ -117,31 +117,45 @@ func CloseRelated() { // 自定义sql查询 -type Condition struct { - list []*conditionInfo +type Query struct { + list []*queryInfo +} + +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, + }) + } +} + + +// And a Condition by and .and 一个条件 +func (c *Query) And(column string, cases string, value interface{}) { + c.AndOnCondition(true,column,cases,value) } -// 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) 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 *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, - }) +// Or a Condition by or .or 一个条件 +func (c *Query) Or(column string, cases string, value interface{}) { + c.OrOnCondition(true,column,cases,value) } -func (c *Condition) Get() (where string, out []interface{}) { +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,>=,<=) diff --git a/data/view/genfunc/genfunc_test.go b/data/view/genfunc/genfunc_test.go index b10dc71..ba25a09 100644 --- a/data/view/genfunc/genfunc_test.go +++ b/data/view/genfunc/genfunc_test.go @@ -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() } diff --git a/data/view/genfunc/model/gen.base.go b/data/view/genfunc/model/gen.base.go index ce3dde4..617387f 100644 --- a/data/view/genfunc/model/gen.base.go +++ b/data/view/genfunc/model/gen.base.go @@ -93,31 +93,43 @@ func CloseRelated() { } // 自定义sql查询 -type Condition struct { - list []*conditionInfo +type Query struct { + list []*queryInfo +} + +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, + }) + } +} + +// And a Condition by and .and 一个条件 +func (c *Query) And(column string, cases string, value interface{}) { + c.AndOnCondition(true, column, cases, value) } -// 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) 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 *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, - }) +// Or a Condition by or .or 一个条件 +func (c *Query) Or(column string, cases string, value interface{}) { + c.OrOnCondition(true, column, cases, value) } -func (c *Condition) Get() (where string, out []interface{}) { +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,>=,<=)