Add support for pointer arguments
Some checks failed
Build / build (push) Failing after 41s

This commit is contained in:
Sebastiaan de Schaetzen 2025-05-08 19:42:57 +02:00
parent dd6be6b9b6
commit 637797116d
2 changed files with 42 additions and 1 deletions

View File

@ -44,6 +44,14 @@ func (q *Query) bindInto(into *int, args ...any) *Query {
}
for i, arg := range args {
*into++
v := reflect.ValueOf(arg)
if v.Kind() == reflect.Ptr {
if v.IsNil() {
q.stmt.BindNull(*into)
continue
}
arg = v.Elem().Interface() // Dereference the pointer
}
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++ {

View File

@ -100,6 +100,39 @@ func TestUpdateQueryWithWrongArguments(t *testing.T) {
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