better console parser, logger, better socket utils, more

Author: RealSuperUser

src/main/java/ir/xenoncommunity/Main.java

@@ -1,7 +1,6 @@
 package ir.xenoncommunity;
 
 import ir.xenoncommunity.utils.CommandLineParser;
-import lombok.Getter;
 
 public class Main {
 	public static MainRunner runner;

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

@@ -1,6 +1,7 @@
 package ir.xenoncommunity;
 
 import ir.xenoncommunity.utils.CommandLineParser;
+import ir.xenoncommunity.utils.Logger;
 import ir.xenoncommunity.utils.SocketUtils;
 import ir.xenoncommunity.utils.TaskManager;
 import lombok.Getter;
@@ -11,14 +12,28 @@ public class MainRunner {
 	public final CommandLineParser parser;
 	public final TaskManager taskManager;
 	public final SocketUtils socketUtils;
+	public final Logger logger = new Logger(true, true);
 
 	public MainRunner(CommandLineParser parser) {
 		this.parser = parser;
 		this.taskManager = new TaskManager();
 		this.socketUtils = new SocketUtils();
 	}
+
 	public void run() {
+		getLogger().print(Logger.LEVEL.INFO,"", "test");
+		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);
+		for(int i = 0; i <= maxThreads; i++){
+			getTaskManager().add(new Thread(() -> {
+				while (true) {
+					socketUtils.connect(ip, port, isResult, isKeepAlive);
+				}
+			}));
+		}
+		getTaskManager().doTasks();
 	}
 }

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

@@ -12,8 +12,10 @@ public CommandLineParser(String[] args) {
 
 	@Nullable
 	public <T> T get(String arg, Class<T> type) {
-		for (int i = 0; i < args.length - 1; i++) {
-			if (args[i].equals(arg))
+		for (int i = 0; i < args.length ; i++) {
+			if(args[i + 1].startsWith("-"))
+				return (T) Boolean.TRUE;
+			else if (args[i].equals(arg))
 				return convertToType(args[i + 1], type);
 		}
 		return null;

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

@@ -0,0 +1,27 @@
+package ir.xenoncommunity.utils;
+
+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;
+    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){
+        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));
+
+    }
+
+    public enum LEVEL{
+        ERROR,
+        WARN,
+        INFO
+    }
+}

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

@@ -1,26 +1,22 @@
 package ir.xenoncommunity.utils;
 
 import ir.xenoncommunity.Main;
-import ir.xenoncommunity.MainRunner;
 import lombok.SneakyThrows;
-import lombok.experimental.UtilityClass;
 import lombok.val;
-
-import java.net.InetSocketAddress;
 import java.net.Socket;
-import java.net.SocketAddress;
 
 public class SocketUtils {
-   
     @SneakyThrows
     public void connect(final String ipIn, final int port, final boolean sendResult, final boolean keepAlive){
-        val socket = new Socket(ipIn, port);
-        if(sendResult && socket.isConnected()) System.out.println(String.format("Connected! ip: %s, port: %s, threads: %s", ipIn, port, Main.runner.getTaskManager().tasks.size()));
-        if(keepAlive) {
-            socket.setKeepAlive(true);
-        } else{
-            socket.close();
+        try(val socket = new Socket(ipIn, port)){
+            if (sendResult && socket.isConnected()) {
+                System.out.printf("Connected! ip: %s, port: %s, threads: %s\n", ipIn, port, Main.runner.getTaskManager().tasks.size());
+            }
+            if (keepAlive) {
+                socket.setKeepAlive(true);
+            } else{
+                socket.close();
+            }
         }
-
     }
 }

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

@@ -1,8 +1,5 @@
 package ir.xenoncommunity.utils;
 
-import lombok.experimental.UtilityClass;
-import lombok.val;
-
 import java.util.ArrayList;
 import java.util.List;
 
@@ -15,8 +12,6 @@ public void remove(final int idIn){
         tasks.get(idIn).interrupt();
     }
     public void doTasks(){
-        tasks.forEach(e ->{
-            e.start();
-        });
+        tasks.forEach(Thread::start);
     }
 }