-
Notifications
You must be signed in to change notification settings - Fork 883
/
server_options.go
184 lines (160 loc) · 5.65 KB
/
server_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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
// 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 topology
import (
"time"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/bsoncodec"
"go.mongodb.org/mongo-driver/event"
"go.mongodb.org/mongo-driver/x/mongo/driver"
"go.mongodb.org/mongo-driver/x/mongo/driver/session"
)
var defaultRegistry = bson.NewRegistryBuilder().Build()
type serverConfig struct {
clock *session.ClusterClock
compressionOpts []string
connectionOpts []ConnectionOption
appname string
heartbeatInterval time.Duration
heartbeatTimeout time.Duration
maxConns uint64
minConns uint64
poolMonitor *event.PoolMonitor
serverMonitor *event.ServerMonitor
connectionPoolMaxIdleTime time.Duration
registry *bsoncodec.Registry
monitoringDisabled bool
serverAPI *driver.ServerAPIOptions
loadBalanced bool
}
func newServerConfig(opts ...ServerOption) (*serverConfig, error) {
cfg := &serverConfig{
heartbeatInterval: 10 * time.Second,
heartbeatTimeout: 10 * time.Second,
maxConns: 100,
registry: defaultRegistry,
}
for _, opt := range opts {
err := opt(cfg)
if err != nil {
return nil, err
}
}
return cfg, nil
}
// ServerOption configures a server.
type ServerOption func(*serverConfig) error
func withMonitoringDisabled(fn func(bool) bool) ServerOption {
return func(cfg *serverConfig) error {
cfg.monitoringDisabled = fn(cfg.monitoringDisabled)
return nil
}
}
// WithConnectionOptions configures the server's connections.
func WithConnectionOptions(fn func(...ConnectionOption) []ConnectionOption) ServerOption {
return func(cfg *serverConfig) error {
cfg.connectionOpts = fn(cfg.connectionOpts...)
return nil
}
}
// WithCompressionOptions configures the server's compressors.
func WithCompressionOptions(fn func(...string) []string) ServerOption {
return func(cfg *serverConfig) error {
cfg.compressionOpts = fn(cfg.compressionOpts...)
return nil
}
}
// WithServerAppName configures the server's application name.
func WithServerAppName(fn func(string) string) ServerOption {
return func(cfg *serverConfig) error {
cfg.appname = fn(cfg.appname)
return nil
}
}
// WithHeartbeatInterval configures a server's heartbeat interval.
func WithHeartbeatInterval(fn func(time.Duration) time.Duration) ServerOption {
return func(cfg *serverConfig) error {
cfg.heartbeatInterval = fn(cfg.heartbeatInterval)
return nil
}
}
// WithHeartbeatTimeout configures how long to wait for a heartbeat socket to
// connection.
func WithHeartbeatTimeout(fn func(time.Duration) time.Duration) ServerOption {
return func(cfg *serverConfig) error {
cfg.heartbeatTimeout = fn(cfg.heartbeatTimeout)
return nil
}
}
// WithMaxConnections configures the maximum number of connections to allow for
// a given server. If max is 0, then the default will be math.MaxInt64.
func WithMaxConnections(fn func(uint64) uint64) ServerOption {
return func(cfg *serverConfig) error {
cfg.maxConns = fn(cfg.maxConns)
return nil
}
}
// WithMinConnections configures the minimum number of connections to allow for
// a given server. If min is 0, then there is no lower limit to the number of
// connections.
func WithMinConnections(fn func(uint64) uint64) ServerOption {
return func(cfg *serverConfig) error {
cfg.minConns = fn(cfg.minConns)
return nil
}
}
// WithConnectionPoolMaxIdleTime configures the maximum time that a connection can remain idle in the connection pool
// before being removed. If connectionPoolMaxIdleTime is 0, then no idle time is set and connections will not be removed
// because of their age
func WithConnectionPoolMaxIdleTime(fn func(time.Duration) time.Duration) ServerOption {
return func(cfg *serverConfig) error {
cfg.connectionPoolMaxIdleTime = fn(cfg.connectionPoolMaxIdleTime)
return nil
}
}
// WithConnectionPoolMonitor configures the monitor for all connection pool actions
func WithConnectionPoolMonitor(fn func(*event.PoolMonitor) *event.PoolMonitor) ServerOption {
return func(cfg *serverConfig) error {
cfg.poolMonitor = fn(cfg.poolMonitor)
return nil
}
}
// WithServerMonitor configures the monitor for all SDAM events for a server
func WithServerMonitor(fn func(*event.ServerMonitor) *event.ServerMonitor) ServerOption {
return func(cfg *serverConfig) error {
cfg.serverMonitor = fn(cfg.serverMonitor)
return nil
}
}
// WithClock configures the ClusterClock for the server to use.
func WithClock(fn func(clock *session.ClusterClock) *session.ClusterClock) ServerOption {
return func(cfg *serverConfig) error {
cfg.clock = fn(cfg.clock)
return nil
}
}
// WithRegistry configures the registry for the server to use when creating
// cursors.
func WithRegistry(fn func(*bsoncodec.Registry) *bsoncodec.Registry) ServerOption {
return func(cfg *serverConfig) error {
cfg.registry = fn(cfg.registry)
return nil
}
}
// WithServerAPI configures the server API options for the server to use.
func WithServerAPI(fn func(serverAPI *driver.ServerAPIOptions) *driver.ServerAPIOptions) ServerOption {
return func(cfg *serverConfig) error {
cfg.serverAPI = fn(cfg.serverAPI)
return nil
}
}
// WithServerLoadBalanced specifies whether or not the server is behind a load balancer.
func WithServerLoadBalanced(fn func(bool) bool) ServerOption {
return func(cfg *serverConfig) error {
cfg.loadBalanced = fn(cfg.loadBalanced)
return nil
}
}