-
Notifications
You must be signed in to change notification settings - Fork 92
/
options.go
112 lines (89 loc) · 2.28 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
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
package jsonrpc
import (
"net/http"
"reflect"
"time"
"github.com/gorilla/websocket"
)
type ParamEncoder func(reflect.Value) (reflect.Value, error)
type clientHandler struct {
ns string
hnd interface{}
}
type Config struct {
reconnectBackoff backoff
pingInterval time.Duration
timeout time.Duration
paramEncoders map[reflect.Type]ParamEncoder
errors *Errors
reverseHandlers []clientHandler
aliasedHandlerMethods map[string]string
httpClient *http.Client
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,
aliasedHandlerMethods: map[string]string{},
paramEncoders: map[reflect.Type]ParamEncoder{},
httpClient: _defaultHTTPClient,
}
}
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
}
}
func WithClientHandler(ns string, hnd interface{}) func(c *Config) {
return func(c *Config) {
c.reverseHandlers = append(c.reverseHandlers, clientHandler{ns, hnd})
}
}
// WithClientHandlerAlias creates an alias for a client HANDLER method - for handlers created
// with WithClientHandler
func WithClientHandlerAlias(alias, original string) func(c *Config) {
return func(c *Config) {
c.aliasedHandlerMethods[alias] = original
}
}
func WithHTTPClient(h *http.Client) func(c *Config) {
return func(c *Config) {
c.httpClient = h
}
}