Add udp and tcp listeners

src/main/java/org/example/ServerApp.java

@@ -4,13 +4,16 @@ import java.io.BufferedReader;
 import java.io.IOException;
 import java.io.InputStreamReader;
 import java.io.PrintWriter;
+import java.net.DatagramPacket;
+import java.net.DatagramSocket;
 import java.net.ServerSocket;
 import java.net.Socket;
 import java.util.ArrayList;
 import java.util.List;
 
 public class ServerApp {
-    private static final int PORT = 1234;
+    private static final int TCP_PORT = 1234;
+    private static final int UDP_PORT = 5678;
     private List<ClientHandler> clients;
 
     public ServerApp() {
@@ -19,36 +22,88 @@ public class ServerApp {
 
     public void start() {
         try {
-            ServerSocket serverSocket = new ServerSocket(PORT);
-            System.out.println("Server started on port " + PORT);
-
-            while (true) {
-                Socket clientSocket = serverSocket.accept();
-                System.out.println("New client connected: " + clientSocket);
-
-                ClientHandler clientHandler = new ClientHandler(clientSocket);
-                clients.add(clientHandler);
-                clientHandler.start();
-            }
+            ServerSocket serverSocket = new ServerSocket(TCP_PORT);
+            DatagramSocket datagramSocket = new DatagramSocket(UDP_PORT);
+            System.out.println("Server started on TCP port " + TCP_PORT + " and UDP port " + UDP_PORT);
+            TCPListener tcpListener = new TCPListener(serverSocket);
+            UDPListener udpListener = new UDPListener(datagramSocket);
+            tcpListener.start();
+            udpListener.start();
         } catch (IOException e) {
             e.printStackTrace();
         }
     }
 
-    private void broadcastMessage(String message, ClientHandler sender) {
+    private void broadcastTCPMessage(String message, ClientHandler sender) {
         for (ClientHandler client : clients) {
             if (client != sender) {
-                client.sendMessage(message);
+                client.sendTCPMessage(message);
             }
         }
     }
 
+    private void broadcastUDPMessage(String message) {
+        for (ClientHandler client : clients) {
+            client.sendUDPMessage(message);
+        }
+    }
+
     private void removeClient(ClientHandler client) {
         clients.remove(client);
     }
 
+    private class TCPListener extends Thread {
+        private ServerSocket serverSocket;
+
+        public TCPListener(ServerSocket serverSocket) {
+            this.serverSocket = serverSocket;
+        }
+
+        @Override
+        public void run() {
+            while (true) {
+                try {
+                    Socket clientSocket = serverSocket.accept();
+                    System.out.println("New client connected: " + clientSocket);
+                    ClientHandler clientHandler = new ClientHandler(clientSocket);
+                    clients.add(clientHandler);
+                    clientHandler.start();
+                } catch (IOException e) {
+                    e.printStackTrace();
+                }
+            }
+        }
+    }
+
+    private class UDPListener extends Thread {
+        private DatagramSocket datagramSocket;
+
+        public UDPListener(DatagramSocket datagramSocket) {
+            this.datagramSocket = datagramSocket;
+        }
+
+        @Override
+        public void run() {
+            try {
+                byte[] buffer = new byte[1024];
+                DatagramPacket packet = new DatagramPacket(buffer, buffer.length);
+
+                while (true) {
+                    datagramSocket.receive(packet);
+                    String message = new String(packet.getData(), 0, packet.getLength());
+                    broadcastTCPMessage(message, null);
+                }
+            } catch (IOException e) {
+                e.printStackTrace();
+            } finally {
+                datagramSocket.close();
+            }
+        }
+    }
+
     private class ClientHandler extends Thread {
         private Socket clientSocket;
+        private DatagramSocket datagramSocket;
         private BufferedReader in;
         private PrintWriter out;
 
@@ -61,9 +116,11 @@ public class ServerApp {
             try {
                 in = new BufferedReader(new InputStreamReader(clientSocket.getInputStream()));
                 out = new PrintWriter(clientSocket.getOutputStream(), true);
+                datagramSocket = new DatagramSocket();
+
                 String clientMessage;
                 while ((clientMessage = in.readLine()) != null) {
-                    broadcastMessage(clientMessage, this);
+                    broadcastTCPMessage(clientMessage, this);
                 }
             } catch (IOException e) {
                 e.printStackTrace();
@@ -79,9 +136,19 @@ public class ServerApp {
             }
         }
 
-        public void sendMessage(String message) {
+        public void sendTCPMessage(String message) {
             out.println(message);
         }
+
+        public void sendUDPMessage(String message) {
+            try {
+                byte[] buffer = message.getBytes();
+                DatagramPacket packet = new DatagramPacket(buffer, buffer.length, clientSocket.getInetAddress(), UDP_PORT);
+                datagramSocket.send(packet);
+            } catch (IOException e) {
+                e.printStackTrace();
+            }
+        }
     }
 
     public static void main(String[] args) {