array_codec.go raw

   1  // Copyright (C) MongoDB, Inc. 2017-present.
   2  //
   3  // Licensed under the Apache License, Version 2.0 (the "License"); you may
   4  // not use this file except in compliance with the License. You may obtain
   5  // a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
   6  
   7  package bsoncodec
   8  
   9  import (
  10  	"reflect"
  11  
  12  	"go.mongodb.org/mongo-driver/bson/bsonrw"
  13  	"go.mongodb.org/mongo-driver/x/bsonx/bsoncore"
  14  )
  15  
  16  // ArrayCodec is the Codec used for bsoncore.Array values.
  17  //
  18  // Deprecated: Use [go.mongodb.org/mongo-driver/bson.NewRegistry] to get a registry with the
  19  // ArrayCodec registered.
  20  type ArrayCodec struct{}
  21  
  22  var defaultArrayCodec = NewArrayCodec()
  23  
  24  // NewArrayCodec returns an ArrayCodec.
  25  //
  26  // Deprecated: Use [go.mongodb.org/mongo-driver/bson.NewRegistry] to get a registry with the
  27  // ArrayCodec registered.
  28  func NewArrayCodec() *ArrayCodec {
  29  	return &ArrayCodec{}
  30  }
  31  
  32  // EncodeValue is the ValueEncoder for bsoncore.Array values.
  33  func (ac *ArrayCodec) EncodeValue(_ EncodeContext, vw bsonrw.ValueWriter, val reflect.Value) error {
  34  	if !val.IsValid() || val.Type() != tCoreArray {
  35  		return ValueEncoderError{Name: "CoreArrayEncodeValue", Types: []reflect.Type{tCoreArray}, Received: val}
  36  	}
  37  
  38  	arr := val.Interface().(bsoncore.Array)
  39  	return bsonrw.Copier{}.CopyArrayFromBytes(vw, arr)
  40  }
  41  
  42  // DecodeValue is the ValueDecoder for bsoncore.Array values.
  43  func (ac *ArrayCodec) DecodeValue(_ DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
  44  	if !val.CanSet() || val.Type() != tCoreArray {
  45  		return ValueDecoderError{Name: "CoreArrayDecodeValue", Types: []reflect.Type{tCoreArray}, Received: val}
  46  	}
  47  
  48  	if val.IsNil() {
  49  		val.Set(reflect.MakeSlice(val.Type(), 0, 0))
  50  	}
  51  
  52  	val.SetLen(0)
  53  	arr, err := bsonrw.Copier{}.AppendArrayBytes(val.Interface().(bsoncore.Array), vr)
  54  	val.Set(reflect.ValueOf(arr))
  55  	return err
  56  }
  57