Jack Christensen
666af9ead5
Though this doesn't follow Go naming conventions exactly it makes names more consistent with PostgreSQL and it is easier to read. For example, TIDOID becomes TidOid. In addition this is one less breaking change in the move to V3.
108 lines
2.2 KiB
Go
108 lines
2.2 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
|
|
}
|
|
|
|
// ConvertFrom converts from src to dst. Note that as pguint32 is not a general
|
|
// number type ConvertFrom does not do automatic type conversion as other number
|
|
// types do.
|
|
func (dst *pguint32) ConvertFrom(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
|
|
}
|
|
|
|
// 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(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(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(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(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
|
|
}
|