Use withNewSocket() when parsing

The Python remoting server currently always closes the socket after each request.
openrewrite · Oct 14, 2024 · d07a0b7 · d07a0b7
1 parent cd847d1
commit d07a0b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/org/openrewrite/python/PythonParser.java b/src/main/java/org/openrewrite/python/PythonParser.java
@@ -92,7 +92,7 @@ public Stream<SourceFile> parseInputs(Iterable<Input> inputs, @Nullable Path rel
             parsingListener.startedParsing(input);
 
             try (EncodingDetectingInputStream is = input.getSource(ctx)) {
-                SourceFile parsed = client.runUsingSocket((socket, messenger) -> requireNonNull(messenger.sendRequest(generator -> {
+                SourceFile parsed = client.withNewSocket((socket, messenger) -> requireNonNull(messenger.sendRequest(generator -> {
                             if (input.isSynthetic() || !Files.isRegularFile(input.getPath())) {
                                 generator.writeString("parse-python-source");
                                 generator.writeString(is.readFully());