-
Notifications
You must be signed in to change notification settings - Fork 883
/
session_options.go
111 lines (100 loc) · 3.54 KB
/
session_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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
// 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"
"time"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo/options"
)
// transactionOptions is a wrapper for *options.transactionOptions. This type implements the bson.Unmarshaler interface
// to convert BSON documents to a transactionOptions instance.
type transactionOptions struct {
*options.TransactionOptions
}
var _ bson.Unmarshaler = (*transactionOptions)(nil)
func (to *transactionOptions) UnmarshalBSON(data []byte) error {
var temp struct {
RC *readConcern `bson:"readConcern"`
RP *ReadPreference `bson:"readPreference"`
WC *writeConcern `bson:"writeConcern"`
MaxCommitTimeMS *int64 `bson:"maxCommitTimeMS"`
Extra map[string]interface{} `bson:",inline"`
}
if err := bson.Unmarshal(data, &temp); err != nil {
return fmt.Errorf("error unmarshalling to temporary transactionOptions object: %v", err)
}
if len(temp.Extra) > 0 {
return fmt.Errorf("unrecognized fields for transactionOptions: %v", mapKeys(temp.Extra))
}
to.TransactionOptions = options.Transaction()
if temp.MaxCommitTimeMS != nil {
mctms := time.Duration(*temp.MaxCommitTimeMS) * time.Millisecond
to.SetMaxCommitTime(&mctms)
}
if rc := temp.RC; rc != nil {
to.SetReadConcern(rc.toReadConcernOption())
}
if rp := temp.RP; rp != nil {
converted, err := rp.ToReadPrefOption()
if err != nil {
return fmt.Errorf("error parsing read preference document: %v", err)
}
to.SetReadPreference(converted)
}
if wc := temp.WC; wc != nil {
converted, err := wc.toWriteConcernOption()
if err != nil {
return fmt.Errorf("error parsing write concern document: %v", err)
}
to.SetWriteConcern(converted)
}
return nil
}
// sessionOptions is a wrapper for *options.sessionOptions. This type implements the bson.Unmarshaler interface to
// convert BSON documents to a sessionOptions instance.
type sessionOptions struct {
*options.SessionOptions
}
var _ bson.Unmarshaler = (*sessionOptions)(nil)
func (so *sessionOptions) UnmarshalBSON(data []byte) error {
var temp struct {
Causal *bool `bson:"causalConsistency"`
MaxCommitTimeMS *int64 `bson:"maxCommitTimeMS"`
TxnOptions *transactionOptions `bson:"defaultTransactionOptions"`
Snapshot *bool `bson:"snapshot"`
Extra map[string]interface{} `bson:",inline"`
}
if err := bson.Unmarshal(data, &temp); err != nil {
return fmt.Errorf("error unmarshalling to temporary sessionOptions object: %v", err)
}
if len(temp.Extra) > 0 {
return fmt.Errorf("unrecognized fields for sessionOptions: %v", mapKeys(temp.Extra))
}
so.SessionOptions = options.Session()
if temp.Causal != nil {
so.SetCausalConsistency(*temp.Causal)
}
if temp.MaxCommitTimeMS != nil {
mctms := time.Duration(*temp.MaxCommitTimeMS) * time.Millisecond
so.SetDefaultMaxCommitTime(&mctms)
}
if temp.TxnOptions != nil {
if rc := temp.TxnOptions.ReadConcern; rc != nil {
so.SetDefaultReadConcern(rc)
}
if rp := temp.TxnOptions.ReadPreference; rp != nil {
so.SetDefaultReadPreference(rp)
}
if wc := temp.TxnOptions.WriteConcern; wc != nil {
so.SetDefaultWriteConcern(wc)
}
}
if temp.Snapshot != nil {
so.SetSnapshot(*temp.Snapshot)
}
return nil
}