74 lines
1.6 KiB
Go
74 lines
1.6 KiB
Go
|
package dao
|
|||
|
|
|
|||
|
|
import "fmt"
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
//logicNoOp = 0
|
|||
|
|
logicOpAnd = 1
|
|||
|
|
logicOpOr = 2
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
//Condition 生成的condition为
|
|||
|
|
// [before] [key] [operator] ? [after]
|
|||
|
|
// 如
|
|||
|
|
// [(] [id] [=] ? [and]
|
|||
|
|
// [ctime] [>] ? [)]
|
|||
|
|
// [order by] [time] nil [] //如果value是nil，不会设置placeholder ?
|
|||
|
|
type Condition struct {
|
|||
|
|
logicOp int
|
|||
|
|
Before string
|
|||
|
|
Key string
|
|||
|
|
Operator string
|
|||
|
|
Value interface{}
|
|||
|
|
After string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//ConcatCondition concat conditions
|
|||
|
|
func ConcatCondition(conditions ...Condition) (conditionStr string, args []interface{}, hasOperator bool) {
|
|||
|
|
hasOperator = false
|
|||
|
|
for _, c := range conditions {
|
|||
|
|
var questionMark = "?"
|
|||
|
|
if c.Value == nil {
|
|||
|
|
questionMark = ""
|
|||
|
|
}
|
|||
|
|
if c.Operator != "" {
|
|||
|
|
hasOperator = true
|
|||
|
|
}
|
|||
|
|
var logicOp = ""
|
|||
|
|
switch c.logicOp {
|
|||
|
|
case logicOpAnd:
|
|||
|
|
logicOp = " and "
|
|||
|
|
case logicOpOr:
|
|||
|
|
logicOp = " or "
|
|||
|
|
}
|
|||
|
|
conditionStr += fmt.Sprintf(" %s %s %s %s %s %s", logicOp, c.Before, c.Key, c.Operator, questionMark, c.After)
|
|||
|
|
if c.Value != nil {
|
|||
|
|
args = append(args, c.Value)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//AndCondition and condition
|
|||
|
|
func AndCondition(conditions ...Condition) (result []Condition) {
|
|||
|
|
return addLogicOperator(logicOpAnd, conditions...)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//OrCondition or condition
|
|||
|
|
func OrCondition(conditions ...Condition) (result []Condition) {
|
|||
|
|
return addLogicOperator(logicOpOr, conditions...)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func addLogicOperator(operator int, conditions ...Condition) (result []Condition) {
|
|||
|
|
var isFirst = true
|
|||
|
|
for _, v := range conditions {
|
|||
|
|
if isFirst {
|
|||
|
|
isFirst = false
|
|||
|
|
} else {
|
|||
|
|
v.logicOp = operator
|
|||
|
|
}
|
|||
|
|
result = append(result, v)
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|