Use count method

openrewrite · Apr 22, 2024 · 38e2367 · 38e2367
1 parent 9e6751b
commit 38e2367
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/src/main/java/org/openrewrite/LanguageComposition.java b/src/main/java/org/openrewrite/LanguageComposition.java
@@ -109,7 +109,7 @@ public TreeVisitor<?, ExecutionContext> getScanner(Accumulator acc) {
                             0,
                             hasParseFailure));
                 } else {
-                    int genericLineCount = genericLineCount(s);
+                    int genericLineCount = LineCounter.count(s);
                     if (s.getClass().getName().startsWith("org.openrewrite.cobol.tree.CobolPreprocessor$Copybook")) {
                         Counts copybookCounts = acc.getFolderToLanguageToCounts()
                                 .computeIfAbsent(folderPath, k -> new HashMap<>())
@@ -361,12 +361,6 @@ private static class Counts {
         int fileCount;
     }
 
-    private static int genericLineCount(SourceFile s) {
-        LineCounter counter = new LineCounter();
-        s.printAll(counter);
-        return counter.getLineCount();
-    }
-
     private static class LineCounter extends PrintOutputCapture<Integer> {
         private int count;