mirror of
https://gitee.com/johng/gf.git
synced 2024-11-30 03:07:45 +08:00
95 lines
3.1 KiB
Go
95 lines
3.1 KiB
Go
// Copyright GoFrame Author(https://github.com/gogf/gf). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
package gdb
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/gogf/gf/text/gstr"
|
|
)
|
|
|
|
// isSubQuery checks and returns whether given string a sub-query sql string.
|
|
func isSubQuery(s string) bool {
|
|
s = gstr.TrimLeft(s, "()")
|
|
if p := gstr.Pos(s, " "); p != -1 {
|
|
if gstr.Equal(s[:p], "select") {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// LeftJoin does "LEFT JOIN ... ON ..." statement on the model.
|
|
// The parameter <table> can be joined table and its joined condition,
|
|
// and also with its alias name, like:
|
|
// Table("user").LeftJoin("user_detail", "user_detail.uid=user.uid")
|
|
// Table("user", "u").LeftJoin("user_detail", "ud", "ud.uid=u.uid")
|
|
// Table("user", "u").LeftJoin("SELECT xxx FROM xxx AS a", "a.uid=u.uid")
|
|
func (m *Model) LeftJoin(table ...string) *Model {
|
|
return m.doJoin("LEFT", table...)
|
|
}
|
|
|
|
// RightJoin does "RIGHT JOIN ... ON ..." statement on the model.
|
|
// The parameter <table> can be joined table and its joined condition,
|
|
// and also with its alias name, like:
|
|
// Table("user").RightJoin("user_detail", "user_detail.uid=user.uid")
|
|
// Table("user", "u").RightJoin("user_detail", "ud", "ud.uid=u.uid")
|
|
// Table("user", "u").RightJoin("SELECT xxx FROM xxx AS a", "a.uid=u.uid")
|
|
func (m *Model) RightJoin(table ...string) *Model {
|
|
return m.doJoin("RIGHT", table...)
|
|
}
|
|
|
|
// InnerJoin does "INNER JOIN ... ON ..." statement on the model.
|
|
// The parameter <table> can be joined table and its joined condition,
|
|
// and also with its alias name, like:
|
|
// Table("user").InnerJoin("user_detail", "user_detail.uid=user.uid")
|
|
// Table("user", "u").InnerJoin("user_detail", "ud", "ud.uid=u.uid")
|
|
// Table("user", "u").InnerJoin("SELECT xxx FROM xxx AS a", "a.uid=u.uid")
|
|
func (m *Model) InnerJoin(table ...string) *Model {
|
|
return m.doJoin("INNER", table...)
|
|
}
|
|
|
|
// doJoin does "LEFT/RIGHT/INNER JOIN ... ON ..." statement on the model.
|
|
// The parameter <table> can be joined table and its joined condition,
|
|
// and also with its alias name, like:
|
|
// Table("user").InnerJoin("user_detail", "user_detail.uid=user.uid")
|
|
// Table("user", "u").InnerJoin("user_detail", "ud", "ud.uid=u.uid")
|
|
// Table("user", "u").InnerJoin("SELECT xxx FROM xxx AS a", "a.uid=u.uid")
|
|
// Related issues:
|
|
// https://github.com/gogf/gf/issues/1024
|
|
func (m *Model) doJoin(operator string, table ...string) *Model {
|
|
var (
|
|
model = m.getModel()
|
|
joinStr = ""
|
|
)
|
|
if len(table) > 0 {
|
|
if isSubQuery(table[0]) {
|
|
joinStr = gstr.Trim(table[0])
|
|
if joinStr[0] != '(' {
|
|
joinStr = "(" + joinStr + ")"
|
|
}
|
|
} else {
|
|
joinStr = m.db.QuotePrefixTableName(table[0])
|
|
}
|
|
}
|
|
if len(table) > 2 {
|
|
model.tables += fmt.Sprintf(
|
|
" %s JOIN %s AS %s ON (%s)",
|
|
operator, joinStr, m.db.QuoteWord(table[1]), table[2],
|
|
)
|
|
} else if len(table) == 2 {
|
|
model.tables += fmt.Sprintf(
|
|
" %s JOIN %s ON (%s)",
|
|
operator, joinStr, table[1],
|
|
)
|
|
} else if len(table) == 1 {
|
|
model.tables += fmt.Sprintf(
|
|
" %s JOIN %s", operator, joinStr,
|
|
)
|
|
}
|
|
return model
|
|
}
|