|
|
package njson
|
|
|
|
|
|
import (
|
|
|
"bytes"
|
|
|
"encoding/json"
|
|
|
"fmt"
|
|
|
"io"
|
|
|
"strings"
|
|
|
)
|
|
|
|
|
|
// Marshal marshals v into json bytes.
|
|
|
// see json.Marshal
|
|
|
func Marshal(v any) ([]byte, error) {
|
|
|
return json.Marshal(v)
|
|
|
}
|
|
|
|
|
|
// MarshalStr marshals v into a string.
|
|
|
func MarshalStr(v any) (string, error) {
|
|
|
data, err := Marshal(v)
|
|
|
if err != nil {
|
|
|
return "", err
|
|
|
}
|
|
|
return string(data), nil
|
|
|
}
|
|
|
|
|
|
// MarshalStrSafe marshals v into a string.
|
|
|
// return an empty string if an error occurred.
|
|
|
func MarshalStrSafe(v any) string {
|
|
|
ret, _ := MarshalStr(v)
|
|
|
return ret
|
|
|
}
|
|
|
|
|
|
// Unmarshal data bytes into v.
|
|
|
func Unmarshal(data []byte, v any) error {
|
|
|
decoder := json.NewDecoder(bytes.NewReader(data))
|
|
|
if err := unmarshalUseNumber(decoder, v); err != nil {
|
|
|
return formatError(string(data), err)
|
|
|
}
|
|
|
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
// UnmarshalFromStr unmarshal v from str.
|
|
|
func UnmarshalFromStr(str string, v any) error {
|
|
|
decoder := json.NewDecoder(strings.NewReader(str))
|
|
|
if err := unmarshalUseNumber(decoder, v); err != nil {
|
|
|
return formatError(str, err)
|
|
|
}
|
|
|
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
// UnmarshalFromReader unmarshal v from reader.
|
|
|
func UnmarshalFromReader(reader io.Reader, v any) error {
|
|
|
var buf strings.Builder
|
|
|
teeReader := io.TeeReader(reader, &buf)
|
|
|
decoder := json.NewDecoder(teeReader)
|
|
|
if err := unmarshalUseNumber(decoder, v); err != nil {
|
|
|
return formatError(buf.String(), err)
|
|
|
}
|
|
|
|
|
|
return nil
|
|
|
}
|
|
|
|
|
|
// unmarshalUseNumber 避免出现科学计算，如1e+100
|
|
|
// 这是由于默认 unmarshal 时会将大于6位的数值类型使用 float64 解析
|
|
|
// 同时也可以避免精度丢失 double -> float64 等
|
|
|
func unmarshalUseNumber(decoder *json.Decoder, v any) error {
|
|
|
decoder.UseNumber()
|
|
|
return decoder.Decode(v)
|
|
|
}
|
|
|
|
|
|
func formatError(v string, err error) error {
|
|
|
return fmt.Errorf("string: `%s`, error: `%w`", v, err)
|
|
|
}
|