Class JsonRpcClient

java/src/main/java/com/github/copilot/JsonRpcClient.java:40–361

JSON-RPC 2.0 client implementation for communicating with the Copilot CLI. @since 1.0.0

 * @since 1.0.0
 */
class JsonRpcClient implements AutoCloseable {

    private static final Logger LOG = Logger.getLogger(JsonRpcClient.class.getName());
    private static final ObjectMapper MAPPER = createObjectMapper();

    private final InputStream inputStream;
    private final OutputStream outputStream;
    private final Socket socket;
    private final Process process;
    private final AtomicLong requestIdCounter = new AtomicLong(0);
    private final Map<Long, CompletableFuture<JsonNode>> pendingRequests = new ConcurrentHashMap<>();
    private final Map<String, BiConsumer<String, JsonNode>> notificationHandlers = new ConcurrentHashMap<>();
    private final ExecutorService readerExecutor;
    private volatile boolean running = true;

    private JsonRpcClient(InputStream inputStream, OutputStream outputStream, Socket socket, Process process) {
        this.inputStream = inputStream;
        this.outputStream = outputStream;
        this.socket = socket;
        this.process = process;
        this.readerExecutor = Executors.newSingleThreadExecutor(r -> {
            Thread t = new Thread(r, "jsonrpc-reader");
            t.setDaemon(true);
            return t;
        });
        startReader();
    }

    static ObjectMapper createObjectMapper() {
        var mapper = new ObjectMapper();
        mapper.registerModule(new JavaTimeModule());
        mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
        mapper.configure(SerializationFeature.WRITE_DATES_AS_TIMESTAMPS, false);
        mapper.setDefaultPropertyInclusion(JsonInclude.Include.NON_NULL);
        return mapper;
    }

    public static ObjectMapper getObjectMapper() {
        return MAPPER;
    }

    /**
     * Creates a JSON-RPC client using stdio with a process.
     */
    public static JsonRpcClient fromProcess(Process process) {
        return new JsonRpcClient(process.getInputStream(), process.getOutputStream(), null, process);
    }

    /**
     * Creates a JSON-RPC client using TCP socket.
     */
    public static JsonRpcClient fromSocket(Socket socket) throws IOException {
        return new JsonRpcClient(socket.getInputStream(), socket.getOutputStream(), socket, null);
    }

    /**
     * Registers a handler for JSON-RPC method calls (requests/notifications from
     * server).