mirror of
https://github.com/go-gitea/gitea.git
synced 2024-08-20 03:23:26 +02:00
* update code.gitea.io/sdk/gitea v0.13.1 -> v0.13.2 * update github.com/go-swagger/go-swagger v0.25.0 -> v0.26.0 * update github.com/google/uuid v1.1.2 -> v1.2.0 * update github.com/klauspost/compress v1.11.3 -> v1.11.7 * update github.com/lib/pq 083382b7e6fc -> v1.9.0 * update github.com/markbates/goth v1.65.0 -> v1.66.1 * update github.com/mattn/go-sqlite3 v1.14.4 -> v1.14.6 * update github.com/mgechev/revive 246eac737dc7 -> v1.0.3 * update github.com/minio/minio-go/v7 v7.0.6 -> v7.0.7 * update github.com/niklasfasching/go-org v1.3.2 -> v1.4.0 * update github.com/olivere/elastic/v7 v7.0.21 -> v7.0.22 * update github.com/pquerna/otp v1.2.0 -> v1.3.0 * update github.com/xanzy/go-gitlab v0.39.0 -> v0.42.0 * update github.com/yuin/goldmark v1.2.1 -> v1.3.1
1329 lines
38 KiB
Go
Vendored
1329 lines
38 KiB
Go
Vendored
// Copyright (C) MongoDB, Inc. 2017-present.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
// not use this file except in compliance with the License. You may obtain
|
|
// a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
package bsoncodec
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"math"
|
|
"net/url"
|
|
"reflect"
|
|
"strconv"
|
|
"time"
|
|
|
|
"go.mongodb.org/mongo-driver/bson/bsonrw"
|
|
"go.mongodb.org/mongo-driver/bson/bsontype"
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
"go.mongodb.org/mongo-driver/x/bsonx/bsoncore"
|
|
)
|
|
|
|
var defaultValueDecoders DefaultValueDecoders
|
|
|
|
func newDefaultStructCodec() *StructCodec {
|
|
codec, err := NewStructCodec(DefaultStructTagParser)
|
|
if err != nil {
|
|
// This function is called from the codec registration path, so errors can't be propagated. If there's an error
|
|
// constructing the StructCodec, we panic to avoid losing it.
|
|
panic(fmt.Errorf("error creating default StructCodec: %v", err))
|
|
}
|
|
return codec
|
|
}
|
|
|
|
// DefaultValueDecoders is a namespace type for the default ValueDecoders used
|
|
// when creating a registry.
|
|
type DefaultValueDecoders struct{}
|
|
|
|
// RegisterDefaultDecoders will register the decoder methods attached to DefaultValueDecoders with
|
|
// the provided RegistryBuilder.
|
|
//
|
|
// There is no support for decoding map[string]interface{} becuase there is no decoder for
|
|
// interface{}, so users must either register this decoder themselves or use the
|
|
// EmptyInterfaceDecoder avaialble in the bson package.
|
|
func (dvd DefaultValueDecoders) RegisterDefaultDecoders(rb *RegistryBuilder) {
|
|
if rb == nil {
|
|
panic(errors.New("argument to RegisterDefaultDecoders must not be nil"))
|
|
}
|
|
|
|
rb.
|
|
RegisterTypeDecoder(tBinary, ValueDecoderFunc(dvd.BinaryDecodeValue)).
|
|
RegisterTypeDecoder(tUndefined, ValueDecoderFunc(dvd.UndefinedDecodeValue)).
|
|
RegisterTypeDecoder(tDateTime, ValueDecoderFunc(dvd.DateTimeDecodeValue)).
|
|
RegisterTypeDecoder(tNull, ValueDecoderFunc(dvd.NullDecodeValue)).
|
|
RegisterTypeDecoder(tRegex, ValueDecoderFunc(dvd.RegexDecodeValue)).
|
|
RegisterTypeDecoder(tDBPointer, ValueDecoderFunc(dvd.DBPointerDecodeValue)).
|
|
RegisterTypeDecoder(tTimestamp, ValueDecoderFunc(dvd.TimestampDecodeValue)).
|
|
RegisterTypeDecoder(tMinKey, ValueDecoderFunc(dvd.MinKeyDecodeValue)).
|
|
RegisterTypeDecoder(tMaxKey, ValueDecoderFunc(dvd.MaxKeyDecodeValue)).
|
|
RegisterTypeDecoder(tJavaScript, ValueDecoderFunc(dvd.JavaScriptDecodeValue)).
|
|
RegisterTypeDecoder(tSymbol, ValueDecoderFunc(dvd.SymbolDecodeValue)).
|
|
RegisterTypeDecoder(tByteSlice, defaultByteSliceCodec).
|
|
RegisterTypeDecoder(tTime, defaultTimeCodec).
|
|
RegisterTypeDecoder(tEmpty, defaultEmptyInterfaceCodec).
|
|
RegisterTypeDecoder(tOID, ValueDecoderFunc(dvd.ObjectIDDecodeValue)).
|
|
RegisterTypeDecoder(tDecimal, ValueDecoderFunc(dvd.Decimal128DecodeValue)).
|
|
RegisterTypeDecoder(tJSONNumber, ValueDecoderFunc(dvd.JSONNumberDecodeValue)).
|
|
RegisterTypeDecoder(tURL, ValueDecoderFunc(dvd.URLDecodeValue)).
|
|
RegisterTypeDecoder(tCoreDocument, ValueDecoderFunc(dvd.CoreDocumentDecodeValue)).
|
|
RegisterTypeDecoder(tCodeWithScope, ValueDecoderFunc(dvd.CodeWithScopeDecodeValue)).
|
|
RegisterDefaultDecoder(reflect.Bool, ValueDecoderFunc(dvd.BooleanDecodeValue)).
|
|
RegisterDefaultDecoder(reflect.Int, ValueDecoderFunc(dvd.IntDecodeValue)).
|
|
RegisterDefaultDecoder(reflect.Int8, ValueDecoderFunc(dvd.IntDecodeValue)).
|
|
RegisterDefaultDecoder(reflect.Int16, ValueDecoderFunc(dvd.IntDecodeValue)).
|
|
RegisterDefaultDecoder(reflect.Int32, ValueDecoderFunc(dvd.IntDecodeValue)).
|
|
RegisterDefaultDecoder(reflect.Int64, ValueDecoderFunc(dvd.IntDecodeValue)).
|
|
RegisterDefaultDecoder(reflect.Uint, defaultUIntCodec).
|
|
RegisterDefaultDecoder(reflect.Uint8, defaultUIntCodec).
|
|
RegisterDefaultDecoder(reflect.Uint16, defaultUIntCodec).
|
|
RegisterDefaultDecoder(reflect.Uint32, defaultUIntCodec).
|
|
RegisterDefaultDecoder(reflect.Uint64, defaultUIntCodec).
|
|
RegisterDefaultDecoder(reflect.Float32, ValueDecoderFunc(dvd.FloatDecodeValue)).
|
|
RegisterDefaultDecoder(reflect.Float64, ValueDecoderFunc(dvd.FloatDecodeValue)).
|
|
RegisterDefaultDecoder(reflect.Array, ValueDecoderFunc(dvd.ArrayDecodeValue)).
|
|
RegisterDefaultDecoder(reflect.Map, defaultMapCodec).
|
|
RegisterDefaultDecoder(reflect.Slice, defaultSliceCodec).
|
|
RegisterDefaultDecoder(reflect.String, defaultStringCodec).
|
|
RegisterDefaultDecoder(reflect.Struct, newDefaultStructCodec()).
|
|
RegisterDefaultDecoder(reflect.Ptr, NewPointerCodec()).
|
|
RegisterTypeMapEntry(bsontype.Double, tFloat64).
|
|
RegisterTypeMapEntry(bsontype.String, tString).
|
|
RegisterTypeMapEntry(bsontype.Array, tA).
|
|
RegisterTypeMapEntry(bsontype.Binary, tBinary).
|
|
RegisterTypeMapEntry(bsontype.Undefined, tUndefined).
|
|
RegisterTypeMapEntry(bsontype.ObjectID, tOID).
|
|
RegisterTypeMapEntry(bsontype.Boolean, tBool).
|
|
RegisterTypeMapEntry(bsontype.DateTime, tDateTime).
|
|
RegisterTypeMapEntry(bsontype.Regex, tRegex).
|
|
RegisterTypeMapEntry(bsontype.DBPointer, tDBPointer).
|
|
RegisterTypeMapEntry(bsontype.JavaScript, tJavaScript).
|
|
RegisterTypeMapEntry(bsontype.Symbol, tSymbol).
|
|
RegisterTypeMapEntry(bsontype.CodeWithScope, tCodeWithScope).
|
|
RegisterTypeMapEntry(bsontype.Int32, tInt32).
|
|
RegisterTypeMapEntry(bsontype.Int64, tInt64).
|
|
RegisterTypeMapEntry(bsontype.Timestamp, tTimestamp).
|
|
RegisterTypeMapEntry(bsontype.Decimal128, tDecimal).
|
|
RegisterTypeMapEntry(bsontype.MinKey, tMinKey).
|
|
RegisterTypeMapEntry(bsontype.MaxKey, tMaxKey).
|
|
RegisterTypeMapEntry(bsontype.Type(0), tD).
|
|
RegisterTypeMapEntry(bsontype.EmbeddedDocument, tD).
|
|
RegisterHookDecoder(tValueUnmarshaler, ValueDecoderFunc(dvd.ValueUnmarshalerDecodeValue)).
|
|
RegisterHookDecoder(tUnmarshaler, ValueDecoderFunc(dvd.UnmarshalerDecodeValue))
|
|
}
|
|
|
|
// BooleanDecodeValue is the ValueDecoderFunc for bool types.
|
|
func (dvd DefaultValueDecoders) BooleanDecodeValue(dctx DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.IsValid() || !val.CanSet() || val.Kind() != reflect.Bool {
|
|
return ValueDecoderError{Name: "BooleanDecodeValue", Kinds: []reflect.Kind{reflect.Bool}, Received: val}
|
|
}
|
|
|
|
var b bool
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.Int32:
|
|
i32, err := vr.ReadInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
b = (i32 != 0)
|
|
case bsontype.Int64:
|
|
i64, err := vr.ReadInt64()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
b = (i64 != 0)
|
|
case bsontype.Double:
|
|
f64, err := vr.ReadDouble()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
b = (f64 != 0)
|
|
case bsontype.Boolean:
|
|
b, err = vr.ReadBoolean()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
case bsontype.Null:
|
|
if err = vr.ReadNull(); err != nil {
|
|
return err
|
|
}
|
|
case bsontype.Undefined:
|
|
if err = vr.ReadUndefined(); err != nil {
|
|
return err
|
|
}
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a boolean", vrType)
|
|
}
|
|
val.SetBool(b)
|
|
return nil
|
|
}
|
|
|
|
// IntDecodeValue is the ValueDecoderFunc for int types.
|
|
func (dvd DefaultValueDecoders) IntDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() {
|
|
return ValueDecoderError{
|
|
Name: "IntDecodeValue",
|
|
Kinds: []reflect.Kind{reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64, reflect.Int},
|
|
Received: val,
|
|
}
|
|
}
|
|
|
|
var i64 int64
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.Int32:
|
|
i32, err := vr.ReadInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
i64 = int64(i32)
|
|
case bsontype.Int64:
|
|
i64, err = vr.ReadInt64()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
case bsontype.Double:
|
|
f64, err := vr.ReadDouble()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !dc.Truncate && math.Floor(f64) != f64 {
|
|
return errors.New("IntDecodeValue can only truncate float64 to an integer type when truncation is enabled")
|
|
}
|
|
if f64 > float64(math.MaxInt64) {
|
|
return fmt.Errorf("%g overflows int64", f64)
|
|
}
|
|
i64 = int64(f64)
|
|
case bsontype.Boolean:
|
|
b, err := vr.ReadBoolean()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if b {
|
|
i64 = 1
|
|
}
|
|
case bsontype.Null:
|
|
if err = vr.ReadNull(); err != nil {
|
|
return err
|
|
}
|
|
case bsontype.Undefined:
|
|
if err = vr.ReadUndefined(); err != nil {
|
|
return err
|
|
}
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into an integer type", vrType)
|
|
}
|
|
|
|
switch val.Kind() {
|
|
case reflect.Int8:
|
|
if i64 < math.MinInt8 || i64 > math.MaxInt8 {
|
|
return fmt.Errorf("%d overflows int8", i64)
|
|
}
|
|
case reflect.Int16:
|
|
if i64 < math.MinInt16 || i64 > math.MaxInt16 {
|
|
return fmt.Errorf("%d overflows int16", i64)
|
|
}
|
|
case reflect.Int32:
|
|
if i64 < math.MinInt32 || i64 > math.MaxInt32 {
|
|
return fmt.Errorf("%d overflows int32", i64)
|
|
}
|
|
case reflect.Int64:
|
|
case reflect.Int:
|
|
if int64(int(i64)) != i64 { // Can we fit this inside of an int
|
|
return fmt.Errorf("%d overflows int", i64)
|
|
}
|
|
default:
|
|
return ValueDecoderError{
|
|
Name: "IntDecodeValue",
|
|
Kinds: []reflect.Kind{reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64, reflect.Int},
|
|
Received: val,
|
|
}
|
|
}
|
|
|
|
val.SetInt(i64)
|
|
return nil
|
|
}
|
|
|
|
// UintDecodeValue is the ValueDecoderFunc for uint types.
|
|
//
|
|
// Deprecated: UintDecodeValue is not registered by default. Use UintCodec.DecodeValue instead.
|
|
func (dvd DefaultValueDecoders) UintDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
var i64 int64
|
|
var err error
|
|
switch vr.Type() {
|
|
case bsontype.Int32:
|
|
i32, err := vr.ReadInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
i64 = int64(i32)
|
|
case bsontype.Int64:
|
|
i64, err = vr.ReadInt64()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
case bsontype.Double:
|
|
f64, err := vr.ReadDouble()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !dc.Truncate && math.Floor(f64) != f64 {
|
|
return errors.New("UintDecodeValue can only truncate float64 to an integer type when truncation is enabled")
|
|
}
|
|
if f64 > float64(math.MaxInt64) {
|
|
return fmt.Errorf("%g overflows int64", f64)
|
|
}
|
|
i64 = int64(f64)
|
|
case bsontype.Boolean:
|
|
b, err := vr.ReadBoolean()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if b {
|
|
i64 = 1
|
|
}
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into an integer type", vr.Type())
|
|
}
|
|
|
|
if !val.CanSet() {
|
|
return ValueDecoderError{
|
|
Name: "UintDecodeValue",
|
|
Kinds: []reflect.Kind{reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uint},
|
|
Received: val,
|
|
}
|
|
}
|
|
|
|
switch val.Kind() {
|
|
case reflect.Uint8:
|
|
if i64 < 0 || i64 > math.MaxUint8 {
|
|
return fmt.Errorf("%d overflows uint8", i64)
|
|
}
|
|
case reflect.Uint16:
|
|
if i64 < 0 || i64 > math.MaxUint16 {
|
|
return fmt.Errorf("%d overflows uint16", i64)
|
|
}
|
|
case reflect.Uint32:
|
|
if i64 < 0 || i64 > math.MaxUint32 {
|
|
return fmt.Errorf("%d overflows uint32", i64)
|
|
}
|
|
case reflect.Uint64:
|
|
if i64 < 0 {
|
|
return fmt.Errorf("%d overflows uint64", i64)
|
|
}
|
|
case reflect.Uint:
|
|
if i64 < 0 || int64(uint(i64)) != i64 { // Can we fit this inside of an uint
|
|
return fmt.Errorf("%d overflows uint", i64)
|
|
}
|
|
default:
|
|
return ValueDecoderError{
|
|
Name: "UintDecodeValue",
|
|
Kinds: []reflect.Kind{reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uint},
|
|
Received: val,
|
|
}
|
|
}
|
|
|
|
val.SetUint(uint64(i64))
|
|
return nil
|
|
}
|
|
|
|
// FloatDecodeValue is the ValueDecoderFunc for float types.
|
|
func (dvd DefaultValueDecoders) FloatDecodeValue(ec DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() {
|
|
return ValueDecoderError{
|
|
Name: "FloatDecodeValue",
|
|
Kinds: []reflect.Kind{reflect.Float32, reflect.Float64},
|
|
Received: val,
|
|
}
|
|
}
|
|
|
|
var f float64
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.Int32:
|
|
i32, err := vr.ReadInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
f = float64(i32)
|
|
case bsontype.Int64:
|
|
i64, err := vr.ReadInt64()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
f = float64(i64)
|
|
case bsontype.Double:
|
|
f, err = vr.ReadDouble()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
case bsontype.Boolean:
|
|
b, err := vr.ReadBoolean()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if b {
|
|
f = 1
|
|
}
|
|
case bsontype.Null:
|
|
if err = vr.ReadNull(); err != nil {
|
|
return err
|
|
}
|
|
case bsontype.Undefined:
|
|
if err = vr.ReadUndefined(); err != nil {
|
|
return err
|
|
}
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a float32 or float64 type", vrType)
|
|
}
|
|
|
|
switch val.Kind() {
|
|
case reflect.Float32:
|
|
if !ec.Truncate && float64(float32(f)) != f {
|
|
return errors.New("FloatDecodeValue can only convert float64 to float32 when truncation is allowed")
|
|
}
|
|
case reflect.Float64:
|
|
default:
|
|
return ValueDecoderError{Name: "FloatDecodeValue", Kinds: []reflect.Kind{reflect.Float32, reflect.Float64}, Received: val}
|
|
}
|
|
|
|
val.SetFloat(f)
|
|
return nil
|
|
}
|
|
|
|
// StringDecodeValue is the ValueDecoderFunc for string types.
|
|
//
|
|
// Deprecated: StringDecodeValue is not registered by default. Use StringCodec.DecodeValue instead.
|
|
func (dvd DefaultValueDecoders) StringDecodeValue(dctx DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
var str string
|
|
var err error
|
|
switch vr.Type() {
|
|
// TODO(GODRIVER-577): Handle JavaScript and Symbol BSON types when allowed.
|
|
case bsontype.String:
|
|
str, err = vr.ReadString()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a string type", vr.Type())
|
|
}
|
|
if !val.CanSet() || val.Kind() != reflect.String {
|
|
return ValueDecoderError{Name: "StringDecodeValue", Kinds: []reflect.Kind{reflect.String}, Received: val}
|
|
}
|
|
|
|
val.SetString(str)
|
|
return nil
|
|
}
|
|
|
|
// JavaScriptDecodeValue is the ValueDecoderFunc for the primitive.JavaScript type.
|
|
func (DefaultValueDecoders) JavaScriptDecodeValue(dctx DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tJavaScript {
|
|
return ValueDecoderError{Name: "JavaScriptDecodeValue", Types: []reflect.Type{tJavaScript}, Received: val}
|
|
}
|
|
|
|
var js string
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.JavaScript:
|
|
js, err = vr.ReadJavascript()
|
|
case bsontype.Null:
|
|
err = vr.ReadNull()
|
|
case bsontype.Undefined:
|
|
err = vr.ReadUndefined()
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a primitive.JavaScript", vrType)
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.SetString(js)
|
|
return nil
|
|
}
|
|
|
|
// SymbolDecodeValue is the ValueDecoderFunc for the primitive.Symbol type.
|
|
func (DefaultValueDecoders) SymbolDecodeValue(dctx DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tSymbol {
|
|
return ValueDecoderError{Name: "SymbolDecodeValue", Types: []reflect.Type{tSymbol}, Received: val}
|
|
}
|
|
|
|
var symbol string
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.String:
|
|
symbol, err = vr.ReadString()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
case bsontype.Symbol:
|
|
symbol, err = vr.ReadSymbol()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
case bsontype.Binary:
|
|
data, subtype, err := vr.ReadBinary()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if subtype != bsontype.BinaryGeneric && subtype != bsontype.BinaryBinaryOld {
|
|
return fmt.Errorf("SymbolDecodeValue can only be used to decode subtype 0x00 or 0x02 for %s, got %v", bsontype.Binary, subtype)
|
|
}
|
|
symbol = string(data)
|
|
case bsontype.Null:
|
|
if err = vr.ReadNull(); err != nil {
|
|
return err
|
|
}
|
|
case bsontype.Undefined:
|
|
if err = vr.ReadUndefined(); err != nil {
|
|
return err
|
|
}
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a primitive.Symbol", vrType)
|
|
}
|
|
|
|
val.SetString(symbol)
|
|
return nil
|
|
}
|
|
|
|
// BinaryDecodeValue is the ValueDecoderFunc for Binary.
|
|
func (DefaultValueDecoders) BinaryDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tBinary {
|
|
return ValueDecoderError{Name: "BinaryDecodeValue", Types: []reflect.Type{tBinary}, Received: val}
|
|
}
|
|
|
|
var data []byte
|
|
var subtype byte
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.Binary:
|
|
data, subtype, err = vr.ReadBinary()
|
|
case bsontype.Null:
|
|
err = vr.ReadNull()
|
|
case bsontype.Undefined:
|
|
err = vr.ReadUndefined()
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a Binary", vrType)
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(primitive.Binary{Subtype: subtype, Data: data}))
|
|
return nil
|
|
}
|
|
|
|
// UndefinedDecodeValue is the ValueDecoderFunc for Undefined.
|
|
func (DefaultValueDecoders) UndefinedDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tUndefined {
|
|
return ValueDecoderError{Name: "UndefinedDecodeValue", Types: []reflect.Type{tUndefined}, Received: val}
|
|
}
|
|
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.Undefined:
|
|
err = vr.ReadUndefined()
|
|
case bsontype.Null:
|
|
err = vr.ReadNull()
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into an Undefined", vr.Type())
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(primitive.Undefined{}))
|
|
return nil
|
|
}
|
|
|
|
// ObjectIDDecodeValue is the ValueDecoderFunc for primitive.ObjectID.
|
|
func (dvd DefaultValueDecoders) ObjectIDDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tOID {
|
|
return ValueDecoderError{Name: "ObjectIDDecodeValue", Types: []reflect.Type{tOID}, Received: val}
|
|
}
|
|
|
|
var oid primitive.ObjectID
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.ObjectID:
|
|
oid, err = vr.ReadObjectID()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
case bsontype.String:
|
|
str, err := vr.ReadString()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(str) != 12 {
|
|
return fmt.Errorf("an ObjectID string must be exactly 12 bytes long (got %v)", len(str))
|
|
}
|
|
byteArr := []byte(str)
|
|
copy(oid[:], byteArr)
|
|
case bsontype.Null:
|
|
if err = vr.ReadNull(); err != nil {
|
|
return err
|
|
}
|
|
case bsontype.Undefined:
|
|
if err = vr.ReadUndefined(); err != nil {
|
|
return err
|
|
}
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into an ObjectID", vrType)
|
|
}
|
|
|
|
val.Set(reflect.ValueOf(oid))
|
|
return nil
|
|
}
|
|
|
|
// DateTimeDecodeValue is the ValueDecoderFunc for DateTime.
|
|
func (DefaultValueDecoders) DateTimeDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tDateTime {
|
|
return ValueDecoderError{Name: "DateTimeDecodeValue", Types: []reflect.Type{tDateTime}, Received: val}
|
|
}
|
|
|
|
var dt int64
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.DateTime:
|
|
dt, err = vr.ReadDateTime()
|
|
case bsontype.Null:
|
|
err = vr.ReadNull()
|
|
case bsontype.Undefined:
|
|
err = vr.ReadUndefined()
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a DateTime", vrType)
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(primitive.DateTime(dt)))
|
|
return nil
|
|
}
|
|
|
|
// NullDecodeValue is the ValueDecoderFunc for Null.
|
|
func (DefaultValueDecoders) NullDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tNull {
|
|
return ValueDecoderError{Name: "NullDecodeValue", Types: []reflect.Type{tNull}, Received: val}
|
|
}
|
|
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.Undefined:
|
|
err = vr.ReadUndefined()
|
|
case bsontype.Null:
|
|
err = vr.ReadNull()
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a Null", vr.Type())
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(primitive.Null{}))
|
|
return nil
|
|
}
|
|
|
|
// RegexDecodeValue is the ValueDecoderFunc for Regex.
|
|
func (DefaultValueDecoders) RegexDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tRegex {
|
|
return ValueDecoderError{Name: "RegexDecodeValue", Types: []reflect.Type{tRegex}, Received: val}
|
|
}
|
|
|
|
var pattern, options string
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.Regex:
|
|
pattern, options, err = vr.ReadRegex()
|
|
case bsontype.Null:
|
|
err = vr.ReadNull()
|
|
case bsontype.Undefined:
|
|
err = vr.ReadUndefined()
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a Regex", vrType)
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(primitive.Regex{Pattern: pattern, Options: options}))
|
|
return nil
|
|
}
|
|
|
|
// DBPointerDecodeValue is the ValueDecoderFunc for DBPointer.
|
|
func (DefaultValueDecoders) DBPointerDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tDBPointer {
|
|
return ValueDecoderError{Name: "DBPointerDecodeValue", Types: []reflect.Type{tDBPointer}, Received: val}
|
|
}
|
|
|
|
var ns string
|
|
var pointer primitive.ObjectID
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.DBPointer:
|
|
ns, pointer, err = vr.ReadDBPointer()
|
|
case bsontype.Null:
|
|
err = vr.ReadNull()
|
|
case bsontype.Undefined:
|
|
err = vr.ReadUndefined()
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a DBPointer", vrType)
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(primitive.DBPointer{DB: ns, Pointer: pointer}))
|
|
return nil
|
|
}
|
|
|
|
// TimestampDecodeValue is the ValueDecoderFunc for Timestamp.
|
|
func (DefaultValueDecoders) TimestampDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tTimestamp {
|
|
return ValueDecoderError{Name: "TimestampDecodeValue", Types: []reflect.Type{tTimestamp}, Received: val}
|
|
}
|
|
|
|
var t, incr uint32
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.Timestamp:
|
|
t, incr, err = vr.ReadTimestamp()
|
|
case bsontype.Null:
|
|
err = vr.ReadNull()
|
|
case bsontype.Undefined:
|
|
err = vr.ReadUndefined()
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a Timestamp", vrType)
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(primitive.Timestamp{T: t, I: incr}))
|
|
return nil
|
|
}
|
|
|
|
// MinKeyDecodeValue is the ValueDecoderFunc for MinKey.
|
|
func (DefaultValueDecoders) MinKeyDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tMinKey {
|
|
return ValueDecoderError{Name: "MinKeyDecodeValue", Types: []reflect.Type{tMinKey}, Received: val}
|
|
}
|
|
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.MinKey:
|
|
err = vr.ReadMinKey()
|
|
case bsontype.Null:
|
|
err = vr.ReadNull()
|
|
case bsontype.Undefined:
|
|
err = vr.ReadUndefined()
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a MinKey", vr.Type())
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(primitive.MinKey{}))
|
|
return nil
|
|
}
|
|
|
|
// MaxKeyDecodeValue is the ValueDecoderFunc for MaxKey.
|
|
func (DefaultValueDecoders) MaxKeyDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tMaxKey {
|
|
return ValueDecoderError{Name: "MaxKeyDecodeValue", Types: []reflect.Type{tMaxKey}, Received: val}
|
|
}
|
|
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.MaxKey:
|
|
err = vr.ReadMaxKey()
|
|
case bsontype.Null:
|
|
err = vr.ReadNull()
|
|
case bsontype.Undefined:
|
|
err = vr.ReadUndefined()
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a MaxKey", vr.Type())
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(primitive.MaxKey{}))
|
|
return nil
|
|
}
|
|
|
|
// Decimal128DecodeValue is the ValueDecoderFunc for primitive.Decimal128.
|
|
func (dvd DefaultValueDecoders) Decimal128DecodeValue(dctx DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tDecimal {
|
|
return ValueDecoderError{Name: "Decimal128DecodeValue", Types: []reflect.Type{tDecimal}, Received: val}
|
|
}
|
|
|
|
var d128 primitive.Decimal128
|
|
var err error
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.Decimal128:
|
|
d128, err = vr.ReadDecimal128()
|
|
case bsontype.Null:
|
|
err = vr.ReadNull()
|
|
case bsontype.Undefined:
|
|
err = vr.ReadUndefined()
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a primitive.Decimal128", vr.Type())
|
|
}
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(d128))
|
|
return err
|
|
}
|
|
|
|
// JSONNumberDecodeValue is the ValueDecoderFunc for json.Number.
|
|
func (dvd DefaultValueDecoders) JSONNumberDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tJSONNumber {
|
|
return ValueDecoderError{Name: "JSONNumberDecodeValue", Types: []reflect.Type{tJSONNumber}, Received: val}
|
|
}
|
|
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.Double:
|
|
f64, err := vr.ReadDouble()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(json.Number(strconv.FormatFloat(f64, 'f', -1, 64))))
|
|
case bsontype.Int32:
|
|
i32, err := vr.ReadInt32()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(json.Number(strconv.FormatInt(int64(i32), 10))))
|
|
case bsontype.Int64:
|
|
i64, err := vr.ReadInt64()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(json.Number(strconv.FormatInt(i64, 10))))
|
|
case bsontype.Null:
|
|
if err := vr.ReadNull(); err != nil {
|
|
return err
|
|
}
|
|
val.SetString("")
|
|
case bsontype.Undefined:
|
|
if err := vr.ReadUndefined(); err != nil {
|
|
return err
|
|
}
|
|
val.SetString("")
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a json.Number", vrType)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// URLDecodeValue is the ValueDecoderFunc for url.URL.
|
|
func (dvd DefaultValueDecoders) URLDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tURL {
|
|
return ValueDecoderError{Name: "URLDecodeValue", Types: []reflect.Type{tURL}, Received: val}
|
|
}
|
|
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.String:
|
|
str, err := vr.ReadString()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
parsedURL, err := url.Parse(str)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(parsedURL).Elem())
|
|
return nil
|
|
case bsontype.Null:
|
|
if err := vr.ReadNull(); err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(url.URL{}))
|
|
return nil
|
|
case bsontype.Undefined:
|
|
if err := vr.ReadUndefined(); err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(url.URL{}))
|
|
return nil
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a *url.URL", vrType)
|
|
}
|
|
}
|
|
|
|
// TimeDecodeValue is the ValueDecoderFunc for time.Time.
|
|
//
|
|
// Deprecated: TimeDecodeValue is not registered by default. Use TimeCodec.DecodeValue instead.
|
|
func (dvd DefaultValueDecoders) TimeDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if vr.Type() != bsontype.DateTime {
|
|
return fmt.Errorf("cannot decode %v into a time.Time", vr.Type())
|
|
}
|
|
|
|
dt, err := vr.ReadDateTime()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !val.CanSet() || val.Type() != tTime {
|
|
return ValueDecoderError{Name: "TimeDecodeValue", Types: []reflect.Type{tTime}, Received: val}
|
|
}
|
|
|
|
val.Set(reflect.ValueOf(time.Unix(dt/1000, dt%1000*1000000).UTC()))
|
|
return nil
|
|
}
|
|
|
|
// ByteSliceDecodeValue is the ValueDecoderFunc for []byte.
|
|
//
|
|
// Deprecated: ByteSliceDecodeValue is not registered by default. Use ByteSliceCodec.DecodeValue instead.
|
|
func (dvd DefaultValueDecoders) ByteSliceDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if vr.Type() != bsontype.Binary && vr.Type() != bsontype.Null {
|
|
return fmt.Errorf("cannot decode %v into a []byte", vr.Type())
|
|
}
|
|
|
|
if !val.CanSet() || val.Type() != tByteSlice {
|
|
return ValueDecoderError{Name: "ByteSliceDecodeValue", Types: []reflect.Type{tByteSlice}, Received: val}
|
|
}
|
|
|
|
if vr.Type() == bsontype.Null {
|
|
val.Set(reflect.Zero(val.Type()))
|
|
return vr.ReadNull()
|
|
}
|
|
|
|
data, subtype, err := vr.ReadBinary()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if subtype != 0x00 {
|
|
return fmt.Errorf("ByteSliceDecodeValue can only be used to decode subtype 0x00 for %s, got %v", bsontype.Binary, subtype)
|
|
}
|
|
|
|
val.Set(reflect.ValueOf(data))
|
|
return nil
|
|
}
|
|
|
|
// MapDecodeValue is the ValueDecoderFunc for map[string]* types.
|
|
//
|
|
// Deprecated: MapDecodeValue is not registered by default. Use MapCodec.DecodeValue instead.
|
|
func (dvd DefaultValueDecoders) MapDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Kind() != reflect.Map || val.Type().Key().Kind() != reflect.String {
|
|
return ValueDecoderError{Name: "MapDecodeValue", Kinds: []reflect.Kind{reflect.Map}, Received: val}
|
|
}
|
|
|
|
switch vr.Type() {
|
|
case bsontype.Type(0), bsontype.EmbeddedDocument:
|
|
case bsontype.Null:
|
|
val.Set(reflect.Zero(val.Type()))
|
|
return vr.ReadNull()
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a %s", vr.Type(), val.Type())
|
|
}
|
|
|
|
dr, err := vr.ReadDocument()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if val.IsNil() {
|
|
val.Set(reflect.MakeMap(val.Type()))
|
|
}
|
|
|
|
eType := val.Type().Elem()
|
|
decoder, err := dc.LookupDecoder(eType)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if eType == tEmpty {
|
|
dc.Ancestor = val.Type()
|
|
}
|
|
|
|
keyType := val.Type().Key()
|
|
for {
|
|
key, vr, err := dr.ReadElement()
|
|
if err == bsonrw.ErrEOD {
|
|
break
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
elem := reflect.New(eType).Elem()
|
|
|
|
err = decoder.DecodeValue(dc, vr, elem)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
val.SetMapIndex(reflect.ValueOf(key).Convert(keyType), elem)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// ArrayDecodeValue is the ValueDecoderFunc for array types.
|
|
func (dvd DefaultValueDecoders) ArrayDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.IsValid() || val.Kind() != reflect.Array {
|
|
return ValueDecoderError{Name: "ArrayDecodeValue", Kinds: []reflect.Kind{reflect.Array}, Received: val}
|
|
}
|
|
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.Array:
|
|
case bsontype.Type(0), bsontype.EmbeddedDocument:
|
|
if val.Type().Elem() != tE {
|
|
return fmt.Errorf("cannot decode document into %s", val.Type())
|
|
}
|
|
case bsontype.Binary:
|
|
if val.Type().Elem() != tByte {
|
|
return fmt.Errorf("ArrayDecodeValue can only be used to decode binary into a byte array, got %v", vrType)
|
|
}
|
|
data, subtype, err := vr.ReadBinary()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if subtype != bsontype.BinaryGeneric && subtype != bsontype.BinaryBinaryOld {
|
|
return fmt.Errorf("ArrayDecodeValue can only be used to decode subtype 0x00 or 0x02 for %s, got %v", bsontype.Binary, subtype)
|
|
}
|
|
|
|
if len(data) > val.Len() {
|
|
return fmt.Errorf("more elements returned in array than can fit inside %s", val.Type())
|
|
}
|
|
|
|
for idx, elem := range data {
|
|
val.Index(idx).Set(reflect.ValueOf(elem))
|
|
}
|
|
return nil
|
|
case bsontype.Null:
|
|
val.Set(reflect.Zero(val.Type()))
|
|
return vr.ReadNull()
|
|
case bsontype.Undefined:
|
|
val.Set(reflect.Zero(val.Type()))
|
|
return vr.ReadUndefined()
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into an array", vrType)
|
|
}
|
|
|
|
var elemsFunc func(DecodeContext, bsonrw.ValueReader, reflect.Value) ([]reflect.Value, error)
|
|
switch val.Type().Elem() {
|
|
case tE:
|
|
elemsFunc = dvd.decodeD
|
|
default:
|
|
elemsFunc = dvd.decodeDefault
|
|
}
|
|
|
|
elems, err := elemsFunc(dc, vr, val)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(elems) > val.Len() {
|
|
return fmt.Errorf("more elements returned in array than can fit inside %s, got %v elements", val.Type(), len(elems))
|
|
}
|
|
|
|
for idx, elem := range elems {
|
|
val.Index(idx).Set(elem)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// SliceDecodeValue is the ValueDecoderFunc for slice types.
|
|
//
|
|
// Deprecated: SliceDecodeValue is not registered by default. Use SliceCodec.DecodeValue instead.
|
|
func (dvd DefaultValueDecoders) SliceDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Kind() != reflect.Slice {
|
|
return ValueDecoderError{Name: "SliceDecodeValue", Kinds: []reflect.Kind{reflect.Slice}, Received: val}
|
|
}
|
|
|
|
switch vr.Type() {
|
|
case bsontype.Array:
|
|
case bsontype.Null:
|
|
val.Set(reflect.Zero(val.Type()))
|
|
return vr.ReadNull()
|
|
case bsontype.Type(0), bsontype.EmbeddedDocument:
|
|
if val.Type().Elem() != tE {
|
|
return fmt.Errorf("cannot decode document into %s", val.Type())
|
|
}
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a slice", vr.Type())
|
|
}
|
|
|
|
var elemsFunc func(DecodeContext, bsonrw.ValueReader, reflect.Value) ([]reflect.Value, error)
|
|
switch val.Type().Elem() {
|
|
case tE:
|
|
dc.Ancestor = val.Type()
|
|
elemsFunc = dvd.decodeD
|
|
default:
|
|
elemsFunc = dvd.decodeDefault
|
|
}
|
|
|
|
elems, err := elemsFunc(dc, vr, val)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if val.IsNil() {
|
|
val.Set(reflect.MakeSlice(val.Type(), 0, len(elems)))
|
|
}
|
|
|
|
val.SetLen(0)
|
|
val.Set(reflect.Append(val, elems...))
|
|
|
|
return nil
|
|
}
|
|
|
|
// ValueUnmarshalerDecodeValue is the ValueDecoderFunc for ValueUnmarshaler implementations.
|
|
func (dvd DefaultValueDecoders) ValueUnmarshalerDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.IsValid() || (!val.Type().Implements(tValueUnmarshaler) && !reflect.PtrTo(val.Type()).Implements(tValueUnmarshaler)) {
|
|
return ValueDecoderError{Name: "ValueUnmarshalerDecodeValue", Types: []reflect.Type{tValueUnmarshaler}, Received: val}
|
|
}
|
|
|
|
if val.Kind() == reflect.Ptr && val.IsNil() {
|
|
if !val.CanSet() {
|
|
return ValueDecoderError{Name: "ValueUnmarshalerDecodeValue", Types: []reflect.Type{tValueUnmarshaler}, Received: val}
|
|
}
|
|
val.Set(reflect.New(val.Type().Elem()))
|
|
}
|
|
|
|
if !val.Type().Implements(tValueUnmarshaler) {
|
|
if !val.CanAddr() {
|
|
return ValueDecoderError{Name: "ValueUnmarshalerDecodeValue", Types: []reflect.Type{tValueUnmarshaler}, Received: val}
|
|
}
|
|
val = val.Addr() // If they type doesn't implement the interface, a pointer to it must.
|
|
}
|
|
|
|
t, src, err := bsonrw.Copier{}.CopyValueToBytes(vr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fn := val.Convert(tValueUnmarshaler).MethodByName("UnmarshalBSONValue")
|
|
errVal := fn.Call([]reflect.Value{reflect.ValueOf(t), reflect.ValueOf(src)})[0]
|
|
if !errVal.IsNil() {
|
|
return errVal.Interface().(error)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// UnmarshalerDecodeValue is the ValueDecoderFunc for Unmarshaler implementations.
|
|
func (dvd DefaultValueDecoders) UnmarshalerDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.IsValid() || (!val.Type().Implements(tUnmarshaler) && !reflect.PtrTo(val.Type()).Implements(tUnmarshaler)) {
|
|
return ValueDecoderError{Name: "UnmarshalerDecodeValue", Types: []reflect.Type{tUnmarshaler}, Received: val}
|
|
}
|
|
|
|
if val.Kind() == reflect.Ptr && val.IsNil() {
|
|
if !val.CanSet() {
|
|
return ValueDecoderError{Name: "UnmarshalerDecodeValue", Types: []reflect.Type{tUnmarshaler}, Received: val}
|
|
}
|
|
val.Set(reflect.New(val.Type().Elem()))
|
|
}
|
|
|
|
if !val.Type().Implements(tUnmarshaler) {
|
|
if !val.CanAddr() {
|
|
return ValueDecoderError{Name: "UnmarshalerDecodeValue", Types: []reflect.Type{tUnmarshaler}, Received: val}
|
|
}
|
|
val = val.Addr() // If they type doesn't implement the interface, a pointer to it must.
|
|
}
|
|
|
|
_, src, err := bsonrw.Copier{}.CopyValueToBytes(vr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
fn := val.Convert(tUnmarshaler).MethodByName("UnmarshalBSON")
|
|
errVal := fn.Call([]reflect.Value{reflect.ValueOf(src)})[0]
|
|
if !errVal.IsNil() {
|
|
return errVal.Interface().(error)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// EmptyInterfaceDecodeValue is the ValueDecoderFunc for interface{}.
|
|
//
|
|
// Deprecated: EmptyInterfaceDecodeValue is not registered by default. Use EmptyInterfaceCodec.DecodeValue instead.
|
|
func (dvd DefaultValueDecoders) EmptyInterfaceDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tEmpty {
|
|
return ValueDecoderError{Name: "EmptyInterfaceDecodeValue", Types: []reflect.Type{tEmpty}, Received: val}
|
|
}
|
|
|
|
rtype, err := dc.LookupTypeMapEntry(vr.Type())
|
|
if err != nil {
|
|
switch vr.Type() {
|
|
case bsontype.EmbeddedDocument:
|
|
if dc.Ancestor != nil {
|
|
rtype = dc.Ancestor
|
|
break
|
|
}
|
|
rtype = tD
|
|
case bsontype.Null:
|
|
val.Set(reflect.Zero(val.Type()))
|
|
return vr.ReadNull()
|
|
default:
|
|
return err
|
|
}
|
|
}
|
|
|
|
decoder, err := dc.LookupDecoder(rtype)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
elem := reflect.New(rtype).Elem()
|
|
err = decoder.DecodeValue(dc, vr, elem)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
val.Set(elem)
|
|
return nil
|
|
}
|
|
|
|
// CoreDocumentDecodeValue is the ValueDecoderFunc for bsoncore.Document.
|
|
func (DefaultValueDecoders) CoreDocumentDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tCoreDocument {
|
|
return ValueDecoderError{Name: "CoreDocumentDecodeValue", Types: []reflect.Type{tCoreDocument}, Received: val}
|
|
}
|
|
|
|
if val.IsNil() {
|
|
val.Set(reflect.MakeSlice(val.Type(), 0, 0))
|
|
}
|
|
|
|
val.SetLen(0)
|
|
|
|
cdoc, err := bsonrw.Copier{}.AppendDocumentBytes(val.Interface().(bsoncore.Document), vr)
|
|
val.Set(reflect.ValueOf(cdoc))
|
|
return err
|
|
}
|
|
|
|
func (dvd DefaultValueDecoders) decodeDefault(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) ([]reflect.Value, error) {
|
|
elems := make([]reflect.Value, 0)
|
|
|
|
ar, err := vr.ReadArray()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
eType := val.Type().Elem()
|
|
|
|
decoder, err := dc.LookupDecoder(eType)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
idx := 0
|
|
for {
|
|
vr, err := ar.ReadValue()
|
|
if err == bsonrw.ErrEOA {
|
|
break
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
elem := reflect.New(eType).Elem()
|
|
|
|
err = decoder.DecodeValue(dc, vr, elem)
|
|
if err != nil {
|
|
return nil, newDecodeError(strconv.Itoa(idx), err)
|
|
}
|
|
elems = append(elems, elem)
|
|
idx++
|
|
}
|
|
|
|
return elems, nil
|
|
}
|
|
|
|
// CodeWithScopeDecodeValue is the ValueDecoderFunc for CodeWithScope.
|
|
func (dvd DefaultValueDecoders) CodeWithScopeDecodeValue(dc DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
|
if !val.CanSet() || val.Type() != tCodeWithScope {
|
|
return ValueDecoderError{Name: "CodeWithScopeDecodeValue", Types: []reflect.Type{tCodeWithScope}, Received: val}
|
|
}
|
|
|
|
switch vrType := vr.Type(); vrType {
|
|
case bsontype.CodeWithScope:
|
|
code, dr, err := vr.ReadCodeWithScope()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
scope := reflect.New(tD).Elem()
|
|
elems, err := dvd.decodeElemsFromDocumentReader(dc, dr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
scope.Set(reflect.MakeSlice(tD, 0, len(elems)))
|
|
scope.Set(reflect.Append(scope, elems...))
|
|
|
|
val.Set(reflect.ValueOf(primitive.CodeWithScope{
|
|
Code: primitive.JavaScript(code),
|
|
Scope: scope.Interface().(primitive.D),
|
|
}))
|
|
return nil
|
|
case bsontype.Null:
|
|
if err := vr.ReadNull(); err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(primitive.CodeWithScope{}))
|
|
return nil
|
|
case bsontype.Undefined:
|
|
if err := vr.ReadUndefined(); err != nil {
|
|
return err
|
|
}
|
|
val.Set(reflect.ValueOf(primitive.CodeWithScope{}))
|
|
return nil
|
|
default:
|
|
return fmt.Errorf("cannot decode %v into a primitive.CodeWithScope", vrType)
|
|
}
|
|
}
|
|
|
|
func (dvd DefaultValueDecoders) decodeD(dc DecodeContext, vr bsonrw.ValueReader, _ reflect.Value) ([]reflect.Value, error) {
|
|
switch vr.Type() {
|
|
case bsontype.Type(0), bsontype.EmbeddedDocument:
|
|
default:
|
|
return nil, fmt.Errorf("cannot decode %v into a D", vr.Type())
|
|
}
|
|
|
|
dr, err := vr.ReadDocument()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return dvd.decodeElemsFromDocumentReader(dc, dr)
|
|
}
|
|
|
|
func (DefaultValueDecoders) decodeElemsFromDocumentReader(dc DecodeContext, dr bsonrw.DocumentReader) ([]reflect.Value, error) {
|
|
decoder, err := dc.LookupDecoder(tEmpty)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
elems := make([]reflect.Value, 0)
|
|
for {
|
|
key, vr, err := dr.ReadElement()
|
|
if err == bsonrw.ErrEOD {
|
|
break
|
|
}
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
val := reflect.New(tEmpty).Elem()
|
|
err = decoder.DecodeValue(dc, vr, val)
|
|
if err != nil {
|
|
return nil, newDecodeError(key, err)
|
|
}
|
|
|
|
elems = append(elems, reflect.ValueOf(primitive.E{Key: key, Value: val.Interface()}))
|
|
}
|
|
|
|
return elems, nil
|
|
}
|