func (Mysql) EscapeBool(w query.Writer, b bool) {
if b {
w.WriteRune('1')
} else {
w.WriteRune('0')
}
}
// Need to turn \x00, \n, \r, \, ', " and \x1a.
// Returns an escaped, quoted string. eg, "hello 'world'" -> "'hello \'world\''".
func (Mysql) EscapeString(w query.Writer, s string) {
w.WriteRune('\'')
for _, char := range s {
switch char {
case '\'':
w.WriteString(`\'`)
case '"':
w.WriteString(`\"`)
case '\\':
w.WriteString(`\\`)
case '\n':
w.WriteString(`\n`)
case '\r':
w.WriteString(`\r`)
case 0:
w.WriteString(`\x00`)
case 0x1a:
w.WriteString(`\x1a`)
default:
w.WriteRune(char)
}
}
w.WriteRune('\'')
}
func (Mysql) EscapeIdent(w query.Writer, ident string) {
w.WriteRune('`')
r := strings.NewReplacer("`", "``", ".", "`.`")
w.WriteString(r.Replace(ident))
w.WriteRune('`')
}
func interpolate(w query.Writer, v interface{}) error {
valuer, ok := v.(driver.Valuer)
if ok {
val, err := valuer.Value()
if err != nil {
return err
}
v = val
}
valueOfV := reflect.ValueOf(v)
kindOfV := valueOfV.Kind()
switch {
case v == nil:
w.WriteString("NULL")
case isInt(kindOfV):
var ival = valueOfV.Int()
w.WriteString(strconv.FormatInt(ival, 10))
case isUint(kindOfV):
var uival = valueOfV.Uint()
w.WriteString(strconv.FormatUint(uival, 10))
case kindOfV == reflect.String:
var str = valueOfV.String()
if !utf8.ValidString(str) {
return ErrNotUTF8
}
D.EscapeString(w, str)
case isFloat(kindOfV):
var fval = valueOfV.Float()
w.WriteString(strconv.FormatFloat(fval, 'f', -1, 64))
case kindOfV == reflect.Bool:
D.EscapeBool(w, valueOfV.Bool())
case kindOfV == reflect.Struct:
if typeOfV := valueOfV.Type(); typeOfV == typeOfTime {
t := valueOfV.Interface().(time.Time)
D.EscapeTime(w, t)
} else {
return ErrInvalidValue
}
case kindOfV == reflect.Slice:
typeOfV := reflect.TypeOf(v)
subtype := typeOfV.Elem()
kindOfSubtype := subtype.Kind()
sliceLen := valueOfV.Len()
stringSlice := make([]string, 0, sliceLen)
switch {
case sliceLen == 0:
return ErrInvalidSliceLength
case isInt(kindOfSubtype):
for i := 0; i < sliceLen; i++ {
var ival = valueOfV.Index(i).Int()
stringSlice = append(stringSlice, strconv.FormatInt(ival, 10))
}
case isUint(kindOfSubtype):
for i := 0; i < sliceLen; i++ {
var uival = valueOfV.Index(i).Uint()
stringSlice = append(stringSlice, strconv.FormatUint(uival, 10))
}
case kindOfSubtype == reflect.String:
for i := 0; i < sliceLen; i++ {
var str = valueOfV.Index(i).String()
if !utf8.ValidString(str) {
return ErrNotUTF8
}
buf := new(bytes.Buffer)
D.EscapeString(buf, str)
stringSlice = append(stringSlice, buf.String())
}
default:
return ErrInvalidSliceValue
}
w.WriteRune('(')
w.WriteString(strings.Join(stringSlice, ","))
w.WriteRune(')')
default:
return ErrInvalidValue
}
return nil
}
