123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829 |
- // Copyright GoFrame Author(https://goframe.org). 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 (
- "bytes"
- "database/sql"
- "fmt"
- "github.com/gogf/gf/errors/gcode"
- "reflect"
- "regexp"
- "strings"
- "time"
- "github.com/gogf/gf/errors/gerror"
- "github.com/gogf/gf/internal/empty"
- "github.com/gogf/gf/internal/json"
- "github.com/gogf/gf/internal/utils"
- "github.com/gogf/gf/os/gtime"
- "github.com/gogf/gf/util/gmeta"
- "github.com/gogf/gf/util/gutil"
- "github.com/gogf/gf/internal/structs"
- "github.com/gogf/gf/text/gregex"
- "github.com/gogf/gf/text/gstr"
- "github.com/gogf/gf/util/gconv"
- )
- // apiString is the type assert api for String.
- type apiString interface {
- String() string
- }
- // apiIterator is the type assert api for Iterator.
- type apiIterator interface {
- Iterator(f func(key, value interface{}) bool)
- }
- // apiInterfaces is the type assert api for Interfaces.
- type apiInterfaces interface {
- Interfaces() []interface{}
- }
- // apiMapStrAny is the interface support for converting struct parameter to map.
- type apiMapStrAny interface {
- MapStrAny() map[string]interface{}
- }
- // apiTableName is the interface for retrieving table name fro struct.
- type apiTableName interface {
- TableName() string
- }
- const (
- OrmTagForStruct = "orm"
- OrmTagForUnique = "unique"
- OrmTagForPrimary = "primary"
- OrmTagForTable = "table"
- OrmTagForWith = "with"
- OrmTagForWithWhere = "where"
- OrmTagForWithOrder = "order"
- )
- var (
- // quoteWordReg is the regular expression object for a word check.
- quoteWordReg = regexp.MustCompile(`^[a-zA-Z0-9\-_]+$`)
- // Priority tags for struct converting for orm field mapping.
- structTagPriority = append([]string{OrmTagForStruct}, gconv.StructTagPriority...)
- )
- // guessPrimaryTableName parses and returns the primary table name.
- func (m *Model) guessPrimaryTableName(tableStr string) string {
- if tableStr == "" {
- return ""
- }
- var (
- guessedTableName = ""
- array1 = gstr.SplitAndTrim(tableStr, ",")
- array2 = gstr.SplitAndTrim(array1[0], " ")
- array3 = gstr.SplitAndTrim(array2[0], ".")
- )
- if len(array3) >= 2 {
- guessedTableName = array3[1]
- } else {
- guessedTableName = array3[0]
- }
- charL, charR := m.db.GetChars()
- if charL != "" || charR != "" {
- guessedTableName = gstr.Trim(guessedTableName, charL+charR)
- }
- if !gregex.IsMatchString(regularFieldNameRegPattern, guessedTableName) {
- return ""
- }
- return guessedTableName
- }
- // getTableNameFromOrmTag retrieves and returns the table name from struct object.
- func getTableNameFromOrmTag(object interface{}) string {
- var tableName string
- // Use the interface value.
- if r, ok := object.(apiTableName); ok {
- tableName = r.TableName()
- }
- // User meta data tag "orm".
- if tableName == "" {
- if ormTag := gmeta.Get(object, OrmTagForStruct); !ormTag.IsEmpty() {
- match, _ := gregex.MatchString(
- fmt.Sprintf(`%s\s*:\s*([^,]+)`, OrmTagForTable),
- ormTag.String(),
- )
- if len(match) > 1 {
- tableName = match[1]
- }
- }
- }
- // Use the struct name of snake case.
- if tableName == "" {
- if t, err := structs.StructType(object); err != nil {
- panic(err)
- } else {
- tableName = gstr.CaseSnakeFirstUpper(
- gstr.StrEx(t.String(), "."),
- )
- }
- }
- return tableName
- }
- // ListItemValues retrieves and returns the elements of all item struct/map with key `key`.
- // Note that the parameter `list` should be type of slice which contains elements of map or struct,
- // or else it returns an empty slice.
- //
- // The parameter `list` supports types like:
- // []map[string]interface{}
- // []map[string]sub-map
- // []struct
- // []struct:sub-struct
- // Note that the sub-map/sub-struct makes sense only if the optional parameter `subKey` is given.
- // See gutil.ListItemValues.
- func ListItemValues(list interface{}, key interface{}, subKey ...interface{}) (values []interface{}) {
- return gutil.ListItemValues(list, key, subKey...)
- }
- // ListItemValuesUnique retrieves and returns the unique elements of all struct/map with key `key`.
- // Note that the parameter `list` should be type of slice which contains elements of map or struct,
- // or else it returns an empty slice.
- // See gutil.ListItemValuesUnique.
- func ListItemValuesUnique(list interface{}, key string, subKey ...interface{}) []interface{} {
- return gutil.ListItemValuesUnique(list, key, subKey...)
- }
- // GetInsertOperationByOption returns proper insert option with given parameter `option`.
- func GetInsertOperationByOption(option int) string {
- var operator string
- switch option {
- case insertOptionReplace:
- operator = "REPLACE"
- case insertOptionIgnore:
- operator = "INSERT IGNORE"
- default:
- operator = "INSERT"
- }
- return operator
- }
- // ConvertDataForTableRecord is a very important function, which does converting for any data that
- // will be inserted into table as a record.
- //
- // The parameter `value` should be type of *map/map/*struct/struct.
- // It supports embedded struct definition for struct.
- func ConvertDataForTableRecord(value interface{}) map[string]interface{} {
- var (
- rvValue reflect.Value
- rvKind reflect.Kind
- data = DataToMapDeep(value)
- )
- for k, v := range data {
- rvValue = reflect.ValueOf(v)
- rvKind = rvValue.Kind()
- for rvKind == reflect.Ptr {
- rvValue = rvValue.Elem()
- rvKind = rvValue.Kind()
- }
- switch rvKind {
- case reflect.Slice, reflect.Array, reflect.Map:
- // It should ignore the bytes type.
- if _, ok := v.([]byte); !ok {
- // Convert the value to JSON.
- data[k], _ = json.Marshal(v)
- }
- case reflect.Struct:
- switch r := v.(type) {
- // If the time is zero, it then updates it to nil,
- // which will insert/update the value to database as "null".
- case time.Time:
- if r.IsZero() {
- data[k] = nil
- }
- case gtime.Time:
- if r.IsZero() {
- data[k] = nil
- }
- case *gtime.Time:
- if r.IsZero() {
- data[k] = nil
- }
- case *time.Time:
- continue
- case Counter, *Counter:
- continue
- default:
- // Use string conversion in default.
- if s, ok := v.(apiString); ok {
- data[k] = s.String()
- } else {
- // Convert the value to JSON.
- data[k], _ = json.Marshal(v)
- }
- }
- }
- }
- return data
- }
- // DataToMapDeep converts `value` to map type recursively(if attribute struct is embedded).
- // The parameter `value` should be type of *map/map/*struct/struct.
- // It supports embedded struct definition for struct.
- func DataToMapDeep(value interface{}) map[string]interface{} {
- m := gconv.Map(value, structTagPriority...)
- for k, v := range m {
- switch v.(type) {
- case time.Time, *time.Time, gtime.Time, *gtime.Time:
- m[k] = v
- default:
- // Use string conversion in default.
- if s, ok := v.(apiString); ok {
- m[k] = s.String()
- } else {
- m[k] = v
- }
- }
- }
- return m
- }
- // doHandleTableName adds prefix string and quote chars for the table. It handles table string like:
- // "user", "user u", "user,user_detail", "user u, user_detail ut", "user as u, user_detail as ut",
- // "user.user u", "`user`.`user` u".
- //
- // Note that, this will automatically checks the table prefix whether already added, if true it does
- // nothing to the table name, or else adds the prefix to the table name.
- func doHandleTableName(table, prefix, charLeft, charRight string) string {
- var (
- index = 0
- chars = charLeft + charRight
- array1 = gstr.SplitAndTrim(table, ",")
- )
- for k1, v1 := range array1 {
- array2 := gstr.SplitAndTrim(v1, " ")
- // Trim the security chars.
- array2[0] = gstr.Trim(array2[0], chars)
- // Check whether it has database name.
- array3 := gstr.Split(gstr.Trim(array2[0]), ".")
- for k, v := range array3 {
- array3[k] = gstr.Trim(v, chars)
- }
- index = len(array3) - 1
- // If the table name already has the prefix, skips the prefix adding.
- if len(array3[index]) <= len(prefix) || array3[index][:len(prefix)] != prefix {
- array3[index] = prefix + array3[index]
- }
- array2[0] = gstr.Join(array3, ".")
- // Add the security chars.
- array2[0] = doQuoteString(array2[0], charLeft, charRight)
- array1[k1] = gstr.Join(array2, " ")
- }
- return gstr.Join(array1, ",")
- }
- // doQuoteWord checks given string `s` a word, if true quotes it with `charLeft` and `charRight`
- // and returns the quoted string; or else returns `s` without any change.
- func doQuoteWord(s, charLeft, charRight string) string {
- if quoteWordReg.MatchString(s) && !gstr.ContainsAny(s, charLeft+charRight) {
- return charLeft + s + charRight
- }
- return s
- }
- // doQuoteString quotes string with quote chars.
- // For example, if quote char is '`':
- // "user" => "`user`"
- // "user u" => "`user` u"
- // "user,user_detail" => "`user`,`user_detail`"
- // "user u, user_detail ut" => "`user` u,`user_detail` ut"
- // "user.user u, user.user_detail ut" => "`user`.`user` u,`user`.`user_detail` ut"
- // "u.id, u.name, u.age" => "`u`.`id`,`u`.`name`,`u`.`age`"
- // "u.id asc" => "`u`.`id` asc"
- func doQuoteString(s, charLeft, charRight string) string {
- array1 := gstr.SplitAndTrim(s, ",")
- for k1, v1 := range array1 {
- array2 := gstr.SplitAndTrim(v1, " ")
- array3 := gstr.Split(gstr.Trim(array2[0]), ".")
- if len(array3) == 1 {
- array3[0] = doQuoteWord(array3[0], charLeft, charRight)
- } else if len(array3) >= 2 {
- array3[0] = doQuoteWord(array3[0], charLeft, charRight)
- // Note:
- // mysql: u.uid
- // mssql double dots: Database..Table
- array3[len(array3)-1] = doQuoteWord(array3[len(array3)-1], charLeft, charRight)
- }
- array2[0] = gstr.Join(array3, ".")
- array1[k1] = gstr.Join(array2, " ")
- }
- return gstr.Join(array1, ",")
- }
- // GetWhereConditionOfStruct returns the where condition sql and arguments by given struct pointer.
- // This function automatically retrieves primary or unique field and its attribute value as condition.
- func GetWhereConditionOfStruct(pointer interface{}) (where string, args []interface{}, err error) {
- tagField, err := structs.TagFields(pointer, []string{OrmTagForStruct})
- if err != nil {
- return "", nil, err
- }
- array := ([]string)(nil)
- for _, field := range tagField {
- array = strings.Split(field.TagValue, ",")
- if len(array) > 1 && gstr.InArray([]string{OrmTagForUnique, OrmTagForPrimary}, array[1]) {
- return array[0], []interface{}{field.Value.Interface()}, nil
- }
- if len(where) > 0 {
- where += " AND "
- }
- where += field.TagValue + "=?"
- args = append(args, field.Value.Interface())
- }
- return
- }
- // GetPrimaryKey retrieves and returns primary key field name from given struct.
- func GetPrimaryKey(pointer interface{}) (string, error) {
- tagField, err := structs.TagFields(pointer, []string{OrmTagForStruct})
- if err != nil {
- return "", err
- }
- array := ([]string)(nil)
- for _, field := range tagField {
- array = strings.Split(field.TagValue, ",")
- if len(array) > 1 && array[1] == OrmTagForPrimary {
- return array[0], nil
- }
- }
- return "", nil
- }
- // GetPrimaryKeyCondition returns a new where condition by primary field name.
- // The optional parameter `where` is like follows:
- // 123 => primary=123
- // []int{1, 2, 3} => primary IN(1,2,3)
- // "john" => primary='john'
- // []string{"john", "smith"} => primary IN('john','smith')
- // g.Map{"id": g.Slice{1,2,3}} => id IN(1,2,3)
- // g.Map{"id": 1, "name": "john"} => id=1 AND name='john'
- // etc.
- //
- // Note that it returns the given `where` parameter directly if the `primary` is empty
- // or length of `where` > 1.
- func GetPrimaryKeyCondition(primary string, where ...interface{}) (newWhereCondition []interface{}) {
- if len(where) == 0 {
- return nil
- }
- if primary == "" {
- return where
- }
- if len(where) == 1 {
- var (
- rv = reflect.ValueOf(where[0])
- kind = rv.Kind()
- )
- if kind == reflect.Ptr {
- rv = rv.Elem()
- kind = rv.Kind()
- }
- switch kind {
- case reflect.Map, reflect.Struct:
- // Ignore the parameter `primary`.
- break
- default:
- return []interface{}{map[string]interface{}{
- primary: where[0],
- }}
- }
- }
- return where
- }
- // formatSql formats the sql string and its arguments before executing.
- // The internal handleArguments function might be called twice during the SQL procedure,
- // but do not worry about it, it's safe and efficient.
- func formatSql(sql string, args []interface{}) (newSql string, newArgs []interface{}) {
- // DO NOT do this as there may be multiple lines and comments in the sql.
- // sql = gstr.Trim(sql)
- // sql = gstr.Replace(sql, "\n", " ")
- // sql, _ = gregex.ReplaceString(`\s{2,}`, ` `, sql)
- return handleArguments(sql, args)
- }
- type formatWhereInput struct {
- Where interface{}
- Args []interface{}
- OmitNil bool
- OmitEmpty bool
- Schema string
- Table string
- }
- // formatWhere formats where statement and its arguments for `Where` and `Having` statements.
- func formatWhere(db DB, in formatWhereInput) (newWhere string, newArgs []interface{}) {
- var (
- buffer = bytes.NewBuffer(nil)
- reflectValue = reflect.ValueOf(in.Where)
- reflectKind = reflectValue.Kind()
- )
- for reflectKind == reflect.Ptr {
- reflectValue = reflectValue.Elem()
- reflectKind = reflectValue.Kind()
- }
- switch reflectKind {
- case reflect.Array, reflect.Slice:
- newArgs = formatWhereInterfaces(db, gconv.Interfaces(in.Where), buffer, newArgs)
- case reflect.Map:
- for key, value := range DataToMapDeep(in.Where) {
- if gregex.IsMatchString(regularFieldNameRegPattern, key) {
- if in.OmitNil && empty.IsNil(value) {
- continue
- }
- if in.OmitEmpty && empty.IsEmpty(value) {
- continue
- }
- }
- newArgs = formatWhereKeyValue(db, buffer, newArgs, key, value)
- }
- case reflect.Struct:
- // If `where` struct implements apiIterator interface,
- // it then uses its Iterate function to iterate its key-value pairs.
- // For example, ListMap and TreeMap are ordered map,
- // which implement apiIterator interface and are index-friendly for where conditions.
- if iterator, ok := in.Where.(apiIterator); ok {
- iterator.Iterator(func(key, value interface{}) bool {
- ketStr := gconv.String(key)
- if gregex.IsMatchString(regularFieldNameRegPattern, ketStr) {
- if in.OmitNil && empty.IsNil(value) {
- return true
- }
- if in.OmitEmpty && empty.IsEmpty(value) {
- return true
- }
- }
- newArgs = formatWhereKeyValue(db, buffer, newArgs, ketStr, value)
- return true
- })
- break
- }
- // Automatically mapping and filtering the struct attribute.
- var (
- reflectType = reflectValue.Type()
- structField reflect.StructField
- )
- data := DataToMapDeep(in.Where)
- if in.Table != "" {
- data, _ = db.GetCore().mappingAndFilterData(in.Schema, in.Table, data, true)
- }
- // Put the struct attributes in sequence in Where statement.
- for i := 0; i < reflectType.NumField(); i++ {
- structField = reflectType.Field(i)
- foundKey, foundValue := gutil.MapPossibleItemByKey(data, structField.Name)
- if foundKey != "" {
- if in.OmitNil && empty.IsNil(foundValue) {
- continue
- }
- if in.OmitEmpty && empty.IsEmpty(foundValue) {
- continue
- }
- newArgs = formatWhereKeyValue(db, buffer, newArgs, foundKey, foundValue)
- }
- }
- default:
- // Usually a string.
- var (
- i = 0
- whereStr = gconv.String(in.Where)
- )
- for {
- if i >= len(in.Args) {
- break
- }
- // Sub query, which is always used along with a string condition.
- if model, ok := in.Args[i].(*Model); ok {
- var (
- index = -1
- )
- whereStr, _ = gregex.ReplaceStringFunc(`(\?)`, whereStr, func(s string) string {
- index++
- if i+len(newArgs) == index {
- sqlWithHolder, holderArgs := model.getFormattedSqlAndArgs(queryTypeNormal, false)
- newArgs = append(newArgs, holderArgs...)
- // Automatically adding the brackets.
- return "(" + sqlWithHolder + ")"
- }
- return s
- })
- in.Args = gutil.SliceDelete(in.Args, i)
- continue
- }
- i++
- }
- buffer.WriteString(whereStr)
- }
- if buffer.Len() == 0 {
- return "", in.Args
- }
- newArgs = append(newArgs, in.Args...)
- newWhere = buffer.String()
- if len(newArgs) > 0 {
- if gstr.Pos(newWhere, "?") == -1 {
- if gregex.IsMatchString(lastOperatorRegPattern, newWhere) {
- // Eg: Where/And/Or("uid>=", 1)
- newWhere += "?"
- } else if gregex.IsMatchString(regularFieldNameRegPattern, newWhere) {
- newWhere = db.GetCore().QuoteString(newWhere)
- if len(newArgs) > 0 {
- if utils.IsArray(newArgs[0]) {
- // Eg:
- // Where("id", []int{1,2,3})
- // Where("user.id", []int{1,2,3})
- newWhere += " IN (?)"
- } else if empty.IsNil(newArgs[0]) {
- // Eg:
- // Where("id", nil)
- // Where("user.id", nil)
- newWhere += " IS NULL"
- newArgs = nil
- } else {
- // Eg:
- // Where/And/Or("uid", 1)
- // Where/And/Or("user.uid", 1)
- newWhere += "=?"
- }
- }
- }
- }
- }
- return handleArguments(newWhere, newArgs)
- }
- // formatWhereInterfaces formats `where` as []interface{}.
- func formatWhereInterfaces(db DB, where []interface{}, buffer *bytes.Buffer, newArgs []interface{}) []interface{} {
- if len(where) == 0 {
- return newArgs
- }
- if len(where)%2 != 0 {
- buffer.WriteString(gstr.Join(gconv.Strings(where), ""))
- return newArgs
- }
- var str string
- for i := 0; i < len(where); i += 2 {
- str = gconv.String(where[i])
- if buffer.Len() > 0 {
- buffer.WriteString(" AND " + db.GetCore().QuoteWord(str) + "=?")
- } else {
- buffer.WriteString(db.GetCore().QuoteWord(str) + "=?")
- }
- if s, ok := where[i+1].(Raw); ok {
- buffer.WriteString(gconv.String(s))
- } else {
- newArgs = append(newArgs, where[i+1])
- }
- }
- return newArgs
- }
- // formatWhereKeyValue handles each key-value pair of the parameter map.
- func formatWhereKeyValue(db DB, buffer *bytes.Buffer, newArgs []interface{}, key string, value interface{}) []interface{} {
- quotedKey := db.GetCore().QuoteWord(key)
- if buffer.Len() > 0 {
- buffer.WriteString(" AND ")
- }
- // If the value is type of slice, and there's only one '?' holder in
- // the key string, it automatically adds '?' holder chars according to its arguments count
- // and converts it to "IN" statement.
- var (
- rv = reflect.ValueOf(value)
- kind = rv.Kind()
- )
- switch kind {
- case reflect.Slice, reflect.Array:
- count := gstr.Count(quotedKey, "?")
- if count == 0 {
- buffer.WriteString(quotedKey + " IN(?)")
- newArgs = append(newArgs, value)
- } else if count != rv.Len() {
- buffer.WriteString(quotedKey)
- newArgs = append(newArgs, value)
- } else {
- buffer.WriteString(quotedKey)
- newArgs = append(newArgs, gconv.Interfaces(value)...)
- }
- default:
- if value == nil || empty.IsNil(rv) {
- if gregex.IsMatchString(regularFieldNameRegPattern, key) {
- // The key is a single field name.
- buffer.WriteString(quotedKey + " IS NULL")
- } else {
- // The key may have operation chars.
- buffer.WriteString(quotedKey)
- }
- } else {
- // It also supports "LIKE" statement, which we considers it an operator.
- quotedKey = gstr.Trim(quotedKey)
- if gstr.Pos(quotedKey, "?") == -1 {
- like := " like"
- if len(quotedKey) > len(like) && gstr.Equal(quotedKey[len(quotedKey)-len(like):], like) {
- // Eg: Where(g.Map{"name like": "john%"})
- buffer.WriteString(quotedKey + " ?")
- } else if gregex.IsMatchString(lastOperatorRegPattern, quotedKey) {
- // Eg: Where(g.Map{"age > ": 16})
- buffer.WriteString(quotedKey + " ?")
- } else if gregex.IsMatchString(regularFieldNameRegPattern, key) {
- // The key is a regular field name.
- buffer.WriteString(quotedKey + "=?")
- } else {
- // The key is not a regular field name.
- // Eg: Where(g.Map{"age > 16": nil})
- // Issue: https://github.com/gogf/gf/issues/765
- if empty.IsEmpty(value) {
- buffer.WriteString(quotedKey)
- break
- } else {
- buffer.WriteString(quotedKey + "=?")
- }
- }
- } else {
- buffer.WriteString(quotedKey)
- }
- if s, ok := value.(Raw); ok {
- buffer.WriteString(gconv.String(s))
- } else {
- newArgs = append(newArgs, value)
- }
- }
- }
- return newArgs
- }
- // handleArguments is an important function, which handles the sql and all its arguments
- // before committing them to underlying driver.
- func handleArguments(sql string, args []interface{}) (newSql string, newArgs []interface{}) {
- newSql = sql
- // insertHolderCount is used to calculate the inserting position for the '?' holder.
- insertHolderCount := 0
- // Handles the slice arguments.
- if len(args) > 0 {
- for index, arg := range args {
- var (
- reflectValue = reflect.ValueOf(arg)
- reflectKind = reflectValue.Kind()
- )
- for reflectKind == reflect.Ptr {
- reflectValue = reflectValue.Elem()
- reflectKind = reflectValue.Kind()
- }
- switch reflectKind {
- case reflect.Slice, reflect.Array:
- // It does not split the type of []byte.
- // Eg: table.Where("name = ?", []byte("john"))
- if _, ok := arg.([]byte); ok {
- newArgs = append(newArgs, arg)
- continue
- }
- if reflectValue.Len() == 0 {
- // Empty slice argument, it converts the sql to a false sql.
- // Eg:
- // Query("select * from xxx where id in(?)", g.Slice{}) -> select * from xxx where 0=1
- // Where("id in(?)", g.Slice{}) -> WHERE 0=1
- if gstr.Contains(newSql, "?") {
- whereKeyWord := " WHERE "
- if p := gstr.PosI(newSql, whereKeyWord); p == -1 {
- return "0=1", []interface{}{}
- } else {
- return gstr.SubStr(newSql, 0, p+len(whereKeyWord)) + "0=1", []interface{}{}
- }
- }
- } else {
- for i := 0; i < reflectValue.Len(); i++ {
- newArgs = append(newArgs, reflectValue.Index(i).Interface())
- }
- }
- // If the '?' holder count equals the length of the slice,
- // it does not implement the arguments splitting logic.
- // Eg: db.Query("SELECT ?+?", g.Slice{1, 2})
- if len(args) == 1 && gstr.Count(newSql, "?") == reflectValue.Len() {
- break
- }
- // counter is used to finding the inserting position for the '?' holder.
- var (
- counter = 0
- replaced = false
- )
- newSql, _ = gregex.ReplaceStringFunc(`\?`, newSql, func(s string) string {
- if replaced {
- return s
- }
- counter++
- if counter == index+insertHolderCount+1 {
- replaced = true
- insertHolderCount += reflectValue.Len() - 1
- return "?" + strings.Repeat(",?", reflectValue.Len()-1)
- }
- return s
- })
- // Special struct handling.
- case reflect.Struct:
- switch v := arg.(type) {
- // The underlying driver supports time.Time/*time.Time types.
- case time.Time, *time.Time:
- newArgs = append(newArgs, arg)
- continue
- // Special handling for gtime.Time/*gtime.Time.
- //
- // DO NOT use its underlying gtime.Time.Time as its argument,
- // because the std time.Time will be converted to certain timezone
- // according to underlying driver. And the underlying driver also
- // converts the time.Time to string automatically as the following does.
- case gtime.Time:
- newArgs = append(newArgs, v.String())
- continue
- case *gtime.Time:
- newArgs = append(newArgs, v.String())
- continue
- default:
- // It converts the struct to string in default
- // if it has implemented the String interface.
- if v, ok := arg.(apiString); ok {
- newArgs = append(newArgs, v.String())
- continue
- }
- }
- newArgs = append(newArgs, arg)
- default:
- newArgs = append(newArgs, arg)
- }
- }
- }
- return
- }
- // formatError customizes and returns the SQL error.
- func formatError(err error, s string, args ...interface{}) error {
- if err != nil && err != sql.ErrNoRows {
- return gerror.NewCodef(gcode.CodeDbOperationError, "%s, %s\n", err.Error(), FormatSqlWithArgs(s, args))
- }
- return err
- }
- // FormatSqlWithArgs binds the arguments to the sql string and returns a complete
- // sql string, just for debugging.
- func FormatSqlWithArgs(sql string, args []interface{}) string {
- index := -1
- newQuery, _ := gregex.ReplaceStringFunc(
- `(\?|:v\d+|\$\d+|@p\d+)`,
- sql,
- func(s string) string {
- index++
- if len(args) > index {
- if args[index] == nil {
- return "null"
- }
- var (
- rv = reflect.ValueOf(args[index])
- kind = rv.Kind()
- )
- if kind == reflect.Ptr {
- if rv.IsNil() || !rv.IsValid() {
- return "null"
- }
- rv = rv.Elem()
- kind = rv.Kind()
- }
- switch kind {
- case reflect.String, reflect.Map, reflect.Slice, reflect.Array:
- return `'` + gstr.QuoteMeta(gconv.String(args[index]), `'`) + `'`
- case reflect.Struct:
- if t, ok := args[index].(time.Time); ok {
- return `'` + t.Format(`2006-01-02 15:04:05`) + `'`
- }
- return `'` + gstr.QuoteMeta(gconv.String(args[index]), `'`) + `'`
- }
- return gconv.String(args[index])
- }
- return s
- })
- return newQuery
- }
|