This commit is contained in:
Lunny Xiao 2023-09-16 22:33:07 +08:00
parent f234597448
commit 52ab5c818f
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
1 changed files with 32 additions and 25 deletions

View File

@ -637,7 +637,10 @@ func (statement *Statement) ConvertSQLOrArgs(sqlOrArgs ...interface{}) (string,
func (statement *Statement) convertSQLOrArgs(sqlOrArgs ...interface{}) (string, []interface{}, error) {
switch sqlOrArgs[0].(type) {
case string:
if len(sqlOrArgs) > 1 {
if len(sqlOrArgs) <= 1 {
return sqlOrArgs[0].(string), sqlOrArgs[1:], nil
}
newArgs := make([]interface{}, 0, len(sqlOrArgs)-1)
for _, arg := range sqlOrArgs[1:] {
if v, ok := arg.(time.Time); ok {
@ -661,13 +664,17 @@ func (statement *Statement) convertSQLOrArgs(sqlOrArgs ...interface{}) (string,
} else {
newArgs = append(newArgs, nil)
}
} else if v, ok := arg.(driver.Valuer); ok {
vv, err := v.Value()
if err != nil {
return "", nil, err
}
newArgs = append(newArgs, vv)
} else {
newArgs = append(newArgs, arg)
}
}
return sqlOrArgs[0].(string), newArgs, nil
}
return sqlOrArgs[0].(string), sqlOrArgs[1:], nil
case *builder.Builder:
return sqlOrArgs[0].(*builder.Builder).ToSQL()
case builder.Builder: