From f626e9a4e02a482e427cfcc1036a826791309bd3 Mon Sep 17 00:00:00 2001
From: "tobias.pobocik" <tobias.pobocik@pantheon.tech>
Date: Wed, 6 Sep 2023 14:26:13 +0200
Subject: [PATCH] Enhance output format in analyze function

Refined the output format in the analyze function to improve
readability and clarity when analyzing multiple modules.

JIRA:LIGHTY-184
Signed-off-by: tobias.pobocik <tobias.pobocik@pantheon.tech>
---
 .../java/io/lighty/yang/validator/formats/Analyzer.java  | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/main/java/io/lighty/yang/validator/formats/Analyzer.java b/src/main/java/io/lighty/yang/validator/formats/Analyzer.java
index 57875227..55576448 100644
--- a/src/main/java/io/lighty/yang/validator/formats/Analyzer.java
+++ b/src/main/java/io/lighty/yang/validator/formats/Analyzer.java
@@ -10,6 +10,7 @@
 import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
 import io.lighty.yang.validator.GroupArguments;
 import java.util.Collection;
+import java.util.Collections;
 import java.util.HashMap;
 import java.util.HashSet;
 import java.util.Map;
@@ -32,11 +33,12 @@ public class Analyzer extends FormatPlugin {
 
     @Override
     void emitFormat(final Module module) {
-        final Set<DeclaredStatement<?>> statements = getRecursivelyDeclaredStatements(modelContext.getModules());
+        final Set<DeclaredStatement<?>> statements = getRecursivelyDeclaredStatements(Collections.singleton(module));
         for (final DeclaredStatement<?> declaredStatement : statements) {
             analyzeSubstatement(declaredStatement);
         }
-        printOut();
+        printOut(module);
+        counter.clear();
     }
 
     private Set<DeclaredStatement<?>> getRecursivelyDeclaredStatements(final Collection<? extends ModuleLike> modules) {
@@ -58,7 +60,8 @@ private static boolean submodulesAreNotEmpty(final Collection<? extends ModuleLi
 
     @SuppressFBWarnings(value = "SLF4J_SIGN_ONLY_FORMAT",
                         justification = "Valid output from LYV is dependent on Logback output")
-    private void printOut() {
+    private void printOut(final Module module) {
+        LOG.info("\n{}:", module.getName());
         for (final Map.Entry<String, Integer> entry : new TreeMap<>(counter).entrySet()) {
             LOG.info("{}: {}", entry.getKey(), entry.getValue());
         }