95 lines
2.5 KiB
Go
95 lines
2.5 KiB
Go
|
// 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 (
|
||
|
|
"fmt"
|
||
|
|
"reflect"
|
||
|
|
|
||
|
|
"go.mongodb.org/mongo-driver/bson/bsonoptions"
|
||
|
|
"go.mongodb.org/mongo-driver/bson/bsonrw"
|
||
|
|
"go.mongodb.org/mongo-driver/bson/bsontype"
|
||
|
|
)
|
||
|
|
|
||
|
|
var defaultStringCodec = NewStringCodec()
|
||
|
|
|
||
|
|
// StringCodec is the Codec used for struct values.
|
||
|
|
type StringCodec struct {
|
||
|
|
DecodeObjectIDAsHex bool
|
||
|
|
}
|
||
|
|
|
||
|
|
var _ ValueCodec = &StringCodec{}
|
||
|
|
|
||
|
|
// NewStringCodec returns a StringCodec with options opts.
|
||
|
|
func NewStringCodec(opts ...*bsonoptions.StringCodecOptions) *StringCodec {
|
||
|
|
stringOpt := bsonoptions.MergeStringCodecOptions(opts...)
|
||
|
|
return &StringCodec{*stringOpt.DecodeObjectIDAsHex}
|
||
|
|
}
|
||
|
|
|
||
|
|
// EncodeValue is the ValueEncoder for string types.
|
||
|
|
func (sc *StringCodec) EncodeValue(ectx EncodeContext, vw bsonrw.ValueWriter, val reflect.Value) error {
|
||
|
|
if val.Kind() != reflect.String {
|
||
|
|
return ValueEncoderError{
|
||
|
|
Name: "StringEncodeValue",
|
||
|
|
Kinds: []reflect.Kind{reflect.String},
|
||
|
|
Received: val,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return vw.WriteString(val.String())
|
||
|
|
}
|
||
|
|
|
||
|
|
// DecodeValue is the ValueDecoder for string types.
|
||
|
|
func (sc *StringCodec) DecodeValue(dctx DecodeContext, vr bsonrw.ValueReader, val reflect.Value) error {
|
||
|
|
if !val.CanSet() || val.Kind() != reflect.String {
|
||
|
|
return ValueDecoderError{Name: "StringDecodeValue", Kinds: []reflect.Kind{reflect.String}, Received: val}
|
||
|
|
}
|
||
|
|
var str string
|
||
|
|
var err error
|
||
|
|
switch vr.Type() {
|
||
|
|
case bsontype.String:
|
||
|
|
str, err = vr.ReadString()
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
case bsontype.ObjectID:
|
||
|
|
oid, err := vr.ReadObjectID()
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if sc.DecodeObjectIDAsHex {
|
||
|
|
str = oid.Hex()
|
||
|
|
} else {
|
||
|
|
byteArray := [12]byte(oid)
|
||
|
|
str = string(byteArray[:])
|
||
|
|
}
|
||
|
|
case bsontype.Symbol:
|
||
|
|
str, 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("SliceDecodeValue can only be used to decode subtype 0x00 or 0x02 for %s, got %v", bsontype.Binary, subtype)
|
||
|
|
}
|
||
|
|
str = string(data)
|
||
|
|
case bsontype.Null:
|
||
|
|
if err = vr.ReadNull(); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
default:
|
||
|
|
return fmt.Errorf("cannot decode %v into a string type", vr.Type())
|
||
|
|
}
|
||
|
|
|
||
|
|
val.SetString(str)
|
||
|
|
return nil
|
||
|
|
}
|