mirror of
https://github.com/redis/go-redis.git
synced 2025-06-05 06:42:39 +03:00
42c32846e6
* utils: expose ParseFloat via new public utils package * add tests for special float values in vector search
31 lines
643 B
Go
31 lines
643 B
Go
package util
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
"strconv"
|
|
)
|
|
|
|
// ParseFloat parses a Redis RESP3 float reply into a Go float64,
|
|
// handling "inf", "-inf", "nan" per Redis conventions.
|
|
func ParseStringToFloat(s string) (float64, error) {
|
|
switch s {
|
|
case "inf":
|
|
return math.Inf(1), nil
|
|
case "-inf":
|
|
return math.Inf(-1), nil
|
|
case "nan", "-nan":
|
|
return math.NaN(), nil
|
|
}
|
|
return strconv.ParseFloat(s, 64)
|
|
}
|
|
|
|
// MustParseFloat is like ParseFloat but panics on parse errors.
|
|
func MustParseFloat(s string) float64 {
|
|
f, err := ParseStringToFloat(s)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("redis: failed to parse float %q: %v", s, err))
|
|
}
|
|
return f
|
|
}
|