forked from mongodb/mongo-go-driver
/
db_collection_options.go
66 lines (56 loc) · 1.96 KB
/
db_collection_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
66
// 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"
)
type dbOrCollectionOptions struct {
DBOptions *options.DatabaseOptions
CollectionOptions *options.CollectionOptions
}
var _ bson.Unmarshaler = (*dbOrCollectionOptions)(nil)
// UnmarshalBSON specifies custom BSON unmarshalling behavior to convert db/collection options from BSON/JSON documents
// to their corresponding Go objects.
func (d *dbOrCollectionOptions) UnmarshalBSON(data []byte) error {
var temp struct {
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 dbOrCollectionOptions object: %v", err)
}
if len(temp.Extra) > 0 {
return fmt.Errorf("unrecognized fields for dbOrCollectionOptions: %v", mapKeys(temp.Extra))
}
d.DBOptions = options.Database()
d.CollectionOptions = options.Collection()
if temp.RC != nil {
rc := temp.RC.toReadConcernOption()
d.DBOptions.SetReadConcern(rc)
d.CollectionOptions.SetReadConcern(rc)
}
if temp.RP != nil {
rp, err := temp.RP.ToReadPrefOption()
if err != nil {
return fmt.Errorf("error parsing read preference document: %v", err)
}
d.DBOptions.SetReadPreference(rp)
d.CollectionOptions.SetReadPreference(rp)
}
if temp.WC != nil {
wc, err := temp.WC.toWriteConcernOption()
if err != nil {
return fmt.Errorf("error parsing write concern document: %v", err)
}
d.DBOptions.SetWriteConcern(wc)
d.CollectionOptions.SetWriteConcern(wc)
}
return nil
}