You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
1.8 KiB
Go
102 lines
1.8 KiB
Go
package sqlx
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/tal-tech/go-zero/core/logx"
|
|
"github.com/tal-tech/go-zero/core/mapping"
|
|
)
|
|
|
|
func desensitize(datasource string) string {
|
|
// remove account
|
|
pos := strings.LastIndex(datasource, "@")
|
|
if 0 <= pos && pos+1 < len(datasource) {
|
|
datasource = datasource[pos+1:]
|
|
}
|
|
|
|
return datasource
|
|
}
|
|
|
|
func escape(input string) string {
|
|
var b strings.Builder
|
|
|
|
for _, ch := range input {
|
|
switch ch {
|
|
case '\x00':
|
|
b.WriteString(`\x00`)
|
|
case '\r':
|
|
b.WriteString(`\r`)
|
|
case '\n':
|
|
b.WriteString(`\n`)
|
|
case '\\':
|
|
b.WriteString(`\\`)
|
|
case '\'':
|
|
b.WriteString(`\'`)
|
|
case '"':
|
|
b.WriteString(`\"`)
|
|
case '\x1a':
|
|
b.WriteString(`\x1a`)
|
|
default:
|
|
b.WriteRune(ch)
|
|
}
|
|
}
|
|
|
|
return b.String()
|
|
}
|
|
|
|
func format(query string, args ...interface{}) (string, error) {
|
|
numArgs := len(args)
|
|
if numArgs == 0 {
|
|
return query, nil
|
|
}
|
|
|
|
var b strings.Builder
|
|
argIndex := 0
|
|
|
|
for _, ch := range query {
|
|
if ch == '?' {
|
|
if argIndex >= numArgs {
|
|
return "", fmt.Errorf("error: %d ? in sql, but less arguments provided", argIndex)
|
|
}
|
|
|
|
arg := args[argIndex]
|
|
argIndex++
|
|
|
|
switch v := arg.(type) {
|
|
case bool:
|
|
if v {
|
|
b.WriteByte('1')
|
|
} else {
|
|
b.WriteByte('0')
|
|
}
|
|
case string:
|
|
b.WriteByte('\'')
|
|
b.WriteString(escape(v))
|
|
b.WriteByte('\'')
|
|
default:
|
|
b.WriteString(mapping.Repr(v))
|
|
}
|
|
} else {
|
|
b.WriteRune(ch)
|
|
}
|
|
}
|
|
|
|
if argIndex < numArgs {
|
|
return "", fmt.Errorf("error: %d ? in sql, but more arguments provided", argIndex)
|
|
}
|
|
|
|
return b.String(), nil
|
|
}
|
|
|
|
func logInstanceError(datasource string, err error) {
|
|
datasource = desensitize(datasource)
|
|
logx.Errorf("Error on getting sql instance of %s: %v", datasource, err)
|
|
}
|
|
|
|
func logSqlError(stmt string, err error) {
|
|
if err != nil && err != ErrNotFound {
|
|
logx.Errorf("stmt: %s, error: %s", stmt, err.Error())
|
|
}
|
|
}
|