Add CloseHandler to support graceful connection termination.

- Extend `test.proto` with `CloseCommand` and `CloseResponse` messages.
- Introduce `CloseHandler` to process "close" actions and respond with connection termination notice.
- Update `DelegateActionManager` to register `CloseHandler`.
- Refactor `IPCManager` to handle "close" response headers and terminate client connections gracefully.

pom.xml

@@ -55,7 +55,7 @@
         </dependency>
         <dependency>
             <groupId>org.apache.logging.log4j</groupId>
-            <artifactId>log4j-slf4j-impl</artifactId>
+            <artifactId>log4j-slf4j2-impl</artifactId>
             <version>${log4j.version}</version>
         </dependency>
 
@@ -83,13 +83,6 @@
             <artifactId>protobuf-java</artifactId>
             <version>4.32.0</version>
         </dependency>
-        <dependency>
-            <groupId>io.zonky.test</groupId>
-            <artifactId>embedded-postgres</artifactId>
-            <version>2.2.0</version>
-        </dependency>
-
-
     </dependencies>
 
     <build>

src/main/java/com/mediamanager/MediaManagerApplication.java

@@ -9,55 +9,22 @@ import java.io.IOException;
 import java.io.InputStream;
 import java.util.Properties;
 
-import com.mediamanager.service.database.LocalDatabaseManager;
+import com.mediamanager.service.database.DatabaseManager;
 
 public class MediaManagerApplication {
     private static final Logger logger = LogManager.getLogger(MediaManagerApplication.class);
     private static Properties config;
-    private static LocalDatabaseManager databaseManager;
+    private static DatabaseManager databaseManager;
     private static DelegateActionManager actionManager;
     private static IPCManager ipcManager;
 
-    public enum ApplicationMode {
-        LOCAL("local"),
-        SERVER("server");
-
-        private final String value;
-
-        ApplicationMode(String value) {
-            this.value = value;
-        }
-    }
-
     public static void main(String[] args) {
         logger.info("Starting MediaManager Core Application...");
 
         try {
             // Load configuration
             loadConfiguration();
-            String runTypeString = config.getProperty("runtype","local");
-            ApplicationMode mode = null;
-
-            for (ApplicationMode am : ApplicationMode.values()) {
-                if (am.value.equalsIgnoreCase(runTypeString)) {
-                    mode = am;
-                    break;
-                }
-            }
-            if (mode == null) {
-                logger.error("Invalid run type: {}", runTypeString);
-                throw new Exception("Invalid run type: " + runTypeString);
-            }
-            logger.info("Run type: {}", mode);
-            switch (mode) {
-                case LOCAL:
-                    logger.info("Starting local database...");
-                    databaseManager = new LocalDatabaseManager(config);
-                    break;
-                case SERVER:
-                    throw new Exception("Server mode not yet implemented");
-                default:
-            }
+            databaseManager = new DatabaseManager(config);
             databaseManager.init();
             actionManager = new DelegateActionManager();
             actionManager.start();

src/main/java/com/mediamanager/service/database/DatabaseManager.java

@@ -1,28 +1,55 @@
 package com.mediamanager.service.database;
-
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 
 import java.sql.Connection;
+import java.sql.DriverManager;
 import java.sql.SQLException;
 import java.sql.Statement;
 import java.util.Properties;
 
-public abstract class DatabaseManager {
-    protected final Properties configuration;
-    protected Connection connection;
-    protected static final Logger logger = LogManager.getLogger(DatabaseManager.class);
 
+public class DatabaseManager {
+    private final Properties configuration;
+    private Connection connection;
+    private static final Logger logger = LogManager.getLogger(DatabaseManager.class);
     public DatabaseManager(Properties config) {
         this.configuration = config;
         logger.debug("DatabaseManager created with configuration:");
+
     }
 
-    public abstract void init() throws Exception;
+    public void init() throws Exception {
+        logger.info("Initializing database connection...");
+        validateConfiguration();
 
-    protected abstract Connection createConnection() throws Exception;
+        String databaseType = configuration.getProperty("database.type");
+        String databaseUrl = configuration.getProperty("database.url");
+        String databaseUsername = configuration.getProperty("database.username");
+        String databasePassword = configuration.getProperty("database.password");
+        String databasePort = configuration.getProperty("database.port");
+        String databaseName = configuration.getProperty("database.name");
 
-    protected void performSanityChecks() throws SQLException {
+
+        String connectionString = String.format("jdbc:postgresql://%s:%s/%s", databaseUrl, databasePort, databaseName);
+        logger.debug("Attempting to connect to: {}", connectionString);
+        try {
+            connection = DriverManager.getConnection(connectionString, databaseUsername, databasePassword);
+            logger.info("Database connection established successfully");
+
+            performSanityChecks();
+            logger.info("Database sanity checks passed successfully");
+
+
+        } catch (SQLException e) {
+            logger.error("Failed to connect to database", e);
+            throw new Exception("Database connection failed: " + e.getMessage(), e);
+        }
+
+
+
+    }
+    private void performSanityChecks() throws SQLException {
         logger.debug("Performing sanity checks...");
         try (Statement stmt = connection.createStatement()) {
             stmt.execute("SELECT 1");
@@ -32,6 +59,27 @@ public abstract class DatabaseManager {
         String databaseProductVersion = connection.getMetaData().getDatabaseProductVersion();
         logger.info("Connected to database: {} v{}", databaseProductName, databaseProductVersion);
     }
+    private void validateConfiguration() throws Exception {
+        String[] requiredProperties = {
+                "database.url",
+                "database.username",
+                "database.password",
+                "database.port",
+                "database.name"
+        };
+
+        for (String property : requiredProperties) {
+            if (configuration.getProperty(property) == null ||
+                    configuration.getProperty(property).trim().isEmpty()) {
+                throw new Exception("Required database configuration missing: " + property);
+            }
+        }
+
+        logger.debug("Database configuration validated successfully");
+
+        }
+
+
 
     public Connection getConnection() {
         return connection;
@@ -50,13 +98,4 @@ public abstract class DatabaseManager {
             logger.debug("No database connection to close");
         }
     }
-    protected boolean testConnection() {
-        try (Statement stmt = connection.createStatement()) {
-            stmt.execute("SELECT 1");
-            return true;
-        } catch (SQLException e) {
-            logger.error("Connection test failed", e);
-            return false;
-        }
-    }
 }

src/main/java/com/mediamanager/service/database/LocalDatabaseManager.java

@@ -1,87 +0,0 @@
-package com.mediamanager.service.database;
-
-import io.zonky.test.db.postgres.embedded.EmbeddedPostgres;
-import org.apache.logging.log4j.LogManager;
-import org.apache.logging.log4j.Logger;
-
-import java.nio.file.Files;
-import java.nio.file.Path;
-import java.nio.file.Paths;
-import java.sql.Connection;
-import java.util.Properties;
-
-public class LocalDatabaseManager extends DatabaseManager {
-    private static final Logger logger = LogManager.getLogger(LocalDatabaseManager.class);
-    private EmbeddedPostgres embeddedPostgres;
-
-    public LocalDatabaseManager(Properties config) {
-        super(config);
-    }
-
-    @Override
-    public void init() throws Exception {
-        logger.info("Initializing embedded PostgreSQL database...");
-
-        // Configurações específicas do modo local
-        String dataDir = configuration.getProperty("local.database.dir",
-                System.getProperty("user.home") + "/.mediamanager/db");
-        int port = Integer.parseInt(
-                configuration.getProperty("local.database.port", "54321")
-        );
-
-        // Cria diretório se não existir
-        Path dataPath = Paths.get(dataDir);
-        Files.createDirectories(dataPath);
-        logger.debug("Using data directory: {}", dataDir);
-
-        // Inicia o PostgreSQL embedded
-        logger.info("Starting embedded PostgreSQL on port {}", port);
-        embeddedPostgres = EmbeddedPostgres.builder()
-                .setDataDirectory(dataPath)
-                .setCleanDataDirectory(false)  // Mantém dados
-                .setPort(port)
-                .start();
-
-        // Cria a conexão
-        this.connection = createConnection();
-        logger.info("Embedded database connection established successfully");
-
-        // Usa o método da classe pai para sanity checks
-        performSanityChecks();
-        logger.info("Database sanity checks passed successfully");
-    }
-
-    @Override
-    protected Connection createConnection() throws Exception {
-        if (embeddedPostgres == null) {
-            throw new IllegalStateException("Embedded PostgreSQL not started");
-        }
-
-        try {
-            // Pega conexão do embedded
-            Connection conn = embeddedPostgres.getPostgresDatabase().getConnection();
-            logger.debug("Got connection from embedded PostgreSQL");
-            return conn;
-        } catch (Exception e) {
-            logger.error("Failed to get embedded connection", e);
-            throw new Exception("Embedded connection failed: " + e.getMessage(), e);
-        }
-    }
-
-    @Override
-    public void close() {
-        // Primeiro fecha a conexão (método da classe pai)
-        super.close();
-
-        // Depois para o embedded
-        if (embeddedPostgres != null) {
-            try {
-                logger.info("Stopping embedded PostgreSQL...");
-                embeddedPostgres.close();
-                logger.info("Embedded PostgreSQL stopped successfully");
-            } catch (Exception e) {
-                logger.error("Error stopping embedded PostgreSQL", e);
-            }
-        }
-    }
-}

src/main/java/com/mediamanager/service/delegate/DelegateActionManager.java

@@ -2,7 +2,9 @@ package com.mediamanager.service.delegate;
 
 import com.google.protobuf.ByteString;
 import com.mediamanager.protocol.TransportProtocol;
+import com.mediamanager.service.delegate.handler.CloseHandler;
 import com.mediamanager.service.delegate.handler.EchoHandler;
+import com.mediamanager.service.delegate.handler.HeartbeatHandler;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 
@@ -23,6 +25,8 @@ public class DelegateActionManager {
 
     private void registerHandlers() {
         handlerRegistry.put("echo",new EchoHandler());
+        handlerRegistry.put("heartbeat",new HeartbeatHandler());
+        handlerRegistry.put("close", new CloseHandler());
     }
 
     public void start(){

src/main/java/com/mediamanager/service/delegate/handler/CloseHandler.java

@@ -0,0 +1,32 @@
+package com.mediamanager.service.delegate.handler;
+
+import com.google.protobuf.ByteString;
+import com.google.protobuf.InvalidProtocolBufferException;
+import com.mediamanager.protocol.TestProtocol.CloseCommand;
+import com.mediamanager.protocol.TestProtocol.CloseResponse;
+import com.mediamanager.protocol.TransportProtocol;
+import com.mediamanager.service.delegate.ActionHandler;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+
+public class CloseHandler implements ActionHandler {
+    private static final Logger logger = LogManager.getLogger(CloseHandler.class);
+
+    @Override
+    public TransportProtocol.Response.Builder handle(ByteString requestPayload)
+            throws InvalidProtocolBufferException {
+
+        CloseCommand.parseFrom(requestPayload); // Valida
+
+        logger.info("Close command received - connection will close");
+
+        CloseResponse response = CloseResponse.newBuilder()
+                .setMessage("Connection closing. Goodbye!")
+                .build();
+
+        return TransportProtocol.Response.newBuilder()
+                .setPayload(ByteString.copyFrom(response.toByteArray()))
+                .setStatusCode(200)
+                .putHeaders("Connection", "close"); // ← Marca para fechar
+    }
+}

src/main/java/com/mediamanager/service/delegate/handler/EchoHandler.java

@@ -1,17 +1,39 @@
 package com.mediamanager.service.delegate.handler;
 
 import com.google.protobuf.ByteString;
+import com.google.protobuf.InvalidProtocolBufferException;
+import com.mediamanager.protocol.TestProtocol.EchoCommand;      // ← Import
+import com.mediamanager.protocol.TestProtocol.EchoResponse;    // ← Import
 import com.mediamanager.protocol.TransportProtocol;
 import com.mediamanager.service.delegate.ActionHandler;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
 
 public class EchoHandler implements ActionHandler {
-    @Override
-    public TransportProtocol.Response.Builder handle(ByteString requestPayload) {
-        String payloadText = requestPayload.toStringUtf8();
-        String responseText = "Server received: " + payloadText;
+    private static final Logger logger = LogManager.getLogger(EchoHandler.class);
 
+    @Override
+    public TransportProtocol.Response.Builder handle(ByteString requestPayload)
+            throws InvalidProtocolBufferException {  // ← Pode lançar exceção
+
+        // 1. Parse Protobuf bytes → EchoCommand
+        EchoCommand command = EchoCommand.parseFrom(requestPayload);
+
+        logger.debug("Echo received: {}", command.getMessage());
+
+        // 2. Cria EchoResponse (Protobuf)
+        EchoResponse echoResponse = EchoResponse.newBuilder()
+                .setMessage(command.getMessage())
+                .setServerTimestamp(System.currentTimeMillis())
+                .build();
+
+        // 3. Serializa EchoResponse → bytes
+        ByteString responsePayload = ByteString.copyFrom(echoResponse.toByteArray());
+
+        // 4. Retorna Response
         return TransportProtocol.Response.newBuilder()
-                .setPayload(ByteString.copyFromUtf8(responseText))
+                .setPayload(responsePayload)
                 .setStatusCode(200);
+
     }
 }

src/main/java/com/mediamanager/service/delegate/handler/HeartbeatHandler.java

@@ -0,0 +1,36 @@
+package com.mediamanager.service.delegate.handler;
+
+import com.google.protobuf.ByteString;
+import com.google.protobuf.InvalidProtocolBufferException;
+import com.mediamanager.protocol.TestProtocol.HeartbeatCommand;
+import com.mediamanager.protocol.TestProtocol.HeartbeatResponse;
+import com.mediamanager.protocol.TransportProtocol;
+import com.mediamanager.service.delegate.ActionHandler;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+
+public class HeartbeatHandler implements ActionHandler {
+    private static final Logger logger = LogManager.getLogger(HeartbeatHandler.class);
+
+    @Override
+    public TransportProtocol.Response.Builder handle(ByteString requestPayload)
+            throws InvalidProtocolBufferException {
+
+        HeartbeatCommand command = HeartbeatCommand.parseFrom(requestPayload);
+
+        long serverTime = System.currentTimeMillis();
+
+        logger.debug("Heartbeat received. Client T1={}, Server T2={}",
+                command.getClientTimestamp(), serverTime);
+
+        HeartbeatResponse response = HeartbeatResponse.newBuilder()
+                .setClientTimestamp(command.getClientTimestamp())  // Echo T1
+                .setServerTimestamp(serverTime)                     // T2
+                .build();
+
+        return TransportProtocol.Response.newBuilder()
+                .setPayload(ByteString.copyFrom(response.toByteArray()))
+                .setStatusCode(200);
+
+    }
+}

src/main/java/com/mediamanager/service/ipc/IPCManager.java

@@ -185,7 +185,7 @@ public class IPCManager {
                     // Nenhum cliente conectado no momento
                     // Dorme por um curto período antes de verificar novamente
                     // Isso evita consumir CPU desnecessariamente em um loop vazio
-                    Thread.sleep(100); // 100 milissegundos
+                    Thread.sleep(1); // 1 milissegundos
                 }
 
             } catch (InterruptedException e) {
@@ -223,9 +223,16 @@ public class IPCManager {
             logger.debug("Client {} handler thread started", clientId);
 
             try {
-                TransportProtocol.Request request = readRequest(channel);
+                // LOOP: processa múltiplas requests na mesma conexão
+                while (channel.isOpen()) {
+                    TransportProtocol.Request request = readRequest(channel);
+
+                    if (request == null) {
+                        // Cliente desconectou gracefully
+                        logger.info("Client {} disconnected (end of stream)", clientId);
+                        break;
+                    }
 
-                if (request != null) {
                     logger.info("Client {} sent request {}", clientId, request.getRequestId());
 
                     // Processa usando o DelegateActionManager
@@ -233,17 +240,29 @@ public class IPCManager {
 
                     // Envia resposta de volta
                     writeResponse(channel, response);
-
                     logger.info("Client {} response sent", clientId);
-                } else {
-                    logger.warn("Client {} sent null message", clientId);
+
+                    // Verifica se é comando CLOSE
+                    String connectionHeader = response.getHeadersOrDefault("Connection", "");
+                    if ("close".equals(connectionHeader)) {
+                        logger.info("Client {} requested connection close", clientId);
+                        break; // Sai do loop e fecha
+                    }
                 }
 
+            } catch (IOException e) {
+                if (channel.isOpen()) {
+                    logger.error("IO error handling client {}", clientId, e);
+                } else {
+                    logger.debug("Client {} connection closed by peer", clientId);
+                }
             } catch (Exception e) {
-                logger.error("Error handling client {}", clientId, e);
+                logger.error("Unexpected error handling client {}", clientId, e);
             } finally {
                 try {
-                    channel.close();
+                    if (channel.isOpen()) {
+                        channel.close();
+                    }
                     logger.debug("Client {} channel closed", clientId);
                 } catch (IOException e) {
                     logger.error("Error closing client {} channel", clientId, e);

src/main/proto/test.proto

@@ -0,0 +1,30 @@
+syntax = "proto3";
+
+option java_package = "com.mediamanager.protocol";
+option java_outer_classname = "TestProtocol";
+
+package mediamanager.test;
+
+message EchoCommand {
+  string message = 1;
+}
+
+message EchoResponse {
+  string message = 1;
+  int64 server_timestamp = 2;
+}
+
+message HeartbeatCommand {
+  int64 client_timestamp = 1;
+}
+
+message HeartbeatResponse {
+  int64 client_timestamp = 1;
+  int64 server_timestamp = 2;
+}
+message CloseCommand {
+  // Vazio - apenas sinaliza fechamento
+}
+message CloseResponse {
+  string message = 1;
+}

src/main/resources/config.properties.example

@@ -23,5 +23,4 @@ ipc.pipe.path=/tmp/mediamanager
 ipc.buffer.size=8192
 
 ipc.socket.path=/tmp/mediamanager
-runtype=local
 

src/main/resources/log4j2.xml

@@ -18,40 +18,17 @@
     </Appenders>
     
     <Loggers>
-        <Logger name="com.mediamanager" level="debug" additivity="false">
+        <Logger name="com.mediamanager" level="INFO" additivity="false">
             <AppenderRef ref="Console"/>
             <AppenderRef ref="FileAppender"/>
         </Logger>
         
-        <Logger name="org.hibernate" level="info" additivity="false">
+        <Logger name="org.hibernate" level="INFO" additivity="false">
             <AppenderRef ref="Console"/>
             <AppenderRef ref="FileAppender"/>
         </Logger>
         
-        <Logger name="io.zonky.test" level="INFO" additivity="false">
-            <AppenderRef ref="Console"/>
-            <AppenderRef ref="FileAppender"/>
-        </Logger>
-
-        <!-- PostgreSQL JDBC Driver -->
-        <Logger name="org.postgresql" level="WARN" additivity="false">
-            <AppenderRef ref="Console"/>
-            <AppenderRef ref="FileAppender"/>
-        </Logger>
-
-        <!-- Processo do PostgreSQL embedded -->
-        <Logger name="org.postgresql.embedded" level="INFO" additivity="false">
-            <AppenderRef ref="Console"/>
-            <AppenderRef ref="FileAppender"/>
-        </Logger>
-
-        <!-- HikariCP Connection Pool -->
-        <Logger name="com.zaxxer.hikari" level="INFO" additivity="false">
-            <AppenderRef ref="Console"/>
-            <AppenderRef ref="FileAppender"/>
-        </Logger>
-        
-        <Root level="info">
+        <Root level="INFO">
             <AppenderRef ref="Console"/>
             <AppenderRef ref="FileAppender"/>
         </Root>