udp protocol

Author: RealSuperUser

build

@@ -23,25 +23,27 @@ public MainRunner(CommandLineParser parser) {
 
 	public void run() {
 		this.isDebug  = Main.runner.parser.get("--debug", Boolean.class);
-		val ip = Main.runner.parser.get("--ip", String.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);
+		val udp = Main.runner.parser.get("--udp", 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);
+			getLogger().print(Logger.LEVEL.INFO, "udp is: " + udp);
 		}
 		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)
-					socketUtils.connect(ip, port, isResult, isKeepAlive);
+					socketUtils.connect(ip, port, isResult, isKeepAlive, udp);
 			}));
 		}
 		getLogger().print(Logger.LEVEL.INFO,  "doing tasks...");

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

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

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

@@ -1,23 +1,39 @@
 package ir.xenoncommunity.utils;
 
 import ir.xenoncommunity.Main;
-import lombok.SneakyThrows;
 import lombok.val;
+
+import java.net.DatagramSocket;
+import java.net.InetAddress;
 import java.net.Socket;
 
 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()) {
-                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);
-            } else{
-                socket.close();
+    public void connect(final String ipAddress, final int port, final boolean sendResult, final boolean keepAlive, final boolean udp) {
+        Main.runner.getLogger().setSection("CONNECT");
+        try {
+            if (udp) {
+                if (keepAlive) {
+                    Main.runner.getLogger().print(Logger.LEVEL.ERROR, "Can't use keepAlive for UDP protocol.");
+                    throw new IllegalStateException("UDP is used with keepAlive.");
+                }
+                try (val socket = new DatagramSocket(port, InetAddress.getByName(ipAddress))) {
+                    if (sendResult && socket.isConnected()) {
+                        Main.runner.getLogger().print(Logger.LEVEL.INFO, String.format("Connected! ip: %s, port: %s, threads: %s\n", ipAddress, port, Main.runner.getTaskManager().tasks.size()));
+                    }
+                }
+            } else {
+                try (val socket = new Socket(ipAddress, port)) {
+                    if (sendResult && socket.isConnected()) {
+                        Main.runner.getLogger().print(Logger.LEVEL.INFO, String.format("Connected! ip: %s, port: %s, threads: %s\n", ipAddress, port, Main.runner.getTaskManager().tasks.size()));
+                    }
+                    if (keepAlive) {
+                        socket.setKeepAlive(true);
+                    } else
+                        socket.close();
+                }
             }
+        } catch (final Exception e) {
+            Main.runner.getLogger().print(Logger.LEVEL.ERROR, e.getMessage());
         }
     }
 }