Send pings from ws server

Author: magik6k

options_server.go

@@ -3,14 +3,17 @@ package jsonrpc
 import (
 	"context"
 	"reflect"
+	"time"
 )
 
 type ParamDecoder func(ctx context.Context, json []byte) (reflect.Value, error)
 
 type ServerConfig struct {
-	paramDecoders  map[reflect.Type]ParamDecoder
 	maxRequestSize int64
-	errors         *Errors
+	pingInterval   time.Duration
+
+	paramDecoders map[reflect.Type]ParamDecoder
+	errors        *Errors
 }
 
 type ServerOption func(c *ServerConfig)
@@ -19,6 +22,8 @@ func defaultServerConfig() ServerConfig {
 	return ServerConfig{
 		paramDecoders:  map[reflect.Type]ParamDecoder{},
 		maxRequestSize: DEFAULT_MAX_REQUEST_SIZE,
+
+		pingInterval: 5 * time.Second,
 	}
 }
 
@@ -39,3 +44,9 @@ func WithServerErrors(es Errors) ServerOption {
 		c.errors = &es
 	}
 }
+
+func WithServerPingInterval(d time.Duration) ServerOption {
+	return func(c *ServerConfig) {
+		c.pingInterval = d
+	}
+}

server.go

@@ -7,6 +7,7 @@ import (
 	"net/http"
 	"reflect"
 	"strings"
+	"time"
 
 	"github.com/gorilla/websocket"
 )
@@ -28,6 +29,7 @@ type RPCServer struct {
 
 	paramDecoders map[reflect.Type]ParamDecoder
 
+	pingInterval   time.Duration
 	maxRequestSize int64
 }
 
@@ -44,6 +46,8 @@ func NewServer(opts ...ServerOption) *RPCServer {
 		paramDecoders:  config.paramDecoders,
 		maxRequestSize: config.maxRequestSize,
 		errors:         config.errors,
+
+		pingInterval: config.pingInterval,
 	}
 }
 
@@ -69,9 +73,10 @@ func (s *RPCServer) handleWS(ctx context.Context, w http.ResponseWriter, r *http
 	}
 
 	(&wsConn{
-		conn:    c,
-		handler: s,
-		exiting: make(chan struct{}),
+		conn:         c,
+		handler:      s,
+		pingInterval: s.pingInterval,
+		exiting:      make(chan struct{}),
 	}).handleWsConn(ctx)
 
 	if err := c.Close(); err != nil {