@ -192,7 +192,8 @@ type Map struct {
reflectTypeToType map [ reflect . Type ] * Type
reflectTypeToType map [ reflect . Type ] * Type
memoizedScanPlans map [ uint32 ] map [ reflect . Type ] [ 2 ] ScanPlan
memoizedScanPlans map [ uint32 ] map [ reflect . Type ] [ 2 ] ScanPlan
memoizedEncodePlans map [ uint32 ] map [ reflect . Type ] [ 2 ] EncodePlan
// TryWrapEncodePlanFuncs is a slice of functions that will wrap a value that cannot be encoded by the Codec. Every
// TryWrapEncodePlanFuncs is a slice of functions that will wrap a value that cannot be encoded by the Codec. Every
// time a wrapper is found the PlanEncode method will be recursively called with the new value. This allows several layers of wrappers
// time a wrapper is found the PlanEncode method will be recursively called with the new value. This allows several layers of wrappers
@ -214,7 +215,8 @@ func NewMap() *Map {
reflectTypeToName : make ( map [ reflect . Type ] string ) ,
reflectTypeToName : make ( map [ reflect . Type ] string ) ,
oidToFormatCode : make ( map [ uint32 ] int16 ) ,
oidToFormatCode : make ( map [ uint32 ] int16 ) ,
memoizedScanPlans : make ( map [ uint32 ] map [ reflect . Type ] [ 2 ] ScanPlan ) ,
memoizedScanPlans : make ( map [ uint32 ] map [ reflect . Type ] [ 2 ] ScanPlan ) ,
memoizedEncodePlans : make ( map [ uint32 ] map [ reflect . Type ] [ 2 ] EncodePlan ) ,
TryWrapEncodePlanFuncs : [ ] TryWrapEncodePlanFunc {
TryWrapEncodePlanFuncs : [ ] TryWrapEncodePlanFunc {
TryWrapDerefPointerEncodePlan ,
TryWrapDerefPointerEncodePlan ,
@ -223,6 +225,7 @@ func NewMap() *Map {
TryWrapStructEncodePlan ,
TryWrapStructEncodePlan ,
TryWrapSliceEncodePlan ,
TryWrapSliceEncodePlan ,
TryWrapMultiDimSliceEncodePlan ,
TryWrapMultiDimSliceEncodePlan ,
TryWrapArrayEncodePlan ,
} ,
} ,
TryWrapScanPlanFuncs : [ ] TryWrapScanPlanFunc {
TryWrapScanPlanFuncs : [ ] TryWrapScanPlanFunc {
@ -232,6 +235,7 @@ func NewMap() *Map {
TryWrapStructScanPlan ,
TryWrapStructScanPlan ,
TryWrapPtrSliceScanPlan ,
TryWrapPtrSliceScanPlan ,
TryWrapPtrMultiDimSliceScanPlan ,
TryWrapPtrMultiDimSliceScanPlan ,
TryWrapPtrArrayScanPlan ,
} ,
} ,
}
}
@ -420,6 +424,9 @@ func (m *Map) RegisterType(t *Type) {
for k := range m . memoizedScanPlans {
for k := range m . memoizedScanPlans {
delete ( m . memoizedScanPlans , k )
delete ( m . memoizedScanPlans , k )
}
}
for k := range m . memoizedEncodePlans {
delete ( m . memoizedEncodePlans , k )
}
} }
// RegisterDefaultPgType registers a mapping of a Go type to a PostgreSQL type name. Typically the data type to be
// RegisterDefaultPgType registers a mapping of a Go type to a PostgreSQL type name. Typically the data type to be
@ -433,6 +440,9 @@ func (m *Map) RegisterDefaultPgType(value any, name string) {
for k := range m . memoizedScanPlans {
for k := range m . memoizedScanPlans {
delete ( m . memoizedScanPlans , k )
delete ( m . memoizedScanPlans , k )
}
}
for k := range m . memoizedEncodePlans {
delete ( m . memoizedEncodePlans , k )
}
} }
func ( m * Map ) TypeForOID ( oid uint32 ) ( * Type , bool ) { func ( m * Map ) TypeForOID ( oid uint32 ) ( * Type , bool ) {
@ -1018,7 +1028,7 @@ func TryWrapStructScanPlan(target any) (plan WrappedScanPlanNextSetter, nextValu
var targetElemValue reflect . Value
var targetElemValue reflect . Value
if targetValue . IsNil ( ) {
if targetValue . IsNil ( ) {
targetElemValue = reflect . New ( targetValue . Type ( ) . Elem ( ) )
targetElemValue = reflect . Zero ( targetValue . Type ( ) . Elem ( ) )
} else {
} else {
targetElemValue = targetValue . Elem ( )
targetElemValue = targetValue . Elem ( )
}
}
@ -1139,6 +1149,31 @@ func (plan *wrapPtrMultiDimSliceScanPlan) Scan(src []byte, target any) error {
return plan . next . Scan ( src , & anyMultiDimSliceArray { slice : reflect . ValueOf ( target ) . Elem ( ) } )
return plan . next . Scan ( src , & anyMultiDimSliceArray { slice : reflect . ValueOf ( target ) . Elem ( ) } )
} }
// TryWrapPtrArrayScanPlan tries to wrap a pointer to a single dimension array.
func TryWrapPtrArrayScanPlan ( target any ) ( plan WrappedScanPlanNextSetter , nextValue any , ok bool ) {
targetValue := reflect . ValueOf ( target )
if targetValue . Kind ( ) != reflect . Ptr {
return nil , nil , false
}
targetElemValue := targetValue . Elem ( )
if targetElemValue . Kind ( ) == reflect . Array {
return & wrapPtrArrayReflectScanPlan { } , & anyArrayArrayReflect { array : targetElemValue } , true
}
return nil , nil , false
}
type wrapPtrArrayReflectScanPlan struct {
next ScanPlan
}
func ( plan * wrapPtrArrayReflectScanPlan ) SetNext ( next ScanPlan ) { plan . next = next }
func ( plan * wrapPtrArrayReflectScanPlan ) Scan ( src [ ] byte , target any ) error {
return plan . next . Scan ( src , & anyArrayArrayReflect { array : reflect . ValueOf ( target ) . Elem ( ) } )
}
// PlanScan prepares a plan to scan a value into target.
// PlanScan prepares a plan to scan a value into target.
func ( m * Map ) PlanScan ( oid uint32 , formatCode int16 , target any ) ScanPlan { func ( m * Map ) PlanScan ( oid uint32 , formatCode int16 , target any ) ScanPlan {
oidMemo := m . memoizedScanPlans [ oid ]
oidMemo := m . memoizedScanPlans [ oid ]
@ -1200,6 +1235,16 @@ func (m *Map) planScan(oid uint32, formatCode int16, target any) ScanPlan {
}
}
}
}
// This needs to happen before trying m.TryWrapScanPlanFuncs. Otherwise, a sql.Scanner would not get called if it was
// defined on a type that could be unwrapped such as `type myString string`.
//
// https://github.com/jackc/pgtype/issues/197
if dt == nil {
if _ , ok := target . ( sql . Scanner ) ; ok {
return & scanPlanSQLScanner { formatCode : formatCode }
}
}
for _ , f := range m . TryWrapScanPlanFuncs {
for _ , f := range m . TryWrapScanPlanFuncs {
if wrapperPlan , nextDst , ok := f ( target ) ; ok {
if wrapperPlan , nextDst , ok := f ( target ) ; ok {
if nextPlan := m . planScan ( oid , formatCode , nextDst ) ; nextPlan != nil {
if nextPlan := m . planScan ( oid , formatCode , nextDst ) ; nextPlan != nil {
@ -1221,10 +1266,6 @@ func (m *Map) planScan(oid uint32, formatCode int16, target any) ScanPlan {
}
}
}
}
if _ , ok := target . ( sql . Scanner ) ; ok {
return & scanPlanSQLScanner { formatCode : formatCode }
}
return & scanPlanFail { m : m , oid : oid , formatCode : formatCode }
return & scanPlanFail { m : m , oid : oid , formatCode : formatCode }
} }
@ -1289,6 +1330,24 @@ func codecDecodeToTextFormat(codec Codec, m *Map, oid uint32, format int16, src
// PlanEncode returns an Encode plan for encoding value into PostgreSQL format for oid and format. If no plan can be
// PlanEncode returns an Encode plan for encoding value into PostgreSQL format for oid and format. If no plan can be
// found then nil is returned.
// found then nil is returned.
func ( m * Map ) PlanEncode ( oid uint32 , format int16 , value any ) EncodePlan { func ( m * Map ) PlanEncode ( oid uint32 , format int16 , value any ) EncodePlan {
oidMemo := m . memoizedEncodePlans [ oid ]
if oidMemo == nil {
oidMemo = make ( map [ reflect . Type ] [ 2 ] EncodePlan )
m . memoizedEncodePlans [ oid ] = oidMemo
}
targetReflectType := reflect . TypeOf ( value )
typeMemo := oidMemo [ targetReflectType ]
plan := typeMemo [ format ]
if plan == nil {
plan = m . planEncode ( oid , format , value )
typeMemo [ format ] = plan
oidMemo [ targetReflectType ] = typeMemo
}
return plan
}
func ( m * Map ) planEncode ( oid uint32 , format int16 , value any ) EncodePlan {
if format == TextFormatCode {
if format == TextFormatCode {
switch value . ( type ) {
switch value . ( type ) {
case string :
case string :
@ -1299,16 +1358,16 @@ func (m *Map) PlanEncode(oid uint32, format int16, value any) EncodePlan {
}
}
var dt * Type
var dt * Type
if dataType , ok := m . TypeForOID ( oid ) ; ok {
if oid == 0 {
dt = dataType
} else {
// If no type for the OID was found, then either it is unknowable (e.g. the simple protocol) or it is an
// unregistered type. In either case try to find the type and OID that matches the value (e.g. a []byte would be
// registered to PostgreSQL bytea).
if dataType , ok := m . TypeForValue ( value ) ; ok {
if dataType , ok := m . TypeForValue ( value ) ; ok {
dt = dataType
dt = dataType
oid = dt . OID // Preserve assumed OID in case we are recursively called below.
oid = dt . OID // Preserve assumed OID in case we are recursively called below.
}
}
} else {
if dataType , ok := m . TypeForOID ( oid ) ; ok {
dt = dataType
}
}
}
if dt != nil {
if dt != nil {
@ -1941,6 +2000,35 @@ func (plan *wrapMultiDimSliceEncodePlan) Encode(value any, buf []byte) (newBuf [
return plan . next . Encode ( & w , buf )
return plan . next . Encode ( & w , buf )
} }
func TryWrapArrayEncodePlan ( value any ) ( plan WrappedEncodePlanNextSetter , nextValue any , ok bool ) {
if _ , ok := value . ( driver . Valuer ) ; ok {
return nil , nil , false
}
if valueType := reflect . TypeOf ( value ) ; valueType != nil && valueType . Kind ( ) == reflect . Array {
w := anyArrayArrayReflect {
array : reflect . ValueOf ( value ) ,
}
return & wrapArrayEncodeReflectPlan { } , w , true
}
return nil , nil , false
}
type wrapArrayEncodeReflectPlan struct {
next EncodePlan
}
func ( plan * wrapArrayEncodeReflectPlan ) SetNext ( next EncodePlan ) { plan . next = next }
func ( plan * wrapArrayEncodeReflectPlan ) Encode ( value any , buf [ ] byte ) ( newBuf [ ] byte , err error ) {
w := anyArrayArrayReflect {
array : reflect . ValueOf ( value ) ,
}
return plan . next . Encode ( w , buf )
}
func newEncodeError ( value any , m * Map , oid uint32 , formatCode int16 , err error ) error { func newEncodeError ( value any , m * Map , oid uint32 , formatCode int16 , err error ) error {
var format string
var format string
switch formatCode {
switch formatCode {
