cache results when parsing multiple files

palantir · Aug 11, 2020 · 2f6e80e · 2f6e80e
1 parent bf882a1
commit 2f6e80e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 4 deletions.
diff --git a/conjure-core/src/main/java/com/palantir/conjure/defs/Conjure.java b/conjure-core/src/main/java/com/palantir/conjure/defs/Conjure.java
@@ -23,7 +23,6 @@
 import java.io.File;
 import java.util.Collection;
 import java.util.List;
-import java.util.stream.Collectors;
 
 public final class Conjure {
     public static final Integer SUPPORTED_IR_VERSION = 1;
@@ -34,8 +33,7 @@ private Conjure() {}
      * Deserializes {@link ConjureDefinition} from their YAML representations in the given files.
      */
     public static ConjureDefinition parse(Collection<File> files) {
-        List<AnnotatedConjureSourceFile> sourceFiles =
-                files.stream().map(ConjureParser::parseAnnotated).collect(Collectors.toList());
+        List<AnnotatedConjureSourceFile> sourceFiles = ConjureParser.parseAnnotated(files);
         ConjureDefinition ir = ConjureParserUtils.parseConjureDef(sourceFiles);
         return NormalizeDefinition.normalize(ir);
     }

diff --git a/conjure-core/src/main/java/com/palantir/conjure/parser/ConjureParser.java b/conjure-core/src/main/java/com/palantir/conjure/parser/ConjureParser.java
@@ -30,8 +30,10 @@
 import java.io.IOException;
 import java.nio.file.Files;
 import java.nio.file.Path;
+import java.util.Collection;
 import java.util.HashMap;
 import java.util.LinkedHashSet;
+import java.util.List;
 import java.util.Map;
 import java.util.Set;
 import java.util.stream.Collectors;
@@ -61,8 +63,18 @@ public static ConjureSourceFile parse(File file) {
     }
 
     public static AnnotatedConjureSourceFile parseAnnotated(File file) {
+        RecursiveParser parser = new RecursiveParser();
+        return parseAnnotated(parser, file);
+    }
+
+    public static List<AnnotatedConjureSourceFile> parseAnnotated(Collection<File> files) {
+        RecursiveParser parser = new RecursiveParser();
+        return files.stream().map(file -> parseAnnotated(parser, file)).collect(Collectors.toList());
+    }
+
+    private static AnnotatedConjureSourceFile parseAnnotated(RecursiveParser parser, File file) {
         return AnnotatedConjureSourceFile.builder()
-                .conjureSourceFile(ConjureParser.parse(file))
+                .conjureSourceFile(parser.parse(file))
                 .sourceFile(file)
                 .build();
     }