|
1 | 1 | #!/usr/bin/env node |
2 | | -// Tool dispatch |
| 2 | +// JSON-RPC stdio transport |
| 3 | +// =================================================================== |
| 4 | + |
| 5 | +let buffer = ""; |
| 6 | +const MAX_BUFFER_BYTES = 2 * 1_048_576; // 2 MB |
| 7 | +======= |
| 8 | +// =================================================================== |
| 9 | +// JSON-RPC stdio transport |
| 10 | +// =================================================================== |
| 11 | + |
| 12 | +let buffer = ""; |
| 13 | +const MAX_BUFFER_BYTES = 2 * 1_048_576; // 2 MB |
| 14 | + |
| 15 | +// Auto-reconnect configuration |
| 16 | +const MAX_RECONNECT_ATTEMPTS = 5; |
| 17 | +const INITIAL_RECONNECT_DELAY = 1000; // 1 second |
| 18 | +let reconnectAttempts = 0; |
| 19 | +let reconnectTimeout = null; |
| 20 | + |
| 21 | +// =================================================================== |
| 22 | +// Auto-reconnect helper |
| 23 | +// =================================================================== |
| 24 | + |
| 25 | +/** |
| 26 | + * Calculate the next reconnect delay with exponential backoff and jitter. |
| 27 | + * @param {number} attempt |
| 28 | + * @returns {number} |
| 29 | + */ |
| 30 | +function getReconnectDelay(attempt) { |
| 31 | + const delay = INITIAL_RECONNECT_DELAY * Math.pow(2, attempt); |
| 32 | + const jitter = delay * 0.2; // 20% jitter |
| 33 | + return delay + Math.random() * jitter; |
| 34 | +} |
| 35 | + |
| 36 | +/** |
| 37 | + * Attempt to reconnect to the stdio transport. |
| 38 | + */ |
| 39 | +function attemptReconnect() { |
| 40 | + if (reconnectAttempts >= MAX_RECONNECT_ATTEMPTS) { |
| 41 | + logError("Max reconnect attempts reached. Giving up."); |
| 42 | + process.exit(1); |
| 43 | + } |
| 44 | + |
| 45 | + const delay = getReconnectDelay(reconnectAttempts); |
| 46 | + logError(`Attempting to reconnect in ${delay}ms... (attempt ${reconnectAttempts + 1} of ${MAX_RECONNECT_ATTEMPTS})`); |
| 47 | + |
| 48 | + reconnectTimeout = setTimeout(() => { |
| 49 | + reconnectAttempts++; |
| 50 | + // In a real stdio transport, you would re-establish the connection here. |
| 51 | + // For now, we'll just log the attempt. |
| 52 | + logError(`Reconnect attempt ${reconnectAttempts}`); |
| 53 | + attemptReconnect(); |
| 54 | + }, delay); |
| 55 | +} |
| 56 | + |
| 57 | +// =================================================================== |
| 58 | +// Error handling |
| 59 | +// =================================================================== |
| 60 | + |
| 61 | +/** |
| 62 | + * Handle errors and attempt to reconnect if necessary. |
| 63 | + * @param {Error} error |
| 64 | + */ |
| 65 | +function handleError(error) { |
| 66 | + logError("Error in MCP bridge:", error); |
| 67 | + if (reconnectTimeout) { |
| 68 | + clearTimeout(reconnectTimeout); |
| 69 | + } |
| 70 | + reconnectAttempts = 0; |
| 71 | + attemptReconnect(); |
| 72 | +} |
| 73 | + |
| 74 | +// Listen for unhandled rejections and errors |
| 75 | +process.on("unhandledRejection", (reason, promise) => { |
| 76 | + logError("Unhandled Rejection at:", promise, "reason:", reason); |
| 77 | + handleError(reason); |
| 78 | +}); |
| 79 | + |
| 80 | +process.on("uncaughtException", (error) => { |
| 81 | + logError("Uncaught Exception:", error); |
| 82 | + handleError(error); |
| 83 | +});Tool dispatch |
3 | 84 | // =================================================================== |
4 | 85 |
|
5 | 86 | /** |
|
0 commit comments