package pgtype import ( "database/sql/driver" "encoding/binary" "fmt" "io" "math" "strconv" "strings" "github.com/jackc/pgx/pgio" ) type Circle struct { P Vec2 R float64 Status Status } func (dst *Circle) Set(src interface{}) error { return fmt.Errorf("cannot convert %v to Circle", src) } func (dst *Circle) Get() interface{} { switch dst.Status { case Present: return dst case Null: return nil default: return dst.Status } } func (src *Circle) AssignTo(dst interface{}) error { return fmt.Errorf("cannot assign %v to %T", src, dst) } func (dst *Circle) DecodeText(ci *ConnInfo, src []byte) error { if src == nil { *dst = Circle{Status: Null} return nil } if len(src) < 9 { return fmt.Errorf("invalid length for Circle: %v", len(src)) } str := string(src[2:]) end := strings.IndexByte(str, ',') x, err := strconv.ParseFloat(str[:end], 64) if err != nil { return err } str = str[end+1:] end = strings.IndexByte(str, ')') y, err := strconv.ParseFloat(str[:end], 64) if err != nil { return err } str = str[end+2 : len(str)-1] r, err := strconv.ParseFloat(str, 64) if err != nil { return err } *dst = Circle{P: Vec2{x, y}, R: r, Status: Present} return nil } func (dst *Circle) DecodeBinary(ci *ConnInfo, src []byte) error { if src == nil { *dst = Circle{Status: Null} return nil } if len(src) != 24 { return fmt.Errorf("invalid length for Circle: %v", len(src)) } x := binary.BigEndian.Uint64(src) y := binary.BigEndian.Uint64(src[8:]) r := binary.BigEndian.Uint64(src[16:]) *dst = Circle{ P: Vec2{math.Float64frombits(x), math.Float64frombits(y)}, R: math.Float64frombits(r), Status: Present, } return nil } func (src *Circle) 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, fmt.Sprintf(`<(%f,%f),%f>`, src.P.X, src.P.Y, src.R)) return false, err } func (src *Circle) EncodeBinary(ci *ConnInfo, w io.Writer) (bool, error) { switch src.Status { case Null: return true, nil case Undefined: return false, errUndefined } if _, err := pgio.WriteUint64(w, math.Float64bits(src.P.X)); err != nil { return false, err } if _, err := pgio.WriteUint64(w, math.Float64bits(src.P.Y)); err != nil { return false, err } _, err := pgio.WriteUint64(w, math.Float64bits(src.R)) return false, err } // Scan implements the database/sql Scanner interface. func (dst *Circle) Scan(src interface{}) error { if src == nil { *dst = Circle{Status: Null} return nil } switch src := src.(type) { case string: return dst.DecodeText(nil, []byte(src)) case []byte: return dst.DecodeText(nil, src) } return fmt.Errorf("cannot scan %T", src) } // Value implements the database/sql/driver Valuer interface. func (src *Circle) Value() (driver.Value, error) { return encodeValueText(src) }