代码拉取完成,页面将自动刷新
package torm
import (
"reflect"
"strings"
"errors"
"github.com/thinkoner/torm/grammar"
"github.com/thinkoner/torm/query"
)
type Builder struct {
Connection *Connection
grammar grammar.Grammar
Query *query.Query
Bindings map[string][]interface{}
}
type Binding []interface{}
func NewBuilder(connection *Connection, grammar grammar.Grammar) *Builder {
return &Builder{
Connection: connection,
grammar: grammar,
Query: &query.Query{},
Bindings: getDefaultBindings(),
}
}
// Select the columns to be selected.
func (b *Builder) Select(columns ...string) *Builder {
if len(columns) == 0 {
columns = []string{"*"}
}
b.Query.Columns = columns
return b
}
// SelectRaw Add a new "raw" select expression to the query.
func (b *Builder) SelectRaw(expression string, bindings ...interface{}) *Builder {
b.AddSelect(expression)
if len(bindings) != 0 {
b.AddBinding(bindings, "select")
}
return b
}
func (b *Builder) AddSelect(columns ...string) *Builder {
for _, column := range columns {
b.Query.Columns = append(b.Query.Columns, column)
}
return b
}
// From Set the table which the query is targeting.
func (b *Builder) From(table string) *Builder {
b.Query.From = table
return b
}
// Join Add a join clause to the query.
func (b *Builder) Join(table string, first string, args ...interface{}) *Builder {
var (
operator string
second string
typtStr string
where bool
)
typtStr = "INNER"
count := len(args)
if count >= 1 {
operator = args[0].(string)
}
if count >= 2 {
second = args[1].(string)
}
if count >= 3 {
typtStr, _ = args[2].(string)
}
if count >= 4 {
where, _ = args[2].(bool)
}
q := &query.Query{}
if where {
where := &query.Where{
Type: "Basic",
Column: first,
Operator: operator,
Value: second,
}
q.Wheres = append(q.Wheres, where)
b.AddBinding(second, "join")
} else {
where := &query.Where{
Type: "Column",
First: first,
Operator: operator,
Second: second,
Boolean: "and",
}
q.JoinClause = true
q.Wheres = append(q.Wheres, where)
}
b.Query.Joins = append(b.Query.Joins, &query.Join{
Type: typtStr,
Table: table,
Query: q,
})
return b
}
// func (b *Builder) Value(column string) error {
// result, err := b.First([]string{column})
// if len(result) > 0 {
// return result[column], err
// }
// return nil, err
// }
func (b *Builder) First(dest interface{}, columns ...interface{}) error {
return b.Take(1).Get(dest, columns...)
}
func (b *Builder) Get(dest interface{}, columns ...interface{}) error {
var cols []string
if len(columns) == 0 {
cols = []string{"*"}
} else {
if _, ok := columns[0].([]string); ok {
cols = columns[0].([]string)
} else {
return errors.New("Invalid parameters")
}
}
if len(cols) == 0 {
cols = []string{"*"}
}
original := b.Query.Columns
if original == nil {
b.Query.Columns = cols
}
err := b.runSelect(dest)
b.Query.Columns = original
return err
}
func (b *Builder) Scan(dest ...interface{}) error {
return b.runScan(dest...)
}
// Where Add a basic where clause to the query.
func (b *Builder) Where(column string, args ...interface{}) *Builder {
var (
operator string
value interface{}
boolean string
)
boolean = "and"
count := len(args)
if count == 1 {
operator = "="
value = args[0]
}
if count >= 2 {
operator, _ = args[0].(string)
value = args[1]
}
if count >= 3 {
boolean, _ = args[2].(string)
}
where := &query.Where{
Type: "Basic",
Column: column,
Operator: operator,
Value: value,
Boolean: boolean,
}
b.Query.Wheres = append(b.Query.Wheres, where)
b.AddBinding(value, "where")
return b
}
// OrWhere Add an "or where" clause to the query.
func (b *Builder) OrWhere(column string, args ...interface{}) *Builder {
var (
operator string
value interface{}
)
count := len(args)
if count == 1 {
operator = "="
value = args[0]
}
if count >= 2 {
operator, _ = args[0].(string)
value = args[1]
}
return b.Where(column, operator, value, "OR")
}
// WhereColumn Add a "where" clause comparing two columns to the query.
func (b *Builder) WhereColumn(first string, args ...interface{}) *Builder {
var (
operator string
value string
boolean string
)
boolean = "AND"
count := len(args)
if count == 1 {
operator = "="
value = args[0].(string)
}
if count >= 2 {
operator, _ = args[0].(string)
value = args[1].(string)
}
if count >= 3 {
boolean, _ = args[2].(string)
}
where := &query.Where{
Type: "Column",
First: first,
Operator: operator,
Second: value,
Boolean: boolean,
}
b.Query.Wheres = append(b.Query.Wheres, where)
return b
}
// OrWhereColumn Add an "or where" clause comparing two columns to the query.
func (b *Builder) OrWhereColumn(first string, args ...interface{}) *Builder {
var (
operator string
value string
)
count := len(args)
if count == 1 {
operator = "="
value = args[0].(string)
}
if count >= 2 {
operator, _ = args[0].(string)
value = args[1].(string)
}
return b.WhereColumn(first, operator, value, "OR")
}
// WhereRaw Add a raw where clause to the query.
func (b *Builder) WhereRaw(sql string, bindings ...interface{}) *Builder {
b.Query.Wheres = append(
b.Query.Wheres,
&query.Where{
Type: "Raw",
Sql: sql,
Boolean: "AND",
},
)
if len(bindings) != 0 {
b.AddBinding(bindings, "where")
}
return b
}
// OrWhereRaw Add a raw or where clause to the query.
func (b *Builder) OrWhereRaw(sql string, bindings ...interface{}) *Builder {
b.Query.Wheres = append(
b.Query.Wheres,
&query.Where{
Type: "Raw",
Sql: sql,
Boolean: "OR",
},
)
if len(bindings) != 0 {
b.AddBinding(bindings, "where")
}
return b
}
// WhereIn Add a "where in" clause to the query.
func (b *Builder) WhereIn(column string, values []interface{}, args ...interface{}) *Builder {
var (
boolean string
not bool
)
count := len(args)
if count == 0 {
boolean = "and"
not = false
}
if count == 1 {
boolean = args[0].(string)
not = false
}
if count >= 2 {
boolean = args[0].(string)
not = args[1].(bool)
}
t := "In"
if not {
t = "NotIn"
}
b.Query.Wheres = append(
b.Query.Wheres,
&query.Where{
Type: t,
Column: column,
Values: values,
Boolean: boolean,
},
)
for _, value := range values {
b.AddBinding(value, "where")
}
return b
}
// OrWhereIn Add an "or where in" clause to the query.
func (b *Builder) OrWhereIn(column string, values []interface{}) *Builder {
return b.WhereIn(column, values, "OR")
}
// WhereNotIn Add a "where not in" clause to the query.
func (b *Builder) WhereNotIn(column string, values []interface{}, args ...interface{}) *Builder {
var boolean string
count := len(args)
if count == 0 {
boolean = "AND"
}
if count == 1 {
boolean = args[0].(string)
}
return b.WhereIn(column, values, boolean, true)
}
// OrWhereNotIn Add an "or where not in" clause to the query.
func (b *Builder) OrWhereNotIn(column string, values []interface{}) *Builder {
return b.WhereNotIn(column, values, "OR")
}
// WhereNull Add a "where null" clause to the query.
func (b *Builder) WhereNull(column string, args ...interface{}) *Builder {
var (
boolean string
not bool
)
count := len(args)
if count == 0 {
boolean = "and"
not = false
}
if count == 1 {
boolean = args[0].(string)
not = false
}
if count >= 2 {
boolean = args[0].(string)
not = args[1].(bool)
}
t := "Null"
if not {
t = "NotNull"
}
b.Query.Wheres = append(
b.Query.Wheres,
&query.Where{
Type: t,
Column: column,
Boolean: boolean,
},
)
return b
}
// OrWhereNull Add an "or where null" clause to the query.
func (b *Builder) OrWhereNull(column string) *Builder {
return b.WhereNull(column, "OR")
}
// WhereNotNull Add a "where not null" clause to the query.
func (b *Builder) WhereNotNull(column string, args ...interface{}) *Builder {
var (
boolean string
)
count := len(args)
if count == 0 {
boolean = "and"
}
if count == 1 {
boolean = args[0].(string)
}
return b.WhereNull(column, boolean, true)
}
// OrWhereNotNull Add an "or where not null" clause to the query.
func (b *Builder) OrWhereNotNull(column string) *Builder {
return b.WhereNotNull(column, "OR")
}
// WhereBetween Add a where between statement to the query.
func (b *Builder) WhereBetween(column string, values []interface{}, args ...interface{}) *Builder {
var (
boolean string
not bool
)
count := len(args)
if count == 0 {
boolean = "and"
not = false
}
if count == 1 {
boolean = args[0].(string)
not = false
}
if count >= 2 {
boolean = args[0].(string)
not = args[1].(bool)
}
b.Query.Wheres = append(
b.Query.Wheres,
&query.Where{
Type: "Between",
Column: column,
Boolean: boolean,
Not: not,
},
)
b.AddBinding(values, "where")
return b
}
// OrWhereBetween Add an or where between statement to the query.
func (b *Builder) OrWhereBetween(column string, values []interface{}) *Builder {
return b.WhereBetween(column, values, "OR")
}
// WhereNotBetween Add a where not between statement to the query.
func (b *Builder) WhereNotBetween(column string, values []interface{}, args ...interface{}) *Builder {
var (
boolean string
)
count := len(args)
if count == 0 {
boolean = "and"
}
if count == 1 {
boolean = args[0].(string)
}
return b.WhereBetween(column, values, boolean, true)
}
// OrWhereNotBetween Add an or where not between statement to the query.
func (b *Builder) OrWhereNotBetween(column string, values []interface{}) *Builder {
return b.WhereNotBetween(column, values, "OR")
}
// GroupBy Add a "group by" clause to the query.
func (b *Builder) GroupBy(groups ...string) *Builder {
if len(groups) != 0 {
for _, group := range groups {
b.Query.Groups = append(b.Query.Groups, group)
}
}
return b
}
// Having Add a "having" clause to the query.
func (b *Builder) Having(column string, args ...interface{}) *Builder {
var (
operator string
value interface{}
boolean string
)
boolean = "and"
count := len(args)
if count == 1 {
operator = "="
value = args[0]
}
if count >= 2 {
operator, _ = args[0].(string)
value = args[1]
}
if count >= 3 {
boolean, _ = args[2].(string)
}
having := &query.Having{
Type: "Basic",
Column: column,
Operator: operator,
Value: value,
Boolean: boolean,
}
b.Query.Havings = append(b.Query.Havings, having)
b.AddBinding(value, "having")
return b
}
// OrderBy Add an "order by" clause to the query.
func (b *Builder) OrderBy(column string, args ...string) *Builder {
direction := "asc"
if len(args) > 0 {
direction = args[0]
}
if len(b.Query.Unions) == 0 {
b.Query.Orders = append(
b.Query.Orders,
&query.Order{
Column: column,
Direction: strings.ToUpper(direction),
},
)
} else {
b.Query.UnionOrders = append(
b.Query.UnionOrders,
&query.UnionOrder{
Column: column,
Direction: strings.ToUpper(direction),
},
)
}
return b
}
// OrderByDesc Add a descending "order by" clause to the query.
func (b *Builder) OrderByDesc(column string) *Builder {
return b.OrderBy(column, "desc")
}
// OrderByRaw Add a raw "order by" clause to the query.
func (b *Builder) OrderByRaw(sql string, bindings ...interface{}) *Builder {
if len(b.Query.Unions) == 0 {
b.Query.Orders = append(
b.Query.Orders,
&query.Order{
Type: "Raw",
Sql: sql,
},
)
} else {
b.Query.UnionOrders = append(
b.Query.UnionOrders,
&query.UnionOrder{
Type: "Raw",
Sql: sql,
},
)
}
if len(bindings) != 0 {
b.AddBinding(bindings, "order")
}
return b
}
// Take Alias to set the "limit" value of the query.
func (b *Builder) Take(value uint64) *Builder {
b.Limit(value)
return b
}
// Limit Set the "limit" value of the query.
func (b *Builder) Limit(value uint64) *Builder {
if len(b.Query.Unions) > 0 {
b.Query.UnionLimit = value
} else {
b.Query.Limit = value
}
return b
}
// Skip Alias to set the "offset" value of the query.
func (b *Builder) Skip(value uint64) *Builder {
return b.Offset(value)
}
// Offset Set the "offset" value of the query.
func (b *Builder) Offset(value uint64) *Builder {
if len(b.Query.Unions) > 0 {
b.Query.UnionOffset = value
} else {
b.Query.Offset = value
}
return b
}
// Count Retrieve the "count" result of the query.
func (b *Builder) Count(dest ...interface{}) error {
return b.Aggregate("COUNT", []string{"*"}, dest...)
}
// Min Retrieve the minimum value of a given column.
func (b *Builder) Min(columns string, dest ...interface{}) error {
return b.Aggregate("MIN", []string{columns}, dest...)
}
// Max Retrieve the maximum value of a given column.
func (b *Builder) Max(columns string, dest ...interface{}) error {
return b.Aggregate("MAX", []string{columns}, dest...)
}
// Sum Retrieve the sum of the values of a given column.
func (b *Builder) Sum(columns string, dest ...interface{}) error {
return b.Aggregate("SUM", []string{columns}, dest...)
}
// Avg Retrieve the average of the values of a given column.
func (b *Builder) Avg(columns string, dest ...interface{}) error {
return b.Aggregate("AVG", []string{columns}, dest...)
}
// Aggregate Execute an aggregate function on the database.
func (b *Builder) Aggregate(function string, columns []string, dest ...interface{}) error {
cols := []string{"*"}
if len(columns) > 0 {
cols = columns
}
nb := b.CloneWithout("columns").CloneWithoutBindings("select").setAggregate(function, cols)
original := nb.Query.Columns
if original == nil {
nb.Query.Columns = cols
}
return nb.Scan(dest...)
}
func (b *Builder) ToSql() string {
return b.grammar.CompileSelect(b.Query)
}
// AddBinding Add a binding to the query.
func (b *Builder) AddBinding(value interface{}, segment string) {
if _, ok := b.Bindings[segment]; !ok {
return
}
t := reflect.TypeOf(value)
switch t.Kind() {
case reflect.Slice:
s := reflect.ValueOf(value)
for i := 0; i < s.Len(); i++ {
b.Bindings[segment] = append(b.Bindings[segment], s.Index(i).Interface())
}
default:
b.Bindings[segment] = append(b.Bindings[segment], value)
}
}
func (b *Builder) GetBindings() []interface{} {
var bindings []interface{}
for _, val := range b.Bindings {
for _, v := range val {
bindings = append(bindings, v)
}
}
return bindings
}
// CloneWithout Clone the query without the given properties.
func (b *Builder) CloneWithout(properties ...string) *Builder {
builder := b
for _, property := range properties {
switch property {
case "columns":
builder.Query.Columns = nil
case "orders":
// builder.Query.Columns = nil
case "limit":
builder.Query.Limit = 0
case "offset":
// builder.Query.Columns = nil
}
}
return builder
}
// CloneWithoutBindings Clone the query without the given bindings.
func (b *Builder) CloneWithoutBindings(except ...string) *Builder {
builder := b
for _, val := range except {
builder.Bindings[val] = nil
}
return builder
}
// GetConnection Get the database connection instance.
func (b *Builder) GetConnection() *Connection {
return b.Connection
}
// GetGrammar Get the query grammar instance.
func (b *Builder) GetGrammar() grammar.Grammar {
return b.grammar
}
func (b *Builder) runSelect(dest interface{}) error {
return b.Connection.Select(
b.ToSql(),
b.GetBindings(),
dest,
)
}
func (b *Builder) runScan(dest ...interface{}) error {
return b.Connection.Scan(
b.ToSql(),
b.GetBindings(),
dest...,
)
}
func (b *Builder) setAggregate(function string, columns []string) *Builder {
b.Query.Aggregate = &query.Aggregate{
Function: function,
Columns: columns,
}
return b
}
// Insert Insert a new record into the database.
func (b *Builder) Insert(value map[string]interface{}) (int64, int64, error) {
var values []map[string]interface{}
values = append(values, value)
return b.Inserts(values)
}
// Inserts Bulk insert records into the database.
func (b *Builder) Inserts(values []map[string]interface{}) (int64, int64, error) {
sql, bindings := b.GetGrammar().CompileInsert(b.Query, values)
return b.Connection.Insert(sql, bindings...)
}
// Update a record in the database.
func (b *Builder) Update(value map[string]interface{}) (int64, error) {
sql := b.GetGrammar().CompileUpdate(b.Query, value)
cleanBindings := cleanBindings(b.GetGrammar().PrepareBindingsForUpdate(b.Bindings, value))
return b.Connection.Update(
sql,
cleanBindings...,
)
}
// Delete a record from the database.
func (b *Builder) Delete(args ...interface{}) (int64, error) {
if len(args) > 0 {
b.Where(b.Query.From+".id", args[0])
}
return b.Connection.Delete(
b.GetGrammar().CompileDelete(b.Query),
b.GetBindings()...,
)
}
// cleanBindings Remove all of the expressions from a list of bindings.
func cleanBindings(bindings []interface{}) []interface{} {
var result []interface{}
for _, b := range bindings {
if _, ok := b.(grammar.Expression); !ok {
result = append(result, b)
}
}
return result
}
func getDefaultBindings() map[string][]interface{} {
return map[string][]interface{}{
"select": make([]interface{}, 0),
"join": make([]interface{}, 0),
"where": make([]interface{}, 0),
"having": make([]interface{}, 0),
"order": make([]interface{}, 0),
"union": make([]interface{}, 0),
}
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。