gopkg/pgx
2
0
Fork 0
Code Activity

pgtype.Numeric implements Float64Valuer

Browse Source
This commit is contained in:
Jack Christensen
2022-02-05 08:39:53 -06:00
parent a280f4db8a
commit a74ebc9e51
2 changed files with 56 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pgtype/numeric.go
+31 -24
View File
@@ -80,6 +80,35 @@ func (n Numeric) NumericValue() (Numeric, error) {
return n, nil
}
func (n Numeric) Float64Value() (Float8, error) {
if !n.Valid {
return Float8{}, nil
} else if n.NaN {
return Float8{Float: math.NaN(), Valid: true}, nil
} else if n.InfinityModifier == Infinity {
return Float8{Float: math.Inf(1), Valid: true}, nil
} else if n.InfinityModifier == NegativeInfinity {
return Float8{Float: math.Inf(-1), Valid: true}, nil
}
buf := make([]byte, 0, 32)
if n.Int == nil {
buf = append(buf, '0')
} else {
buf = append(buf, n.Int.String()...)
}
buf = append(buf, 'e')
buf = append(buf, strconv.FormatInt(int64(n.Exp), 10)...)
f, err := strconv.ParseFloat(string(buf), 64)
if err != nil {
return Float8{}, err
}
return Float8{Float: f, Valid: true}, nil
}
func (n *Numeric) toBigInt() (*big.Int, error) {
if n.Exp == 0 {
return n.Int, nil
@@ -104,28 +133,6 @@ func (n *Numeric) toBigInt() (*big.Int, error) {
return num, nil
}
func (n *Numeric) toFloat64() (float64, error) {
if n.NaN {
return math.NaN(), nil
} else if n.InfinityModifier == Infinity {
return math.Inf(1), nil
} else if n.InfinityModifier == NegativeInfinity {
return math.Inf(-1), nil
}
buf := make([]byte, 0, 32)
buf = append(buf, n.Int.String()...)
buf = append(buf, 'e')
buf = append(buf, strconv.FormatInt(int64(n.Exp), 10)...)
f, err := strconv.ParseFloat(string(buf), 64)
if err != nil {
return 0, err
}
return f, nil
}
func parseNumericString(str string) (n *big.Int, exp int32, err error) {
parts := strings.SplitN(str, ".", 2)
digits := strings.Join(parts, "")
@@ -642,12 +649,12 @@ func (scanPlanBinaryNumericToFloat64Scanner) Scan(src []byte, dst interface{}) e
return err
}
f64, err := n.toFloat64()
f8, err := n.Float64Value()
if err != nil {
return err
}
return scanner.ScanFloat64(Float8{Float: f64, Valid: true})
return scanner.ScanFloat64(f8)
}
type scanPlanBinaryNumericToInt64Scanner struct{}