Skip to content

Commit

Permalink
Add support for NamedValueChecker interface
Browse files Browse the repository at this point in the history
  • Loading branch information
tamird committed Apr 28, 2023
1 parent 2a217b9 commit b25c22b
Show file tree
Hide file tree
Showing 3 changed files with 108 additions and 3 deletions.
7 changes: 4 additions & 3 deletions array.go
Expand Up @@ -19,10 +19,11 @@ var typeSQLScanner = reflect.TypeOf((*sql.Scanner)(nil)).Elem()
// slice of any dimension.
//
// For example:
// db.Query(`SELECT * FROM t WHERE id = ANY($1)`, pq.Array([]int{235, 401}))
//
// var x []sql.NullInt64
// db.QueryRow(`SELECT ARRAY[235, 401]`).Scan(pq.Array(&x))
// db.Query(`SELECT * FROM t WHERE id = ANY($1)`, pq.Array([]int{235, 401}))
//
// var x []sql.NullInt64
// db.QueryRow(`SELECT ARRAY[235, 401]`).Scan(pq.Array(&x))
//
// Scanning multi-dimensional arrays is not supported. Arrays where the lower
// bound is not one (such as `[0:0]={1}') are not supported.
Expand Down
35 changes: 35 additions & 0 deletions conn_go19.go
@@ -0,0 +1,35 @@
//go:build go1.9
// +build go1.9

package pq

import (
"database/sql/driver"
"reflect"
)

var _ driver.NamedValueChecker = (*conn)(nil)

func (c *conn) CheckNamedValue(nv *driver.NamedValue) error {
if _, ok := nv.Value.(driver.Valuer); ok {
// Ignore Valuer, for backward compatiblity with pq.Array()
return driver.ErrSkip
}

// Ignoring []byte / []uint8
if _, ok := nv.Value.([]uint8); ok {
return driver.ErrSkip
}

v := reflect.ValueOf(nv.Value)
if v.Kind() == reflect.Ptr {
v = v.Elem()
}
if v.Kind() == reflect.Slice {
var err error
nv.Value, err = Array(nv.Value).Value()
return err
}

return driver.ErrSkip
}
69 changes: 69 additions & 0 deletions conn_go19_test.go
@@ -0,0 +1,69 @@
//go:build go1.9
// +build go1.9

package pq

import (
"fmt"
"reflect"
"testing"
)

func TestArrayArg(t *testing.T) {
db := openTestConn(t)
defer db.Close()

for _, tc := range []struct {
in, out interface{}
}{
{
in: []int{245, 231},
out: []int64{245, 231},
},
{
in: &[]int{245, 231},
out: []int64{245, 231},
},
{
in: []int64{245, 231},
},
{
in: &[]int64{245, 231},
out: []int64{245, 231},
},
} {
if tc.out == nil {
tc.out = tc.in
}
t.Run(fmt.Sprintf("%#v", tc.in), func(t *testing.T) {
r, err := db.Query("SELECT $1::int[]", tc.in)
if err != nil {
t.Fatal(err)
}
defer r.Close()

if !r.Next() {
if r.Err() != nil {
t.Fatal(r.Err())
}
t.Fatal("expected row")
}

defer func() {
if r.Next() {
t.Fatal("unexpected row")
}
}()

got := reflect.New(reflect.TypeOf(tc.out))
if err := r.Scan(got.Interface()); err != nil {
t.Fatal(err)
}

if !reflect.DeepEqual(tc.out, got.Elem().Interface()) {
t.Errorf("got %v, want %v", got, tc.out)
}
})
}

}

0 comments on commit b25c22b

Please sign in to comment.