Rename JSONRPCServer to JSONRPCRouter

Author: tutman96

internal/jsonrpc/rpc_server.go renamed to internal/jsonrpc/router.go

@@ -12,7 +12,7 @@ import (
 	"time"
 )
 
-type JSONRPCServer struct {
+type JSONRPCRouter struct {
 	writer io.Writer
 
 	handlers map[string]*RPCHandler
@@ -22,16 +22,14 @@ type JSONRPCServer struct {
 	responseChannels      map[int64]chan JSONRPCResponse
 }
 
-func NewJSONRPCServer(writer io.Writer, handlers map[string]*RPCHandler) *JSONRPCServer {
-	return &JSONRPCServer{
-		writer:           writer,
-		handlers:         handlers,
-		responseChannels: make(map[int64]chan JSONRPCResponse),
-		nextId:           atomic.Int64{},
+func NewJSONRPCRouter(writer io.Writer, handlers map[string]*RPCHandler) *JSONRPCRouter {
+	return &JSONRPCRouter{
+		writer:   writer,
+		handlers: handlers,
 	}
 }
 
-func (s *JSONRPCServer) Request(method string, params map[string]interface{}, result interface{}) *JSONRPCResponseError {
+func (s *JSONRPCRouter) Request(method string, params map[string]interface{}, result interface{}) *JSONRPCResponseError {
 	id := s.nextId.Add(1)
 	request := JSONRPCRequest{
 		JSONRPC: "2.0",
@@ -108,7 +106,7 @@ type JSONRPCMessage struct {
 	ID     *int64  `json:"id,omitempty"`
 }
 
-func (s *JSONRPCServer) HandleMessage(data []byte) error {
+func (s *JSONRPCRouter) HandleMessage(data []byte) error {
 	// Data will either be a JSONRPCRequest or JSONRPCResponse object
 	// We need to determine which one it is
 	var raw JSONRPCMessage
@@ -193,7 +191,7 @@ func (s *JSONRPCServer) HandleMessage(data []byte) error {
 	return s.writeResponse(response)
 }
 
-func (s *JSONRPCServer) writeResponse(response JSONRPCResponse) error {
+func (s *JSONRPCRouter) writeResponse(response JSONRPCResponse) error {
 	responseBytes, err := json.Marshal(response)
 	if err != nil {
 		return err

jsonrpc.go

@@ -40,8 +40,8 @@ func (w *DataChannelWriter) Write(data []byte) (int, error) {
 	return len(data), nil
 }
 
-func NewDataChannelJsonRpcServer(dataChannel *webrtc.DataChannel) *jsonrpc.JSONRPCServer {
-	return jsonrpc.NewJSONRPCServer(
+func NewDataChannelJsonRpcRouter(dataChannel *webrtc.DataChannel) *jsonrpc.JSONRPCRouter {
+	return jsonrpc.NewJSONRPCRouter(
 		NewDataChannelWriter(dataChannel),
 		rpcHandlers,
 	)

webrtc.go

@@ -102,14 +102,7 @@ func newSession(config SessionConfig) (*Session, error) {
 		switch d.Label() {
 		case "rpc":
 			session.RPCChannel = d
-			fmt.Println("starting rpc server")
-			rpcServer := NewDataChannelJsonRpcServer(d)
-			d.OnError(func(err error) {
-				fmt.Println("rpc error", err)
-			})
-			d.OnClose(func() {
-				fmt.Println("rpc closed")
-			})
+			rpcServer := NewDataChannelJsonRpcRouter(d)
 			d.OnMessage(func(msg webrtc.DataChannelMessage) {
 				go rpcServer.HandleMessage(msg.Data)
 			})