Jack Christensen
19c6689752
Because reading a record type requires the decoder to be able to look up oid to type mapping and types such as hstore have types that are not fixed between different PostgreSQL servers it was necessary to restructure the pgtype system so all encoders and decodes take a *ConnInfo that includes oid/name/type information.
119 lines
2.3 KiB
Go
119 lines
2.3 KiB
Go
package pgtype
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
|
|
"github.com/jackc/pgx/pgio"
|
|
)
|
|
|
|
// pguint32 is the core type that is used to implement PostgreSQL types such as
|
|
// Cid and Xid.
|
|
type pguint32 struct {
|
|
Uint uint32
|
|
Status Status
|
|
}
|
|
|
|
// Set converts from src to dst. Note that as pguint32 is not a general
|
|
// number type Set does not do automatic type conversion as other number
|
|
// types do.
|
|
func (dst *pguint32) Set(src interface{}) error {
|
|
switch value := src.(type) {
|
|
case uint32:
|
|
*dst = pguint32{Uint: value, Status: Present}
|
|
default:
|
|
return fmt.Errorf("cannot convert %v to pguint32", value)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (dst *pguint32) Get() interface{} {
|
|
switch dst.Status {
|
|
case Present:
|
|
return dst.Uint
|
|
case Null:
|
|
return nil
|
|
default:
|
|
return dst.Status
|
|
}
|
|
}
|
|
|
|
// AssignTo assigns from src to dst. Note that as pguint32 is not a general number
|
|
// type AssignTo does not do automatic type conversion as other number types do.
|
|
func (src *pguint32) AssignTo(dst interface{}) error {
|
|
switch v := dst.(type) {
|
|
case *uint32:
|
|
if src.Status == Present {
|
|
*v = src.Uint
|
|
} else {
|
|
return fmt.Errorf("cannot assign %v into %T", src, dst)
|
|
}
|
|
case **uint32:
|
|
if src.Status == Present {
|
|
n := src.Uint
|
|
*v = &n
|
|
} else {
|
|
*v = nil
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (dst *pguint32) DecodeText(ci *ConnInfo, src []byte) error {
|
|
if src == nil {
|
|
*dst = pguint32{Status: Null}
|
|
return nil
|
|
}
|
|
|
|
n, err := strconv.ParseUint(string(src), 10, 32)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
*dst = pguint32{Uint: uint32(n), Status: Present}
|
|
return nil
|
|
}
|
|
|
|
func (dst *pguint32) DecodeBinary(ci *ConnInfo, src []byte) error {
|
|
if src == nil {
|
|
*dst = pguint32{Status: Null}
|
|
return nil
|
|
}
|
|
|
|
if len(src) != 4 {
|
|
return fmt.Errorf("invalid length: %v", len(src))
|
|
}
|
|
|
|
n := binary.BigEndian.Uint32(src)
|
|
*dst = pguint32{Uint: n, Status: Present}
|
|
return nil
|
|
}
|
|
|
|
func (src pguint32) EncodeText(ci *ConnInfo, w io.Writer) (bool, error) {
|
|
switch src.Status {
|
|
case Null:
|
|
return true, nil
|
|
case Undefined:
|
|
return false, errUndefined
|
|
}
|
|
|
|
_, err := io.WriteString(w, strconv.FormatUint(uint64(src.Uint), 10))
|
|
return false, err
|
|
}
|
|
|
|
func (src pguint32) EncodeBinary(ci *ConnInfo, w io.Writer) (bool, error) {
|
|
switch src.Status {
|
|
case Null:
|
|
return true, nil
|
|
case Undefined:
|
|
return false, errUndefined
|
|
}
|
|
|
|
_, err := pgio.WriteUint32(w, src.Uint)
|
|
return false, err
|
|
}
|