昔念
83ecb90baf
- 更新 README.md 中的项目结构说明 - 添加 pprof 性能分析工具的使用说明 - 更新 build.bat 文件，增加 proto 文件编译命令 - 升级 go-logr/logr 依赖至 v1.3.0 - 降级 go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracegrpc 依赖至 v1.16.0 - 降级 go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp 依赖至 v1.16.0 - 升级 go.opentelemetry.io/otel/trace 依赖至 v1.20.0 - 移除 logic/main.go 中的冗余代码 - 重构 logic/server.go 中的 Start 函数 - 更新 login/main.go 文件
123 lines
2.5 KiB
Go
123 lines
2.5 KiB
Go
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
|
|
|
|
methodNamer MethodNameFormatter
|
|
}
|
|
|
|
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,
|
|
|
|
methodNamer: DefaultMethodNameFormatter,
|
|
}
|
|
}
|
|
|
|
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
|
|
}
|
|
}
|
|
|
|
func WithMethodNameFormatter(namer MethodNameFormatter) func(c *Config) {
|
|
return func(c *Config) {
|
|
c.methodNamer = namer
|
|
}
|
|
}
|