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
|
// 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/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
connectionPoolMaxIdleTime time.Duration
registry *bsoncodec.Registry
}
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
// 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
}
}
// 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 100
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
}
}
// 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
}
}
|
