123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441 |
- // Copyright 2019 The Xorm Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style
- // license that can be found in the LICENSE file.
-
- package statements
-
- import (
- "errors"
- "fmt"
- "reflect"
- "strings"
-
- "xorm.io/builder"
- "xorm.io/xorm/schemas"
- )
-
- func (statement *Statement) GenQuerySQL(sqlOrArgs ...interface{}) (string, []interface{}, error) {
- if len(sqlOrArgs) > 0 {
- return statement.ConvertSQLOrArgs(sqlOrArgs...)
- }
-
- if statement.RawSQL != "" {
- return statement.GenRawSQL(), statement.RawParams, nil
- }
-
- if len(statement.TableName()) <= 0 {
- return "", nil, ErrTableNotFound
- }
-
- var columnStr = statement.ColumnStr()
- if len(statement.SelectStr) > 0 {
- columnStr = statement.SelectStr
- } else {
- if statement.JoinStr == "" {
- if columnStr == "" {
- if statement.GroupByStr != "" {
- columnStr = statement.quoteColumnStr(statement.GroupByStr)
- } else {
- columnStr = statement.genColumnStr()
- }
- }
- } else {
- if columnStr == "" {
- if statement.GroupByStr != "" {
- columnStr = statement.quoteColumnStr(statement.GroupByStr)
- } else {
- columnStr = "*"
- }
- }
- }
- if columnStr == "" {
- columnStr = "*"
- }
- }
-
- if err := statement.ProcessIDParam(); err != nil {
- return "", nil, err
- }
-
- sqlStr, condArgs, err := statement.genSelectSQL(columnStr, true, true)
- if err != nil {
- return "", nil, err
- }
- args := append(statement.joinArgs, condArgs...)
-
- // for mssql and use limit
- qs := strings.Count(sqlStr, "?")
- if len(args)*2 == qs {
- args = append(args, args...)
- }
-
- return sqlStr, args, nil
- }
-
- func (statement *Statement) GenSumSQL(bean interface{}, columns ...string) (string, []interface{}, error) {
- if statement.RawSQL != "" {
- return statement.GenRawSQL(), statement.RawParams, nil
- }
-
- statement.SetRefBean(bean)
-
- var sumStrs = make([]string, 0, len(columns))
- for _, colName := range columns {
- if !strings.Contains(colName, " ") && !strings.Contains(colName, "(") {
- colName = statement.quote(colName)
- } else {
- colName = statement.ReplaceQuote(colName)
- }
- sumStrs = append(sumStrs, fmt.Sprintf("COALESCE(sum(%s),0)", colName))
- }
- sumSelect := strings.Join(sumStrs, ", ")
-
- if err := statement.mergeConds(bean); err != nil {
- return "", nil, err
- }
-
- sqlStr, condArgs, err := statement.genSelectSQL(sumSelect, true, true)
- if err != nil {
- return "", nil, err
- }
-
- return sqlStr, append(statement.joinArgs, condArgs...), nil
- }
-
- func (statement *Statement) GenGetSQL(bean interface{}) (string, []interface{}, error) {
- v := rValue(bean)
- isStruct := v.Kind() == reflect.Struct
- if isStruct {
- statement.SetRefBean(bean)
- }
-
- var columnStr = statement.ColumnStr()
- if len(statement.SelectStr) > 0 {
- columnStr = statement.SelectStr
- } else {
- // TODO: always generate column names, not use * even if join
- if len(statement.JoinStr) == 0 {
- if len(columnStr) == 0 {
- if len(statement.GroupByStr) > 0 {
- columnStr = statement.quoteColumnStr(statement.GroupByStr)
- } else {
- columnStr = statement.genColumnStr()
- }
- }
- } else {
- if len(columnStr) == 0 {
- if len(statement.GroupByStr) > 0 {
- columnStr = statement.quoteColumnStr(statement.GroupByStr)
- }
- }
- }
- }
-
- if len(columnStr) == 0 {
- columnStr = "*"
- }
-
- if isStruct {
- if err := statement.mergeConds(bean); err != nil {
- return "", nil, err
- }
- } else {
- if err := statement.ProcessIDParam(); err != nil {
- return "", nil, err
- }
- }
-
- sqlStr, condArgs, err := statement.genSelectSQL(columnStr, true, true)
- if err != nil {
- return "", nil, err
- }
-
- return sqlStr, append(statement.joinArgs, condArgs...), nil
- }
-
- // GenCountSQL generates the SQL for counting
- func (statement *Statement) GenCountSQL(beans ...interface{}) (string, []interface{}, error) {
- if statement.RawSQL != "" {
- return statement.GenRawSQL(), statement.RawParams, nil
- }
-
- var condArgs []interface{}
- var err error
- if len(beans) > 0 {
- statement.SetRefBean(beans[0])
- if err := statement.mergeConds(beans[0]); err != nil {
- return "", nil, err
- }
- }
-
- var selectSQL = statement.SelectStr
- if len(selectSQL) <= 0 {
- if statement.IsDistinct {
- selectSQL = fmt.Sprintf("count(DISTINCT %s)", statement.ColumnStr())
- } else if statement.ColumnStr() != "" {
- selectSQL = fmt.Sprintf("count(%s)", statement.ColumnStr())
- } else {
- selectSQL = "count(*)"
- }
- }
- sqlStr, condArgs, err := statement.genSelectSQL(selectSQL, false, false)
- if err != nil {
- return "", nil, err
- }
-
- return sqlStr, append(statement.joinArgs, condArgs...), nil
- }
-
- func (statement *Statement) genSelectSQL(columnStr string, needLimit, needOrderBy bool) (string, []interface{}, error) {
- var (
- distinct string
- dialect = statement.dialect
- quote = statement.quote
- fromStr = " FROM "
- top, mssqlCondi, whereStr string
- )
- if statement.IsDistinct && !strings.HasPrefix(columnStr, "count") {
- distinct = "DISTINCT "
- }
-
- condSQL, condArgs, err := statement.GenCondSQL(statement.cond)
- if err != nil {
- return "", nil, err
- }
- if len(condSQL) > 0 {
- whereStr = " WHERE " + condSQL
- }
-
- if dialect.URI().DBType == schemas.MSSQL && strings.Contains(statement.TableName(), "..") {
- fromStr += statement.TableName()
- } else {
- fromStr += quote(statement.TableName())
- }
-
- if statement.TableAlias != "" {
- if dialect.URI().DBType == schemas.ORACLE {
- fromStr += " " + quote(statement.TableAlias)
- } else {
- fromStr += " AS " + quote(statement.TableAlias)
- }
- }
- if statement.JoinStr != "" {
- fromStr = fmt.Sprintf("%v %v", fromStr, statement.JoinStr)
- }
-
- pLimitN := statement.LimitN
- if dialect.URI().DBType == schemas.MSSQL {
- if pLimitN != nil {
- LimitNValue := *pLimitN
- top = fmt.Sprintf("TOP %d ", LimitNValue)
- }
- if statement.Start > 0 {
- var column string
- if len(statement.RefTable.PKColumns()) == 0 {
- for _, index := range statement.RefTable.Indexes {
- if len(index.Cols) == 1 {
- column = index.Cols[0]
- break
- }
- }
- if len(column) == 0 {
- column = statement.RefTable.ColumnsSeq()[0]
- }
- } else {
- column = statement.RefTable.PKColumns()[0].Name
- }
- if statement.needTableName() {
- if len(statement.TableAlias) > 0 {
- column = statement.TableAlias + "." + column
- } else {
- column = statement.TableName() + "." + column
- }
- }
-
- var orderStr string
- if needOrderBy && len(statement.OrderStr) > 0 {
- orderStr = " ORDER BY " + statement.OrderStr
- }
-
- var groupStr string
- if len(statement.GroupByStr) > 0 {
- groupStr = " GROUP BY " + statement.GroupByStr
- }
- mssqlCondi = fmt.Sprintf("(%s NOT IN (SELECT TOP %d %s%s%s%s%s))",
- column, statement.Start, column, fromStr, whereStr, orderStr, groupStr)
- }
- }
-
- var buf strings.Builder
- fmt.Fprintf(&buf, "SELECT %v%v%v%v%v", distinct, top, columnStr, fromStr, whereStr)
- if len(mssqlCondi) > 0 {
- if len(whereStr) > 0 {
- fmt.Fprint(&buf, " AND ", mssqlCondi)
- } else {
- fmt.Fprint(&buf, " WHERE ", mssqlCondi)
- }
- }
-
- if statement.GroupByStr != "" {
- fmt.Fprint(&buf, " GROUP BY ", statement.GroupByStr)
- }
- if statement.HavingStr != "" {
- fmt.Fprint(&buf, " ", statement.HavingStr)
- }
- if needOrderBy && statement.OrderStr != "" {
- fmt.Fprint(&buf, " ORDER BY ", statement.OrderStr)
- }
- if needLimit {
- if dialect.URI().DBType != schemas.MSSQL && dialect.URI().DBType != schemas.ORACLE {
- if statement.Start > 0 {
- if pLimitN != nil {
- fmt.Fprintf(&buf, " LIMIT %v OFFSET %v", *pLimitN, statement.Start)
- } else {
- fmt.Fprintf(&buf, "LIMIT 0 OFFSET %v", statement.Start)
- }
- } else if pLimitN != nil {
- fmt.Fprint(&buf, " LIMIT ", *pLimitN)
- }
- } else if dialect.URI().DBType == schemas.ORACLE {
- if statement.Start != 0 || pLimitN != nil {
- oldString := buf.String()
- buf.Reset()
- rawColStr := columnStr
- if rawColStr == "*" {
- rawColStr = "at.*"
- }
- fmt.Fprintf(&buf, "SELECT %v FROM (SELECT %v,ROWNUM RN FROM (%v) at WHERE ROWNUM <= %d) aat WHERE RN > %d",
- columnStr, rawColStr, oldString, statement.Start+*pLimitN, statement.Start)
- }
- }
- }
- if statement.IsForUpdate {
- return dialect.ForUpdateSQL(buf.String()), condArgs, nil
- }
-
- return buf.String(), condArgs, nil
- }
-
- func (statement *Statement) GenExistSQL(bean ...interface{}) (string, []interface{}, error) {
- if statement.RawSQL != "" {
- return statement.GenRawSQL(), statement.RawParams, nil
- }
-
- var sqlStr string
- var args []interface{}
- var joinStr string
- var err error
- if len(bean) == 0 {
- tableName := statement.TableName()
- if len(tableName) <= 0 {
- return "", nil, ErrTableNotFound
- }
-
- tableName = statement.quote(tableName)
- if len(statement.JoinStr) > 0 {
- joinStr = statement.JoinStr
- }
-
- if statement.Conds().IsValid() {
- condSQL, condArgs, err := statement.GenCondSQL(statement.Conds())
- if err != nil {
- return "", nil, err
- }
-
- if statement.dialect.URI().DBType == schemas.MSSQL {
- sqlStr = fmt.Sprintf("SELECT TOP 1 * FROM %s %s WHERE %s", tableName, joinStr, condSQL)
- } else if statement.dialect.URI().DBType == schemas.ORACLE {
- sqlStr = fmt.Sprintf("SELECT * FROM %s WHERE (%s) %s AND ROWNUM=1", tableName, joinStr, condSQL)
- } else {
- sqlStr = fmt.Sprintf("SELECT * FROM %s %s WHERE %s LIMIT 1", tableName, joinStr, condSQL)
- }
- args = condArgs
- } else {
- if statement.dialect.URI().DBType == schemas.MSSQL {
- sqlStr = fmt.Sprintf("SELECT TOP 1 * FROM %s %s", tableName, joinStr)
- } else if statement.dialect.URI().DBType == schemas.ORACLE {
- sqlStr = fmt.Sprintf("SELECT * FROM %s %s WHERE ROWNUM=1", tableName, joinStr)
- } else {
- sqlStr = fmt.Sprintf("SELECT * FROM %s %s LIMIT 1", tableName, joinStr)
- }
- args = []interface{}{}
- }
- } else {
- beanValue := reflect.ValueOf(bean[0])
- if beanValue.Kind() != reflect.Ptr {
- return "", nil, errors.New("needs a pointer")
- }
-
- if beanValue.Elem().Kind() == reflect.Struct {
- if err := statement.SetRefBean(bean[0]); err != nil {
- return "", nil, err
- }
- }
-
- if len(statement.TableName()) <= 0 {
- return "", nil, ErrTableNotFound
- }
- statement.Limit(1)
- sqlStr, args, err = statement.GenGetSQL(bean[0])
- if err != nil {
- return "", nil, err
- }
- }
-
- return sqlStr, args, nil
- }
-
- func (statement *Statement) GenFindSQL(autoCond builder.Cond) (string, []interface{}, error) {
- if statement.RawSQL != "" {
- return statement.GenRawSQL(), statement.RawParams, nil
- }
-
- var sqlStr string
- var args []interface{}
- var err error
-
- if len(statement.TableName()) <= 0 {
- return "", nil, ErrTableNotFound
- }
-
- var columnStr = statement.ColumnStr()
- if len(statement.SelectStr) > 0 {
- columnStr = statement.SelectStr
- } else {
- if statement.JoinStr == "" {
- if columnStr == "" {
- if statement.GroupByStr != "" {
- columnStr = statement.quoteColumnStr(statement.GroupByStr)
- } else {
- columnStr = statement.genColumnStr()
- }
- }
- } else {
- if columnStr == "" {
- if statement.GroupByStr != "" {
- columnStr = statement.quoteColumnStr(statement.GroupByStr)
- } else {
- columnStr = "*"
- }
- }
- }
- if columnStr == "" {
- columnStr = "*"
- }
- }
-
- statement.cond = statement.cond.And(autoCond)
-
- sqlStr, condArgs, err := statement.genSelectSQL(columnStr, true, true)
- if err != nil {
- return "", nil, err
- }
- args = append(statement.joinArgs, condArgs...)
- // for mssql and use limit
- qs := strings.Count(sqlStr, "?")
- if len(args)*2 == qs {
- args = append(args, args...)
- }
-
- return sqlStr, args, nil
- }
|