add color for logs

jetoile · Aug 9, 2017 · b9a22ba · b9a22ba
1 parent defe700
commit b9a22ba
Show file tree

Hide file tree

Showing 18 changed files with 156 additions and 14 deletions.
diff --git a/hadoop-unit-cassandra/pom.xml b/hadoop-unit-cassandra/pom.xml
@@ -23,6 +23,11 @@
             <groupId>info.archinnov</groupId>
             <artifactId>achilles-embedded</artifactId>
         </dependency>
+
+        <dependency>
+            <groupId>ch.qos.logback</groupId>
+            <artifactId>logback-classic</artifactId>
+        </dependency>
     </dependencies>
 
 </project>
diff --git a/hadoop-unit-cassandra/src/main/resources/logback.xml b/hadoop-unit-cassandra/src/main/resources/logback.xml
@@ -1,7 +1,16 @@
 <configuration>
     <appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender">
         <encoder>
-            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %msg%n
+            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %msg %n
+
+            </pattern>
+        </encoder>
+    </appender>
+
+    <appender name="STDOUT_COLOR" class="ch.qos.logback.core.ConsoleAppender">
+        <withJansi>true</withJansi>
+        <encoder>
+            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %green(%msg) %n
 
             </pattern>
         </encoder>
@@ -14,4 +23,7 @@
         <appender-ref ref="STDOUT" />
     </root>
 
+    <logger name="fr.jetoile.hadoopunit" level="INFO">
+        <appender-ref ref="STDOUT_COLOR" />
+    </logger>
 </configuration>
diff --git a/hadoop-unit-commons/pom.xml b/hadoop-unit-commons/pom.xml
@@ -55,6 +55,10 @@
             <artifactId>javax.servlet-api</artifactId>
         </dependency>
 
+        <dependency>
+            <groupId>org.fusesource.jansi</groupId>
+            <artifactId>jansi</artifactId>
+        </dependency>
     </dependencies>
 
     <build>

diff --git a/hadoop-unit-commons/src/main/java/fr/jetoile/hadoopunit/HadoopBootstrap.java b/hadoop-unit-commons/src/main/java/fr/jetoile/hadoopunit/HadoopBootstrap.java
@@ -21,6 +21,7 @@
 import java.util.*;
 
 import static java.util.stream.Collectors.toList;
+import static org.fusesource.jansi.Ansi.Color.GREEN;
 
 public enum HadoopBootstrap {
     INSTANCE;
@@ -101,7 +102,7 @@ private void internalStart(List<Bootstrap> componentsToStart) {
             startService(c);
         });
         HadoopUtils.INSTANCE.printBanner(System.out);
-        componentsToStart.stream().forEach(c -> System.out.println("\t\t - " + c.getName() + " " + c.getProperties()));
+        componentsToStart.stream().forEach(c -> HadoopUtils.printColorLine(System.out, GREEN, "\t\t - " + c.getName() + " " + c.getProperties()));
         System.out.println();
     }
 

diff --git a/hadoop-unit-commons/src/main/java/fr/jetoile/hadoopunit/HadoopUtils.java b/hadoop-unit-commons/src/main/java/fr/jetoile/hadoopunit/HadoopUtils.java
@@ -19,11 +19,16 @@
 import org.apache.commons.io.FileUtils;
 import org.apache.commons.io.IOUtils;
 import org.apache.commons.lang3.StringUtils;
+import org.fusesource.jansi.Ansi;
+import org.fusesource.jansi.AnsiConsole;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 import java.io.*;
 
+import static org.fusesource.jansi.Ansi.Color.BLUE;
+import static org.fusesource.jansi.Ansi.ansi;
+
 public enum HadoopUtils {
     INSTANCE;
 
@@ -73,6 +78,12 @@ private HadoopUtils() {
         }
     }
 
+    public static void printColorLine(PrintStream out, Ansi.Color color, String line) {
+        AnsiConsole.systemInstall();
+        out.println(ansi().fg(color).a(line).reset());
+        AnsiConsole.systemUninstall();
+    }
+
     private void extractAndLoadDll(String lib) throws IOException {
         InputStream in = HadoopUtils.class.getResourceAsStream(lib);
         // always write to different location
@@ -113,7 +124,7 @@ public void printBanner(PrintStream out) {
             BufferedReader br = new BufferedReader(new InputStreamReader(banner));
             String line = null;
             while ((line = br.readLine()) != null) {
-                out.println(line);
+                printColorLine(out, BLUE, line);
             }
         } catch (Exception ex) {
             LOGGER.warn("Banner not printable", ex);

diff --git a/hadoop-unit-elasticsearch/pom.xml b/hadoop-unit-elasticsearch/pom.xml
@@ -40,6 +40,11 @@
             <version>2.8.2</version>
         </dependency>
 
+        <dependency>
+            <groupId>ch.qos.logback</groupId>
+            <artifactId>logback-classic</artifactId>
+        </dependency>
+
     </dependencies>
 
 </project>
diff --git a/hadoop-unit-elasticsearch/src/main/resources/logback.xml b/hadoop-unit-elasticsearch/src/main/resources/logback.xml
@@ -1,14 +1,29 @@
 <configuration>
     <appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender">
         <encoder>
-            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %msg%n
+            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %msg %n
 
             </pattern>
         </encoder>
     </appender>
 
+    <appender name="STDOUT_COLOR" class="ch.qos.logback.core.ConsoleAppender">
+        <withJansi>true</withJansi>
+        <encoder>
+            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %green(%msg) %n
+
+            </pattern>
+        </encoder>
+    </appender>
+
+    <logger name="BlockStateChange" level="OFF"/>
+    <logger name="BlockManager" level="OFF"/>
+
     <root level="info">
         <appender-ref ref="STDOUT" />
     </root>
 
+    <logger name="fr.jetoile.hadoopunit" level="INFO">
+        <appender-ref ref="STDOUT_COLOR" />
+    </logger>
 </configuration>
diff --git a/hadoop-unit-hbase/src/main/resources/logback.xml b/hadoop-unit-hbase/src/main/resources/logback.xml
@@ -1,7 +1,16 @@
 <configuration>
     <appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender">
         <encoder>
-            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %msg%n
+            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %msg %n
+
+            </pattern>
+        </encoder>
+    </appender>
+
+    <appender name="STDOUT_COLOR" class="ch.qos.logback.core.ConsoleAppender">
+        <withJansi>true</withJansi>
+        <encoder>
+            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %green(%msg) %n
 
             </pattern>
         </encoder>
@@ -14,4 +23,7 @@
         <appender-ref ref="STDOUT" />
     </root>
 
+    <logger name="fr.jetoile.hadoopunit" level="INFO">
+        <appender-ref ref="STDOUT_COLOR" />
+    </logger>
 </configuration>
diff --git a/hadoop-unit-maven-plugin/src/main/java/fr/jetoile/hadoopunit/HadoopBootstrapStarter.java b/hadoop-unit-maven-plugin/src/main/java/fr/jetoile/hadoopunit/HadoopBootstrapStarter.java
@@ -57,6 +57,8 @@
 import java.util.Map;
 import java.util.stream.Collectors;
 
+import static org.fusesource.jansi.Ansi.Color.GREEN;
+
 
 @Mojo(name = "embedded-start", defaultPhase = LifecyclePhase.PRE_INTEGRATION_TEST, threadSafe = false)
 public class HadoopBootstrapStarter extends AbstractMojo {
@@ -200,8 +202,7 @@ private void printBanner() {
                 } catch (NoSuchMethodException | IllegalAccessException | InvocationTargetException e) {
                     getLog().error("unable to reflect main", e);
                 }
-
-                System.out.println("\t\t - " + name + " " + prop);
+                HadoopUtils.printColorLine(System.out, GREEN, "\t\t - " + name + " " + prop);
             }
         });
         System.out.println();

diff --git a/hadoop-unit-mongodb/pom.xml b/hadoop-unit-mongodb/pom.xml
@@ -23,6 +23,11 @@
             <artifactId>hadoop-unit-commons</artifactId>
         </dependency>
 
+        <dependency>
+            <groupId>ch.qos.logback</groupId>
+            <artifactId>logback-classic</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>com.github.sakserv</groupId>
             <artifactId>hadoop-mini-clusters-mongodb</artifactId>

diff --git a/hadoop-unit-mongodb/src/main/resources/logback.xml b/hadoop-unit-mongodb/src/main/resources/logback.xml
@@ -1,7 +1,16 @@
 <configuration>
     <appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender">
         <encoder>
-            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %msg%n
+            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %msg %n
+
+            </pattern>
+        </encoder>
+    </appender>
+
+    <appender name="STDOUT_COLOR" class="ch.qos.logback.core.ConsoleAppender">
+        <withJansi>true</withJansi>
+        <encoder>
+            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %green(%msg) %n
 
             </pattern>
         </encoder>
@@ -14,4 +23,7 @@
         <appender-ref ref="STDOUT" />
     </root>
 
+    <logger name="fr.jetoile.hadoopunit" level="INFO">
+        <appender-ref ref="STDOUT_COLOR" />
+    </logger>
 </configuration>
diff --git a/hadoop-unit-neo4j/pom.xml b/hadoop-unit-neo4j/pom.xml
@@ -32,6 +32,11 @@
             <artifactId>neo4j-bolt</artifactId>
         </dependency>
 
+        <dependency>
+            <groupId>ch.qos.logback</groupId>
+            <artifactId>logback-classic</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>org.neo4j.driver</groupId>
             <artifactId>neo4j-java-driver</artifactId>

diff --git a/hadoop-unit-neo4j/src/main/resources/logback.xml b/hadoop-unit-neo4j/src/main/resources/logback.xml
@@ -1,7 +1,16 @@
 <configuration>
     <appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender">
         <encoder>
-            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %msg%n
+            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %msg %n
+
+            </pattern>
+        </encoder>
+    </appender>
+
+    <appender name="STDOUT_COLOR" class="ch.qos.logback.core.ConsoleAppender">
+        <withJansi>true</withJansi>
+        <encoder>
+            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %green(%msg) %n
 
             </pattern>
         </encoder>
@@ -14,4 +23,7 @@
         <appender-ref ref="STDOUT" />
     </root>
 
+    <logger name="fr.jetoile.hadoopunit" level="INFO">
+        <appender-ref ref="STDOUT_COLOR" />
+    </logger>
 </configuration>
diff --git a/hadoop-unit-solr/src/main/resources/logback.xml b/hadoop-unit-solr/src/main/resources/logback.xml
@@ -1,7 +1,16 @@
 <configuration>
     <appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender">
         <encoder>
-            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %msg%n
+            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %msg %n
+
+            </pattern>
+        </encoder>
+    </appender>
+
+    <appender name="STDOUT_COLOR" class="ch.qos.logback.core.ConsoleAppender">
+        <withJansi>true</withJansi>
+        <encoder>
+            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %green(%msg) %n
 
             </pattern>
         </encoder>
@@ -14,4 +23,7 @@
         <appender-ref ref="STDOUT" />
     </root>
 
+    <logger name="fr.jetoile.hadoopunit" level="INFO">
+        <appender-ref ref="STDOUT_COLOR" />
+    </logger>
 </configuration>
diff --git a/hadoop-unit-standalone/pom.xml b/hadoop-unit-standalone/pom.xml
@@ -29,6 +29,11 @@
             <artifactId>hadoop-unit-commons</artifactId>
         </dependency>
 
+        <dependency>
+            <groupId>ch.qos.logback</groupId>
+            <artifactId>logback-classic</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>org.apache.commons</groupId>
             <artifactId>commons-lang3</artifactId>

diff --git a/hadoop-unit-standalone/src/main/java/fr/jetoile/hadoopunit/HadoopStandaloneBootstrap.java b/hadoop-unit-standalone/src/main/java/fr/jetoile/hadoopunit/HadoopStandaloneBootstrap.java
@@ -53,6 +53,8 @@
 import java.util.Arrays;
 import java.util.List;
 
+import static org.fusesource.jansi.Ansi.Color.GREEN;
+
 public class HadoopStandaloneBootstrap {
 
     final private static Logger LOGGER = LoggerFactory.getLogger(HadoopStandaloneBootstrap.class);
@@ -186,8 +188,7 @@ private static void printBanner() {
                 } catch (NoSuchMethodException | IllegalAccessException | InvocationTargetException e) {
                     LOGGER.error("unable to reflect main", e);
                 }
-
-                System.out.println("\t\t - " + name + " " + prop);
+                HadoopUtils.printColorLine(System.out, GREEN, "\t\t - " + name + " " + prop);
             }
         });
         System.out.println();

diff --git a/hadoop-unit-standalone/src/main/resources/logback.xml b/hadoop-unit-standalone/src/main/resources/logback.xml
@@ -1,7 +1,16 @@
 <configuration>
     <appender name="STDOUT" class="ch.qos.logback.core.ConsoleAppender">
         <encoder>
-            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %msg%n
+            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %msg %n
+
+            </pattern>
+        </encoder>
+    </appender>
+
+    <appender name="STDOUT_COLOR" class="ch.qos.logback.core.ConsoleAppender">
+        <withJansi>true</withJansi>
+        <encoder>
+            <pattern>%d{HH:mm:ss.SSS} %-5level %logger{36} - %green(%msg) %n
 
             </pattern>
         </encoder>
@@ -14,4 +23,7 @@
         <appender-ref ref="STDOUT" />
     </root>
 
+    <logger name="fr.jetoile.hadoopunit" level="INFO">
+        <appender-ref ref="STDOUT_COLOR" />
+    </logger>
 </configuration>
diff --git a/pom.xml b/pom.xml
@@ -123,7 +123,7 @@
         <hadoop-unit.version>${project.version}</hadoop-unit.version>
 
         <slf4j.version>1.7.12</slf4j.version>
-        <logback-classic.version>1.1.3</logback-classic.version>
+        <logback-classic.version>1.2.3</logback-classic.version>
 
         <commons-io.version>2.5</commons-io.version>
         <commons-configuration.version>1.10</commons-configuration.version>
@@ -166,11 +166,23 @@
         <ant.version>1.10.1</ant.version>
         <redis-unit.version>1.0.3</redis-unit.version>
         <plexus.version>2.2</plexus.version>
+        <jansi.version>1.16</jansi.version>
     </properties>
 
 
     <dependencyManagement>
         <dependencies>
+            <dependency>
+                <groupId>org.fusesource.jansi</groupId>
+                <artifactId>jansi</artifactId>
+                <version>${jansi.version}</version>
+            </dependency>
+
+            <dependency>
+                <groupId>ch.qos.logback</groupId>
+                <artifactId>logback-classic</artifactId>
+                <version>${logback-classic.version}</version>
+            </dependency>
 
             <dependency>
                 <groupId>com.ninja-squad</groupId>