123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754 |
- // 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 (
- "bytes"
- "fmt"
- "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/gutil"
- "reflect"
- "regexp"
- "strings"
- "time"
- "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{}
- }
- const (
- ORM_TAG_FOR_STRUCT = "orm"
- ORM_TAG_FOR_UNIQUE = "unique"
- ORM_TAG_FOR_PRIMARY = "primary"
- )
- 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{ORM_TAG_FOR_STRUCT}, gconv.StructTagPriority...)
- )
- // 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 <obj> should be type of *map/map/*struct/struct.
- // It supports inherit 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 v.(type) {
- case time.Time, *time.Time, gtime.Time, *gtime.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.
- // The parameter <value> should be type of *map/map/*struct/struct.
- // It supports inherit struct definition for struct.
- func DataToMapDeep(value interface{}) map[string]interface{} {
- if v, ok := value.(apiMapStrAny); ok {
- return v.MapStrAny()
- }
- var (
- rvValue reflect.Value
- rvField reflect.Value
- rvKind reflect.Kind
- rtField reflect.StructField
- )
- if v, ok := value.(reflect.Value); ok {
- rvValue = v
- } else {
- rvValue = reflect.ValueOf(value)
- }
- rvKind = rvValue.Kind()
- if rvKind == reflect.Ptr {
- rvValue = rvValue.Elem()
- rvKind = rvValue.Kind()
- }
- // If given <value> is not a struct, it uses gconv.Map for converting.
- if rvKind != reflect.Struct {
- return gconv.Map(value, structTagPriority...)
- }
- // Struct handling.
- var (
- fieldTag reflect.StructTag
- rvType = rvValue.Type()
- name = ""
- data = make(map[string]interface{})
- )
- for i := 0; i < rvValue.NumField(); i++ {
- rtField = rvType.Field(i)
- rvField = rvValue.Field(i)
- fieldName := rtField.Name
- if !utils.IsLetterUpper(fieldName[0]) {
- continue
- }
- // Struct attribute inherit
- if rtField.Anonymous {
- for k, v := range DataToMapDeep(rvField) {
- data[k] = v
- }
- continue
- }
- // Other attributes.
- name = ""
- fieldTag = rtField.Tag
- for _, tag := range structTagPriority {
- if s := fieldTag.Get(tag); s != "" {
- name = s
- break
- }
- }
- if name == "" {
- name = fieldName
- } else {
- // The "orm" tag supports json tag feature: -, omitempty
- // The "orm" tag would be like: "id,priority", so it should use splitting handling.
- name = gstr.Trim(name)
- if name == "-" {
- continue
- }
- array := gstr.SplitAndTrim(name, ",")
- if len(array) > 1 {
- switch array[1] {
- case "omitempty":
- if empty.IsEmpty(rvField.Interface()) {
- continue
- } else {
- name = array[0]
- }
- default:
- name = array[0]
- }
- }
- }
- // The underlying driver supports time.Time/*time.Time types.
- fieldValue := rvField.Interface()
- switch fieldValue.(type) {
- case time.Time, *time.Time, gtime.Time, *gtime.Time:
- data[name] = fieldValue
- default:
- // Use string conversion in default.
- if s, ok := fieldValue.(apiString); ok {
- data[name] = s.String()
- } else {
- data[name] = fieldValue
- }
- }
- }
- return data
- }
- // 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. It handles strings like:
- // "user",
- // "user u",
- // "user,user_detail",
- // "user u, user_detail ut",
- // "user.user u, user.user_detail ut",
- // "u.id, u.name, u.age",
- // "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{ORM_TAG_FOR_STRUCT})
- 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{ORM_TAG_FOR_UNIQUE, ORM_TAG_FOR_PRIMARY}, array[1]) {
- return array[0], []interface{}{field.Value()}, nil
- }
- if len(where) > 0 {
- where += " AND "
- }
- where += field.TagValue + "=?"
- args = append(args, field.Value())
- }
- return
- }
- // GetPrimaryKey retrieves and returns primary key field name from given struct.
- func GetPrimaryKey(pointer interface{}) (string, error) {
- tagField, err := structs.TagFields(pointer, []string{ORM_TAG_FOR_STRUCT})
- if err != nil {
- return "", err
- }
- array := ([]string)(nil)
- for _, field := range tagField {
- array = strings.Split(field.TagValue, ",")
- if len(array) > 1 && array[1] == ORM_TAG_FOR_PRIMARY {
- 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)
- }
- // formatWhere formats where statement and its arguments.
- func formatWhere(db DB, where interface{}, args []interface{}, omitEmpty bool) (newWhere string, newArgs []interface{}) {
- var (
- buffer = bytes.NewBuffer(nil)
- rv = reflect.ValueOf(where)
- kind = rv.Kind()
- )
- if kind == reflect.Ptr {
- rv = rv.Elem()
- kind = rv.Kind()
- }
- switch kind {
- case reflect.Array, reflect.Slice:
- newArgs = formatWhereInterfaces(db, gconv.Interfaces(where), buffer, newArgs)
- case reflect.Map:
- for key, value := range DataToMapDeep(where) {
- if gregex.IsMatchString(regularFieldNameRegPattern, key) && 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 iterates 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 := where.(apiIterator); ok {
- iterator.Iterator(func(key, value interface{}) bool {
- ketStr := gconv.String(key)
- if gregex.IsMatchString(regularFieldNameRegPattern, ketStr) && omitEmpty && empty.IsEmpty(value) {
- return true
- }
- newArgs = formatWhereKeyValue(db, buffer, newArgs, ketStr, value)
- return true
- })
- break
- }
- for key, value := range DataToMapDeep(where) {
- if omitEmpty && empty.IsEmpty(value) {
- continue
- }
- newArgs = formatWhereKeyValue(db, buffer, newArgs, key, value)
- }
- default:
- buffer.WriteString(gconv.String(where))
- }
- if buffer.Len() == 0 {
- return "", args
- }
- newArgs = append(newArgs, 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.QuoteString(newWhere)
- if len(newArgs) > 0 {
- if utils.IsArray(newArgs[0]) {
- // Eg: Where("id", []int{1,2,3})
- newWhere += " IN (?)"
- } else if empty.IsNil(newArgs[0]) {
- // Eg: Where("id", nil)
- newWhere += " IS NULL"
- newArgs = nil
- } else {
- // Eg: Where/And/Or("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.QuoteWord(str) + "=?")
- } else {
- buffer.WriteString(db.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.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, sql string, args ...interface{}) error {
- if err != nil && err != ErrNoRows {
- return gerror.New(fmt.Sprintf("%s, %s\n", err.Error(), FormatSqlWithArgs(sql, 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
- }
- // convertMapToStruct maps the <data> to given struct.
- // Note that the given parameter <pointer> should be a pointer to s struct.
- func convertMapToStruct(data map[string]interface{}, pointer interface{}) error {
- tagNameMap, err := structs.TagMapName(pointer, []string{ORM_TAG_FOR_STRUCT})
- if err != nil {
- return err
- }
- // It retrieves and returns the mapping between orm tag and the struct attribute name.
- mapping := make(map[string]string)
- for tag, attr := range tagNameMap {
- mapping[strings.Split(tag, ",")[0]] = attr
- }
- return gconv.Struct(data, pointer, mapping)
- }
|