1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
|
// 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 unified
import (
"fmt"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo/options"
)
// gridFSBucketOptions is a wrapper for *options.BucketOptions. This type implements the bson.Unmarshaler interface to
// convert BSON documents to a BucketOptions instance.
type gridFSBucketOptions struct {
*options.BucketOptions
}
var _ bson.Unmarshaler = (*gridFSBucketOptions)(nil)
func (bo *gridFSBucketOptions) UnmarshalBSON(data []byte) error {
var temp struct {
Name *string `bson:"name"`
ChunkSize *int32 `bson:"chunkSizeBytes"`
RC *readConcern `bson:"readConcern"`
RP *ReadPreference `bson:"readPreference"`
WC *writeConcern `bson:"writeConcern"`
Extra map[string]interface{} `bson:",inline"`
}
if err := bson.Unmarshal(data, &temp); err != nil {
return fmt.Errorf("error unmarshalling to temporary gridFSBucketOptions object: %v", err)
}
if len(temp.Extra) > 0 {
return fmt.Errorf("unrecognized fields for gridFSBucketOptions: %v", mapKeys(temp.Extra))
}
bo.BucketOptions = options.GridFSBucket()
if temp.Name != nil {
bo.SetName(*temp.Name)
}
if temp.ChunkSize != nil {
bo.SetChunkSizeBytes(*temp.ChunkSize)
}
if temp.RC != nil {
bo.SetReadConcern(temp.RC.toReadConcernOption())
}
if temp.RP != nil {
rp, err := temp.RP.ToReadPrefOption()
if err != nil {
return fmt.Errorf("error parsing read preference document: %v", err)
}
bo.SetReadPreference(rp)
}
if temp.WC != nil {
wc, err := temp.WC.toWriteConcernOption()
if err != nil {
return fmt.Errorf("error parsing write concern document: %v", err)
}
bo.SetWriteConcern(wc)
}
return nil
}
|