Add zeronull float8

non-blocking
Jack Christensen 2021-08-26 13:21:02 -05:00
parent 94f8441f4e
commit 39aa071b15
2 changed files with 113 additions and 0 deletions

90
zeronull/float8.go Normal file
View File

@ -0,0 +1,90 @@
package zeronull
import (
"database/sql/driver"
"github.com/jackc/pgtype"
)
type Float8 int64
func (dst *Float8) DecodeText(ci *pgtype.ConnInfo, src []byte) error {
var nullable pgtype.Float8
err := nullable.DecodeText(ci, src)
if err != nil {
return err
}
if nullable.Status == pgtype.Present {
*dst = Float8(nullable.Float)
} else {
*dst = 0
}
return nil
}
func (dst *Float8) DecodeBinary(ci *pgtype.ConnInfo, src []byte) error {
var nullable pgtype.Float8
err := nullable.DecodeBinary(ci, src)
if err != nil {
return err
}
if nullable.Status == pgtype.Present {
*dst = Float8(nullable.Float)
} else {
*dst = 0
}
return nil
}
func (src Float8) EncodeText(ci *pgtype.ConnInfo, buf []byte) ([]byte, error) {
if src == 0 {
return nil, nil
}
nullable := pgtype.Float8{
Float: float64(src),
Status: pgtype.Present,
}
return nullable.EncodeText(ci, buf)
}
func (src Float8) EncodeBinary(ci *pgtype.ConnInfo, buf []byte) ([]byte, error) {
if src == 0 {
return nil, nil
}
nullable := pgtype.Float8{
Float: float64(src),
Status: pgtype.Present,
}
return nullable.EncodeBinary(ci, buf)
}
// Scan implements the database/sql Scanner interface.
func (dst *Float8) Scan(src interface{}) error {
if src == nil {
*dst = 0
return nil
}
var nullable pgtype.Float8
err := nullable.Scan(src)
if err != nil {
return err
}
*dst = Float8(nullable.Float)
return nil
}
// Value implements the database/sql/driver Valuer interface.
func (src Float8) Value() (driver.Value, error) {
return pgtype.EncodeValueText(src)
}

23
zeronull/float8_test.go Normal file
View File

@ -0,0 +1,23 @@
package zeronull_test
import (
"testing"
"github.com/jackc/pgtype/testutil"
"github.com/jackc/pgtype/zeronull"
)
func TestFloat8Transcode(t *testing.T) {
testutil.TestSuccessfulTranscode(t, "float8", []interface{}{
(zeronull.Float8)(1),
(zeronull.Float8)(0),
})
}
func TestFloat8ConvertsGoZeroToNull(t *testing.T) {
testutil.TestGoZeroToNullConversion(t, "float8", (zeronull.Float8)(0))
}
func TestFloat8ConvertsNullToGoZero(t *testing.T) {
testutil.TestNullToGoZeroConversion(t, "float8", (zeronull.Float8)(0))
}