forked from filecoin-project/go-jsonrpc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
options.go
77 lines (62 loc) · 1.49 KB
/
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
package jsonrpc
import (
"reflect"
"time"
"github.com/gorilla/websocket"
)
type ParamEncoder func(reflect.Value) (reflect.Value, error)
type Config struct {
reconnectBackoff backoff
pingInterval time.Duration
timeout time.Duration
paramEncoders map[reflect.Type]ParamEncoder
errors *Errors
noReconnect bool
proxyConnFactory func(func() (*websocket.Conn, error)) func() (*websocket.Conn, error) // for testing
}
func defaultConfig() Config {
return Config{
reconnectBackoff: backoff{
minDelay: 100 * time.Millisecond,
maxDelay: 5 * time.Second,
},
pingInterval: 5 * time.Second,
timeout: 30 * time.Second,
paramEncoders: map[reflect.Type]ParamEncoder{},
}
}
type Option func(c *Config)
func WithReconnectBackoff(minDelay, maxDelay time.Duration) func(c *Config) {
return func(c *Config) {
c.reconnectBackoff = backoff{
minDelay: minDelay,
maxDelay: maxDelay,
}
}
}
// Must be < Timeout/2
func WithPingInterval(d time.Duration) func(c *Config) {
return func(c *Config) {
c.pingInterval = d
}
}
func WithTimeout(d time.Duration) func(c *Config) {
return func(c *Config) {
c.timeout = d
}
}
func WithNoReconnect() func(c *Config) {
return func(c *Config) {
c.noReconnect = true
}
}
func WithParamEncoder(t interface{}, encoder ParamEncoder) func(c *Config) {
return func(c *Config) {
c.paramEncoders[reflect.TypeOf(t).Elem()] = encoder
}
}
func WithErrors(es Errors) func(c *Config) {
return func(c *Config) {
c.errors = &es
}
}