Merge pull request 'Add support for pointer arguments' (#1) from add-support-for-ptr-args into master
All checks were successful
Build / build (push) Successful in 36s
All checks were successful
Build / build (push) Successful in 36s
Reviewed-on: #1
This commit is contained in:
commit
029cf6ce01
10
query.go
10
query.go
@ -44,6 +44,14 @@ func (q *Query) bindInto(into *int, args ...any) *Query {
|
||||
}
|
||||
for i, arg := range args {
|
||||
*into++
|
||||
if arg == nil {
|
||||
q.stmt.BindNull(*into)
|
||||
continue
|
||||
}
|
||||
v := reflect.ValueOf(arg)
|
||||
if v.Kind() == reflect.Ptr {
|
||||
arg = v.Elem().Interface()
|
||||
}
|
||||
if asString, ok := arg.(string); ok {
|
||||
q.stmt.BindText(*into, asString)
|
||||
} else if asInt, ok := arg.(int); ok {
|
||||
@ -54,7 +62,7 @@ func (q *Query) bindInto(into *int, args ...any) *Query {
|
||||
q.stmt.BindBool(*into, asBool)
|
||||
} else {
|
||||
// Check if the argument is a slice or array of any type
|
||||
v := reflect.ValueOf(arg)
|
||||
v = reflect.ValueOf(arg)
|
||||
if v.Kind() == reflect.Slice || v.Kind() == reflect.Array {
|
||||
*into--
|
||||
for i := 0; i < v.Len(); i++ {
|
||||
|
@ -94,12 +94,50 @@ func TestUpdateQuery(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestUpdateQueryWithWrongArguments(t *testing.T) {
|
||||
type S struct {
|
||||
Field string
|
||||
}
|
||||
db := openTestDb(t)
|
||||
value := "ipsum"
|
||||
err := db.Query("insert into mytable(key, value) values ('lorem', ?)").Bind(&value).Exec()
|
||||
abc := S{
|
||||
Field: "ipsum",
|
||||
}
|
||||
err := db.Query("insert into mytable(key, value) values ('lorem', ?)").Bind(abc).Exec()
|
||||
require.Error(t, err)
|
||||
}
|
||||
|
||||
func TestUpdateQueryWithPointerValue(t *testing.T) {
|
||||
db := openTestDb(t)
|
||||
func() {
|
||||
tx := db.MustBegin()
|
||||
defer tx.MustRollback()
|
||||
tx.Query("insert into mytable(key, value) values ('lorem', 'bar')").MustExec()
|
||||
value := "ipsum"
|
||||
key := "lorem"
|
||||
tx.Query("update mytable set value = ? where key = ?").Bind(&value, key).MustExec()
|
||||
tx.MustCommit()
|
||||
}()
|
||||
|
||||
var value string
|
||||
db.Query("select value from mytable where key = 'lorem'").MustScanSingle(&value)
|
||||
require.Equal(t, "ipsum", value)
|
||||
}
|
||||
|
||||
func TestUpdateQueryWithNullValue(t *testing.T) {
|
||||
db := openTestDb(t)
|
||||
func() {
|
||||
tx := db.MustBegin()
|
||||
defer tx.MustRollback()
|
||||
tx.Query("insert into mytable(key, value) values ('lorem', 'bar')").MustExec()
|
||||
key := "lorem"
|
||||
tx.Query("update mytable set value = ? where key = ?").Bind(nil, key).MustExec()
|
||||
tx.MustCommit()
|
||||
}()
|
||||
|
||||
var value *string
|
||||
db.Query("select value from mytable where key = 'lorem'").MustScanSingle(&value)
|
||||
require.Nil(t, value)
|
||||
}
|
||||
|
||||
func TestQueryWithPointerStringArguments(t *testing.T) {
|
||||
db := openTestDb(t)
|
||||
var result *string
|
||||
|
Loading…
x
Reference in New Issue
Block a user