openbilibili/app/admin/main/up/dao/utils.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
}