Add Request Timeout, Cleanup Request map

rpc/call.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/json"
 	"fmt"
+	"time"
 
 	"github.com/google/uuid"
 	"nhooyr.io/websocket"
@@ -57,9 +58,14 @@ func (s *server) Call(ctx context.Context, c *websocket.Conn, method string, par
 		return nil, fmt.Errorf("Error Writing Request: %w", err)
 	}
 
-	// Wait for Response, TODO add Select Timeout
+	timeout := time.NewTimer(time.Second * 10)
-	response := <-resp
 
+	select {
+	case response := <-resp:
+		// Cleanup timeout
+		if !timeout.Stop() {
+			<-timeout.C
+		}
 		if response.Error != nil {
 			return response, fmt.Errorf("Call Error: %w", err)
 		}
@@ -77,6 +83,14 @@ func (s *server) Call(ctx context.Context, c *websocket.Conn, method string, par
 			return response, fmt.Errorf("Error Parsing Result: %w", err)
 		}
 		return response, nil
+	case <-timeout.C:
+		s.requestMutex.Lock()
+		defer s.requestMutex.Unlock()
+
+		// remove request from map
+		delete(s.requests, id)
+		return nil, fmt.Errorf("Request timed out")
+	}
 }
 
 // TODO Call with Multiple Response (Chunked file upload)

rpc/response.go

@@ -22,6 +22,9 @@ func (s *server) handleResponse(ctx context.Context, data []byte) {
 		return
 	}
 
+	// remove request from map
+	delete(s.requests, response.ID)
+
 	// Send Response to Original Caller
 	r <- &response
 }