mirror of https://github.com/jackc/pgx.git
156 lines
3.0 KiB
Go
156 lines
3.0 KiB
Go
package pgtype
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"net"
|
|
|
|
errors "golang.org/x/xerrors"
|
|
)
|
|
|
|
type Macaddr struct {
|
|
Addr net.HardwareAddr
|
|
Status Status
|
|
}
|
|
|
|
func (dst *Macaddr) Set(src interface{}) error {
|
|
if src == nil {
|
|
*dst = Macaddr{Status: Null}
|
|
return nil
|
|
}
|
|
|
|
switch value := src.(type) {
|
|
case net.HardwareAddr:
|
|
addr := make(net.HardwareAddr, len(value))
|
|
copy(addr, value)
|
|
*dst = Macaddr{Addr: addr, Status: Present}
|
|
case string:
|
|
addr, err := net.ParseMAC(value)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*dst = Macaddr{Addr: addr, Status: Present}
|
|
default:
|
|
if originalSrc, ok := underlyingPtrType(src); ok {
|
|
return dst.Set(originalSrc)
|
|
}
|
|
return errors.Errorf("cannot convert %v to Macaddr", value)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (dst *Macaddr) Get() interface{} {
|
|
switch dst.Status {
|
|
case Present:
|
|
return dst.Addr
|
|
case Null:
|
|
return nil
|
|
default:
|
|
return dst.Status
|
|
}
|
|
}
|
|
|
|
func (src *Macaddr) AssignTo(dst interface{}) error {
|
|
switch src.Status {
|
|
case Present:
|
|
switch v := dst.(type) {
|
|
case *net.HardwareAddr:
|
|
*v = make(net.HardwareAddr, len(src.Addr))
|
|
copy(*v, src.Addr)
|
|
return nil
|
|
case *string:
|
|
*v = src.Addr.String()
|
|
return nil
|
|
default:
|
|
if nextDst, retry := GetAssignToDstType(dst); retry {
|
|
return src.AssignTo(nextDst)
|
|
}
|
|
return errors.Errorf("unable to assign to %T", dst)
|
|
}
|
|
case Null:
|
|
return NullAssignTo(dst)
|
|
}
|
|
|
|
return errors.Errorf("cannot decode %#v into %T", src, dst)
|
|
}
|
|
|
|
func (dst *Macaddr) DecodeText(ci *ConnInfo, src []byte) error {
|
|
if src == nil {
|
|
*dst = Macaddr{Status: Null}
|
|
return nil
|
|
}
|
|
|
|
addr, err := net.ParseMAC(string(src))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
*dst = Macaddr{Addr: addr, Status: Present}
|
|
return nil
|
|
}
|
|
|
|
func (dst *Macaddr) DecodeBinary(ci *ConnInfo, src []byte) error {
|
|
if src == nil {
|
|
*dst = Macaddr{Status: Null}
|
|
return nil
|
|
}
|
|
|
|
if len(src) != 6 {
|
|
return errors.Errorf("Received an invalid size for a macaddr: %d", len(src))
|
|
}
|
|
|
|
addr := make(net.HardwareAddr, 6)
|
|
copy(addr, src)
|
|
|
|
*dst = Macaddr{Addr: addr, Status: Present}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (src *Macaddr) EncodeText(ci *ConnInfo, buf []byte) ([]byte, error) {
|
|
switch src.Status {
|
|
case Null:
|
|
return nil, nil
|
|
case Undefined:
|
|
return nil, errUndefined
|
|
}
|
|
|
|
return append(buf, src.Addr.String()...), nil
|
|
}
|
|
|
|
// EncodeBinary encodes src into w.
|
|
func (src *Macaddr) EncodeBinary(ci *ConnInfo, buf []byte) ([]byte, error) {
|
|
switch src.Status {
|
|
case Null:
|
|
return nil, nil
|
|
case Undefined:
|
|
return nil, errUndefined
|
|
}
|
|
|
|
return append(buf, src.Addr...), nil
|
|
}
|
|
|
|
// Scan implements the database/sql Scanner interface.
|
|
func (dst *Macaddr) Scan(src interface{}) error {
|
|
if src == nil {
|
|
*dst = Macaddr{Status: Null}
|
|
return nil
|
|
}
|
|
|
|
switch src := src.(type) {
|
|
case string:
|
|
return dst.DecodeText(nil, []byte(src))
|
|
case []byte:
|
|
srcCopy := make([]byte, len(src))
|
|
copy(srcCopy, src)
|
|
return dst.DecodeText(nil, srcCopy)
|
|
}
|
|
|
|
return errors.Errorf("cannot scan %T", src)
|
|
}
|
|
|
|
// Value implements the database/sql/driver Valuer interface.
|
|
func (src *Macaddr) Value() (driver.Value, error) {
|
|
return EncodeValueText(src)
|
|
}
|