Servidor de órdenes con sockets e hilos

1import java.io.BufferedReader;
2import java.io.IOException;
3import java.io.InputStreamReader;
4import java.io.OutputStreamWriter;
5import java.io.PrintWriter;
6import java.net.ServerSocket;
7import java.net.Socket;
8import java.nio.charset.StandardCharsets;
9import java.util.Map;
10import java.util.Scanner;
11import java.util.concurrent.ConcurrentHashMap;
12
13public class Main {
14    // Compartido por todas las conexiones: cada cliente se atiende en su propio hilo
15    static final Map<String, String> almacen = new ConcurrentHashMap<>();
16
17    /** La respuesta del protocolo a una línea. */
18    static String responder(String linea) {
19        String[] p = linea.trim().split("\\s+", 2);
20        String orden = p[0].toUpperCase();
21        String resto = p.length > 1 ? p[1] : "";
22        switch (orden) {
23            case "" -> { return "ERR línea vacía"; }
24            case "SUMA" -> {
25                String[] n = resto.split("\\s+");
26                if (n.length != 2 || !n[0].matches("-?\\d{1,9}") || !n[1].matches("-?\\d{1,9}")) return "ERR números no válidos";
27                return "OK " + (Long.parseLong(n[0]) + Long.parseLong(n[1]));
28            }
29            case "MAYUS" -> { return resto.isEmpty() ? "ERR falta el texto" : "OK " + resto.toUpperCase(); }
30            case "INVIERTE" -> { return resto.isEmpty() ? "ERR falta el texto" : "OK " + new StringBuilder(resto).reverse(); }
31            case "GUARDA" -> {
32                String[] kv = resto.split("\\s+", 2);
33                if (kv.length != 2) return "ERR uso: GUARDA clave valor";
34                String anterior = almacen.put(kv[0], kv[1]);
35                return anterior == null ? "OK guardado" : "OK sustituido (antes: " + anterior + ")";
36            }
37            case "LEE" -> {
38                String valor = almacen.get(resto);
39                return valor == null ? "ERR no existe " + resto : "OK " + valor;
40            }
41            case "CUENTA" -> { return "OK " + almacen.size(); }
42            case "ADIOS" -> { return "OK adiós"; }
43            default -> { return "ERR orden desconocida"; }
44        }
45    }
46
47    /** Atiende a un cliente: una respuesta por línea hasta ADIOS o hasta que cierre la conexión. */
48    static void atender(Socket socket) {
49        try (socket;
50             BufferedReader in = new BufferedReader(new InputStreamReader(socket.getInputStream(), StandardCharsets.UTF_8));
51             PrintWriter out = new PrintWriter(new OutputStreamWriter(socket.getOutputStream(), StandardCharsets.UTF_8), true)) {
52            String linea;
53            while ((linea = in.readLine()) != null) {
54                out.println(responder(linea));
55                if (linea.trim().equalsIgnoreCase("ADIOS")) break;
56            }
57        } catch (IOException e) {
58            System.out.println("Error con un cliente: " + e.getMessage());
59        }
60    }
61
62    public static void main(String[] args) throws Exception {
63        // Puerto 0: el sistema elige uno libre
64        ServerSocket servidor = new ServerSocket(0);
65        Thread aceptador = new Thread(() -> {
66            try {
67                while (true) {
68                    Socket cliente = servidor.accept();
69                    new Thread(() -> atender(cliente)).start();
70                }
71            } catch (IOException e) {
72                // El servidor se ha cerrado: se termina el bucle
73            }
74        });
75        aceptador.start();
76
77        // El cliente: cada «---» de la entrada abre una conexión nueva
78        Scanner sc = new Scanner(System.in);
79        int numero = 0;
80        Socket conexion = null;
81        BufferedReader in = null;
82        PrintWriter out = null;
83        while (sc.hasNextLine()) {
84            String linea = sc.nextLine();
85            if (conexion == null || linea.equals("---")) {
86                if (conexion != null) conexion.close();
87                conexion = new Socket("localhost", servidor.getLocalPort());
88                in = new BufferedReader(new InputStreamReader(conexion.getInputStream(), StandardCharsets.UTF_8));
89                out = new PrintWriter(new OutputStreamWriter(conexion.getOutputStream(), StandardCharsets.UTF_8), true);
90                System.out.println("== Cliente " + (++numero) + " conectado");
91                if (linea.equals("---")) continue;
92            }
93            out.println(linea);
94            String respuesta = in.readLine();
95            System.out.println(linea.isEmpty() ? ">" : "> " + linea);
96            System.out.println("< " + (respuesta == null ? "(conexión cerrada)" : respuesta));
97            if (respuesta != null && linea.trim().equalsIgnoreCase("ADIOS")) {
98                conexion.close();
99                conexion = null;
100            }
101        }
102        if (conexion != null) conexion.close();
103        servidor.close();
104        aceptador.join();
105        System.out.println("Servidor cerrado · claves guardadas: " + almacen.size());
106    }
107}