-
Notifications
You must be signed in to change notification settings - Fork 883
/
bucket_options.go
65 lines (58 loc) · 1.94 KB
/
bucket_options.go
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: %w", 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: %w", err)
}
bo.SetReadPreference(rp)
}
if temp.WC != nil {
wc, err := temp.WC.toWriteConcernOption()
if err != nil {
return fmt.Errorf("error parsing write concern document: %w", err)
}
bo.SetWriteConcern(wc)
}
return nil
}