debug mode

Author: RealSuperUser

src/main/java/ir/xenoncommunity/MainRunner.java

@@ -9,6 +9,7 @@
 
 @Getter
 public class MainRunner {
+	public boolean isDebug;
 	public final CommandLineParser parser;
 	public final TaskManager taskManager;
 	public final SocketUtils socketUtils;
@@ -21,19 +22,29 @@ public MainRunner(CommandLineParser parser) {
 	}
 
 	public void run() {
-		getLogger().print(Logger.LEVEL.INFO,"", "test");
+		this.isDebug  = Main.runner.parser.get("--debug", Boolean.class);
 		val ip = Main.runner.parser.get("--ip", String.class);
 		val port = Main.runner.parser.get("--port", Integer.class);
 		val maxThreads = Main.runner.parser.get("--threads", Integer.class);
 		val isResult = Main.runner.parser.get("--sendResult", Boolean.class);
 		val isKeepAlive = Main.runner.parser.get("--keepAlive", Boolean.class);
+		if(this.isDebug()){
+			getLogger().setSection("DEBUG");
+			getLogger().print(Logger.LEVEL.INFO, "ip is: " + ip);
+			getLogger().print(Logger.LEVEL.INFO, "port is: " + port);
+			getLogger().print(Logger.LEVEL.INFO, "maxThreads is: " + maxThreads);
+			getLogger().print(Logger.LEVEL.INFO, "isResult is: " + isResult);
+			getLogger().print(Logger.LEVEL.INFO, "isKeepAlive is: " + isKeepAlive);
+		}
 		for(int i = 0; i <= maxThreads; i++){
+			if(this.isDebug)
+				getLogger().print(Logger.LEVEL.INFO, "adding new thread. max: " + maxThreads);
 			getTaskManager().add(new Thread(() -> {
-				while (true) {
+				while (true)
 					socketUtils.connect(ip, port, isResult, isKeepAlive);
-				}
 			}));
 		}
+		getLogger().print(Logger.LEVEL.INFO,  "doing tasks...");
 		getTaskManager().doTasks();
 	}
 }

src/main/java/ir/xenoncommunity/utils/Logger.java

@@ -1,21 +1,23 @@
 package ir.xenoncommunity.utils;
 
+import lombok.Setter;
 import lombok.val;
 
 public class Logger {
     public String osName = System.getProperty("os.name");
     public String osVersion = System.getProperty("os.version");
     private boolean sendOsName;
     private boolean sendOsVersion;
+    @Setter
+    public String section = "NONE";
     public Logger(final boolean sendOsName, final boolean sendOSVersion){
         this.sendOsName = sendOsName;
         this.sendOsVersion = sendOSVersion;
     }
-    public void print(final LEVEL levelIn, final String sectionIn, final String message){
+    public void print(final LEVEL levelIn, final String message){
         val osName = sendOsName ? this.osName : "";
         val osVersion = sendOsVersion ? this.osVersion : "";
-        val section = sectionIn.equals("") ? "NONE" : sectionIn;
-        System.out.println(String.format("[%s-%s] [%s] [%s]>> %s", osName, osVersion, levelIn, section, message));
+        System.out.println(String.format("[%s-%s] [%s] [%s]>> %s", osName, osVersion, levelIn, this.section, message));
 
     }
 

src/main/java/ir/xenoncommunity/utils/SocketUtils.java

@@ -9,8 +9,9 @@ public class SocketUtils {
     @SneakyThrows
     public void connect(final String ipIn, final int port, final boolean sendResult, final boolean keepAlive){
         try(val socket = new Socket(ipIn, port)){
+            Main.runner.getLogger().setSection("CONNECT");
             if (sendResult && socket.isConnected()) {
-                System.out.printf("Connected! ip: %s, port: %s, threads: %s\n", ipIn, port, Main.runner.getTaskManager().tasks.size());
+                Main.runner.getLogger().print(Logger.LEVEL. INFO, String.format("Connected! ip: %s, port: %s, threads: %s\n", ipIn, port, Main.runner.getTaskManager().tasks.size()));
             }
             if (keepAlive) {
                 socket.setKeepAlive(true);