Fix HTTP transport message handling (#76)

Co-authored-by: Pavel Anni <[email protected]>

Author: pavelanni

transport/http/http.go

@@ -20,7 +20,6 @@ type HTTPTransport struct {
 	closeHandler   func()
 	mu             sync.RWMutex
 	addr           string
-	responseMap    map[int64]chan *transport.BaseJsonRpcMessage
 }
 
 // NewHTTPTransport creates a new HTTP transport that listens on the specified endpoint
@@ -29,7 +28,6 @@ func NewHTTPTransport(endpoint string) *HTTPTransport {
 		baseTransport: newBaseTransport(),
 		endpoint:      endpoint,
 		addr:          ":8080", // Default port
-		responseMap:   make(map[int64]chan *transport.BaseJsonRpcMessage),
 	}
 }
 
@@ -55,14 +53,27 @@ func (t *HTTPTransport) Start(ctx context.Context) error {
 // Send implements Transport.Send
 func (t *HTTPTransport) Send(ctx context.Context, message *transport.BaseJsonRpcMessage) error {
 	key := message.JsonRpcResponse.Id
-	responseChannel := t.responseMap[int64(key)]
+	fmt.Printf("[Send] Attempting to send response with key: %d\n", key)
+
+	responseChannel := t.baseTransport.responseMap[int64(key)]
 	if responseChannel == nil {
+		fmt.Printf("[Send] Response map keys: %v\n", t.getResponseMapKeys())
+
 		return fmt.Errorf("no response channel found for key: %d", key)
 	}
 	responseChannel <- message
 	return nil
 }
 
+// Helper method to get keys
+func (t *HTTPTransport) getResponseMapKeys() []int64 {
+	keys := make([]int64, 0, len(t.baseTransport.responseMap))
+	for k := range t.baseTransport.responseMap {
+		keys = append(keys, k)
+	}
+	return keys
+}
+
 // Close implements Transport.Close
 func (t *HTTPTransport) Close() error {
 	if t.server != nil {
@@ -94,6 +105,7 @@ func (t *HTTPTransport) SetErrorHandler(handler func(error)) {
 func (t *HTTPTransport) SetMessageHandler(handler func(ctx context.Context, message *transport.BaseJsonRpcMessage)) {
 	t.mu.Lock()
 	defer t.mu.Unlock()
+	t.baseTransport.messageHandler = handler
 	t.messageHandler = handler
 }