Initial commit

vendor/gorm.io/gorm/logger/sql.go

@@ -0,0 +1,133 @@
+package logger
+
+import (
+	"database/sql/driver"
+	"fmt"
+	"reflect"
+	"regexp"
+	"strconv"
+	"strings"
+	"time"
+	"unicode"
+
+	"gorm.io/gorm/utils"
+)
+
+const (
+	tmFmtWithMS = "2006-01-02 15:04:05.999"
+	tmFmtZero   = "0000-00-00 00:00:00"
+	nullStr     = "NULL"
+)
+
+func isPrintable(s []byte) bool {
+	for _, r := range s {
+		if !unicode.IsPrint(rune(r)) {
+			return false
+		}
+	}
+	return true
+}
+
+var convertibleTypes = []reflect.Type{reflect.TypeOf(time.Time{}), reflect.TypeOf(false), reflect.TypeOf([]byte{})}
+
+func ExplainSQL(sql string, numericPlaceholder *regexp.Regexp, escaper string, avars ...interface{}) string {
+	var convertParams func(interface{}, int)
+	var vars = make([]string, len(avars))
+
+	convertParams = func(v interface{}, idx int) {
+		switch v := v.(type) {
+		case bool:
+			vars[idx] = strconv.FormatBool(v)
+		case time.Time:
+			if v.IsZero() {
+				vars[idx] = escaper + tmFmtZero + escaper
+			} else {
+				vars[idx] = escaper + v.Format(tmFmtWithMS) + escaper
+			}
+		case *time.Time:
+			if v != nil {
+				if v.IsZero() {
+					vars[idx] = escaper + tmFmtZero + escaper
+				} else {
+					vars[idx] = escaper + v.Format(tmFmtWithMS) + escaper
+				}
+			} else {
+				vars[idx] = nullStr
+			}
+		case driver.Valuer:
+			reflectValue := reflect.ValueOf(v)
+			if v != nil && reflectValue.IsValid() && ((reflectValue.Kind() == reflect.Ptr && !reflectValue.IsNil()) || reflectValue.Kind() != reflect.Ptr) {
+				r, _ := v.Value()
+				convertParams(r, idx)
+			} else {
+				vars[idx] = nullStr
+			}
+		case fmt.Stringer:
+			reflectValue := reflect.ValueOf(v)
+			if v != nil && reflectValue.IsValid() && ((reflectValue.Kind() == reflect.Ptr && !reflectValue.IsNil()) || reflectValue.Kind() != reflect.Ptr) {
+				vars[idx] = escaper + strings.Replace(fmt.Sprintf("%v", v), escaper, "\\"+escaper, -1) + escaper
+			} else {
+				vars[idx] = nullStr
+			}
+		case []byte:
+			if isPrintable(v) {
+				vars[idx] = escaper + strings.Replace(string(v), escaper, "\\"+escaper, -1) + escaper
+			} else {
+				vars[idx] = escaper + "<binary>" + escaper
+			}
+		case int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64:
+			vars[idx] = utils.ToString(v)
+		case float64, float32:
+			vars[idx] = fmt.Sprintf("%.6f", v)
+		case string:
+			vars[idx] = escaper + strings.Replace(v, escaper, "\\"+escaper, -1) + escaper
+		default:
+			rv := reflect.ValueOf(v)
+			if v == nil || !rv.IsValid() || rv.Kind() == reflect.Ptr && rv.IsNil() {
+				vars[idx] = nullStr
+			} else if valuer, ok := v.(driver.Valuer); ok {
+				v, _ = valuer.Value()
+				convertParams(v, idx)
+			} else if rv.Kind() == reflect.Ptr && !rv.IsZero() {
+				convertParams(reflect.Indirect(rv).Interface(), idx)
+			} else {
+				for _, t := range convertibleTypes {
+					if rv.Type().ConvertibleTo(t) {
+						convertParams(rv.Convert(t).Interface(), idx)
+						return
+					}
+				}
+				vars[idx] = escaper + strings.Replace(fmt.Sprint(v), escaper, "\\"+escaper, -1) + escaper
+			}
+		}
+	}
+
+	for idx, v := range avars {
+		convertParams(v, idx)
+	}
+
+	if numericPlaceholder == nil {
+		var idx int
+		var newSQL strings.Builder
+
+		for _, v := range []byte(sql) {
+			if v == '?' {
+				if len(vars) > idx {
+					newSQL.WriteString(vars[idx])
+					idx++
+					continue
+				}
+			}
+			newSQL.WriteByte(v)
+		}
+
+		sql = newSQL.String()
+	} else {
+		sql = numericPlaceholder.ReplaceAllString(sql, "$$$1$$")
+		for idx, v := range vars {
+			sql = strings.Replace(sql, "$"+strconv.Itoa(idx+1)+"$", v, 1)
+		}
+	}
+
+	return sql
+}