mirror of https://github.com/jackc/pgx.git
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
|
|
}
|