Merge pull request #5 from gmbrax/feature/IPC-Unix-Socket

Feature/ipc unix socket
This commit is contained in:
Gustavo Henrique Miranda 2025-11-20 09:16:50 -03:00 committed by GitHub
commit 77259308c6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
7 changed files with 251 additions and 17 deletions

55
pom.xml
View File

@ -78,9 +78,22 @@
<version>${junit.version}</version> <version>${junit.version}</version>
<scope>test</scope> <scope>test</scope>
</dependency> </dependency>
<dependency>
<groupId>com.google.protobuf</groupId>
<artifactId>protobuf-java</artifactId>
<version>4.32.0</version>
</dependency>
</dependencies> </dependencies>
<build> <build>
<extensions>
<extension>
<groupId>kr.motd.maven</groupId>
<artifactId>os-maven-plugin</artifactId>
<version>1.7.1</version>
</extension>
</extensions>
<plugins> <plugins>
<plugin> <plugin>
<groupId>org.apache.maven.plugins</groupId> <groupId>org.apache.maven.plugins</groupId>
@ -91,12 +104,54 @@
<target>17</target> <target>17</target>
</configuration> </configuration>
</plugin> </plugin>
<plugin> <plugin>
<groupId>org.apache.maven.plugins</groupId> <groupId>org.apache.maven.plugins</groupId>
<artifactId>maven-surefire-plugin</artifactId> <artifactId>maven-surefire-plugin</artifactId>
<version>3.2.5</version> <version>3.2.5</version>
</plugin> </plugin>
<!-- Protocol Buffers Plugin -->
<plugin>
<groupId>org.xolstice.maven.plugins</groupId>
<artifactId>protobuf-maven-plugin</artifactId>
<version>0.6.1</version>
<configuration>
<protocExecutable>/usr/bin/protoc</protocExecutable>
<protoSourceRoot>${project.basedir}/src/main/proto</protoSourceRoot>
<outputDirectory>${project.build.directory}/generated-sources/protobuf/java</outputDirectory>
<clearOutputDirectory>false</clearOutputDirectory>
</configuration>
<executions>
<execution>
<goals>
<goal>compile</goal>
</goals>
</execution>
</executions>
</plugin>
<!-- Build Helper Plugin - Adiciona generated sources ao classpath -->
<plugin>
<groupId>org.codehaus.mojo</groupId>
<artifactId>build-helper-maven-plugin</artifactId>
<version>3.5.0</version>
<executions>
<execution>
<id>add-source</id>
<phase>generate-sources</phase>
<goals>
<goal>add-source</goal>
</goals>
<configuration>
<sources>
<source>${project.build.directory}/generated-sources/protobuf/java</source>
</sources>
</configuration>
</execution>
</executions>
</plugin>
<!-- Maven Shade Plugin - Creates executable JAR with all dependencies --> <!-- Maven Shade Plugin - Creates executable JAR with all dependencies -->
<plugin> <plugin>
<groupId>org.apache.maven.plugins</groupId> <groupId>org.apache.maven.plugins</groupId>

View File

@ -1,5 +1,6 @@
package com.mediamanager; package com.mediamanager;
import com.mediamanager.service.delegate.DelegateActionManager;
import com.mediamanager.service.ipc.IPCManager; import com.mediamanager.service.ipc.IPCManager;
import org.apache.logging.log4j.LogManager; import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger; import org.apache.logging.log4j.Logger;
@ -14,6 +15,7 @@ public class MediaManagerApplication {
private static final Logger logger = LogManager.getLogger(MediaManagerApplication.class); private static final Logger logger = LogManager.getLogger(MediaManagerApplication.class);
private static Properties config; private static Properties config;
private static DatabaseManager databaseManager; private static DatabaseManager databaseManager;
private static DelegateActionManager actionManager;
private static IPCManager ipcManager; private static IPCManager ipcManager;
public static void main(String[] args) { public static void main(String[] args) {
@ -24,8 +26,9 @@ public class MediaManagerApplication {
loadConfiguration(); loadConfiguration();
databaseManager = new DatabaseManager(config); databaseManager = new DatabaseManager(config);
databaseManager.init(); databaseManager.init();
actionManager = new DelegateActionManager();
ipcManager = new IPCManager(config); actionManager.start();
ipcManager = new IPCManager(config,actionManager);
ipcManager.init(); ipcManager.init();
// TODO: Start application services // TODO: Start application services
@ -51,6 +54,9 @@ public class MediaManagerApplication {
} }
} }
if (actionManager != null) {
actionManager.stop();
}
logger.info("MediaManager Core shutdown successfully"); logger.info("MediaManager Core shutdown successfully");
logger.info("Goodbye!"); logger.info("Goodbye!");

View File

@ -0,0 +1,11 @@
package com.mediamanager.service.delegate;
import com.google.protobuf.ByteString;
import com.google.protobuf.InvalidProtocolBufferException;
import com.mediamanager.protocol.TransportProtocol;
@FunctionalInterface
public interface ActionHandler {
TransportProtocol.Response.Builder handle(ByteString requestPayload) throws InvalidProtocolBufferException;
}

View File

@ -0,0 +1,60 @@
package com.mediamanager.service.delegate;
import com.google.protobuf.ByteString;
import com.mediamanager.protocol.TransportProtocol;
import com.mediamanager.service.delegate.handler.EchoHandler;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import java.util.HashMap;
import java.util.Map;
public class DelegateActionManager {
private static final Logger logger = LogManager.getLogger(DelegateActionManager.class);
private final Map<String, ActionHandler> handlerRegistry;
public DelegateActionManager() {
logger.debug("DelegateActionManager created");
this.handlerRegistry = new HashMap<>();
registerHandlers();
}
private void registerHandlers() {
handlerRegistry.put("echo",new EchoHandler());
}
public void start(){
logger.info("DelegateActionManager started");
}
public void stop(){
logger.info("DelegateActionManager stopped");
}
public TransportProtocol.Response ProcessedRequest(TransportProtocol.Request request){
String requestId = request.getRequestId();
logger.info("Processing request: {}", requestId);
String action = request.getHeadersMap().getOrDefault("action", "unknown");
ActionHandler handler = handlerRegistry.get(action);
TransportProtocol.Response.Builder responseBuilder;
if (handler == null) {
logger.warn("No handler found for action: {}", action);
responseBuilder = TransportProtocol.Response.newBuilder()
.setStatusCode(404) // 404 Not Found
.setPayload(ByteString.copyFromUtf8("Error: Action '" + action + "' not found."));
} else{
try {
logger.debug("Delegating action '{}' to handler...", action);
responseBuilder = handler.handle(request.getPayload());
}catch (Exception e) {
logger.error("Handler for action '{}' threw an exception:", action, e);
responseBuilder = TransportProtocol.Response.newBuilder()
.setStatusCode(500) // 500 Internal Server Error
.setPayload(ByteString.copyFromUtf8("Error: " + e.getMessage()));
}
}
return responseBuilder.setRequestId(requestId).build();
}
}

View File

@ -0,0 +1,17 @@
package com.mediamanager.service.delegate.handler;
import com.google.protobuf.ByteString;
import com.mediamanager.protocol.TransportProtocol;
import com.mediamanager.service.delegate.ActionHandler;
public class EchoHandler implements ActionHandler {
@Override
public TransportProtocol.Response.Builder handle(ByteString requestPayload) {
String payloadText = requestPayload.toStringUtf8();
String responseText = "Server received: " + payloadText;
return TransportProtocol.Response.newBuilder()
.setPayload(ByteString.copyFromUtf8(responseText))
.setStatusCode(200);
}
}

View File

@ -1,10 +1,11 @@
package com.mediamanager.service.ipc; package com.mediamanager.service.ipc;
import com.mediamanager.protocol.TransportProtocol;
import com.mediamanager.service.delegate.DelegateActionManager;
import org.apache.logging.log4j.LogManager; import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger; import org.apache.logging.log4j.Logger;
import java.io.File;
import java.io.IOException; import java.io.IOException;
import java.net.StandardProtocolFamily; import java.net.StandardProtocolFamily;
import java.net.UnixDomainSocketAddress; import java.net.UnixDomainSocketAddress;
@ -26,6 +27,8 @@ import java.util.concurrent.atomic.AtomicInteger;
public class IPCManager { public class IPCManager {
private final Properties configuration; private final Properties configuration;
private static final Logger logger = LogManager.getLogger(IPCManager.class); private static final Logger logger = LogManager.getLogger(IPCManager.class);
private final DelegateActionManager actionManager;
private Path socketPath; private Path socketPath;
private UnixDomainSocketAddress socketAddress; private UnixDomainSocketAddress socketAddress;
private ServerSocketChannel serverChannel; private ServerSocketChannel serverChannel;
@ -35,8 +38,9 @@ public class IPCManager {
private final AtomicInteger clientIdCounter = new AtomicInteger(0); private final AtomicInteger clientIdCounter = new AtomicInteger(0);
public IPCManager(Properties config){ public IPCManager(Properties config, DelegateActionManager actionManager){
configuration = config; configuration = config;
this.actionManager = actionManager;
logger.debug("IPCManager created with configuration:"); logger.debug("IPCManager created with configuration:");
@ -219,24 +223,25 @@ public class IPCManager {
logger.debug("Client {} handler thread started", clientId); logger.debug("Client {} handler thread started", clientId);
try { try {
// TODO: No próximo passo, vamos implementar: TransportProtocol.Request request = readRequest(channel);
// 1. Ler mensagens JSON do SocketChannel
// 2. Parsear o JSON para objetos Java
// 3. Processar a requisição
// 4. Criar uma resposta JSON
// 5. Escrever a resposta de volta no SocketChannel
// Por enquanto, apenas mantém a conexão aberta brevemente if (request != null) {
// para testar que o sistema de aceitação e threads está funcionando logger.info("Client {} sent request {}", clientId, request.getRequestId());
Thread.sleep(100);
logger.debug("Client {} processing complete", clientId); // Processa usando o DelegateActionManager
TransportProtocol.Response response = actionManager.ProcessedRequest(request);
// Envia resposta de volta
writeResponse(channel, response);
logger.info("Client {} response sent", clientId);
} else {
logger.warn("Client {} sent null message", clientId);
}
} catch (Exception e) { } catch (Exception e) {
logger.error("Error handling client {}", clientId, e); logger.error("Error handling client {}", clientId, e);
} finally { } finally {
// SEMPRE fecha o canal quando terminar
// O finally garante que isso acontece mesmo se houver exceção
try { try {
channel.close(); channel.close();
logger.debug("Client {} channel closed", clientId); logger.debug("Client {} channel closed", clientId);
@ -245,6 +250,67 @@ public class IPCManager {
} }
} }
} }
private TransportProtocol.Request readRequest(SocketChannel channel) throws IOException {
// Primeiro, o tamanho da mensagem (4 bytes = int32)
java.nio.ByteBuffer sizeBuffer = java.nio.ByteBuffer.allocate(4);
int bytesRead = 0;
while (bytesRead < 4) {
int read = channel.read(sizeBuffer);
if (read == -1) {
logger.debug("Client disconnected before sending size");
return null;
}
bytesRead += read;
}
sizeBuffer.flip();
int messageSize = sizeBuffer.getInt();
logger.debug("Expecting message of {} bytes", messageSize);
// Validação básica de segurança
if (messageSize <= 0 || messageSize > 1024 * 1024) { // Max 1MB
throw new IOException("Invalid message size: " + messageSize);
}
// Agora a mensagem completa
java.nio.ByteBuffer messageBuffer = java.nio.ByteBuffer.allocate(messageSize);
bytesRead = 0;
while (bytesRead < messageSize) {
int read = channel.read(messageBuffer);
if (read == -1) {
throw new IOException("Client disconnected while reading message");
}
bytesRead += read;
}
messageBuffer.flip();
// Deserializa o Protocol Buffers
byte[] messageBytes = new byte[messageSize];
messageBuffer.get(messageBytes);
return TransportProtocol.Request.parseFrom(messageBytes);
}
private void writeResponse(SocketChannel channel, TransportProtocol.Response response) throws IOException {
byte[] messageBytes = response.toByteArray();
int messageSize = messageBytes.length;
logger.debug("Writing response of {} bytes", messageSize);
java.nio.ByteBuffer buffer = java.nio.ByteBuffer.allocate(4 + messageSize);
buffer.putInt(messageSize);
buffer.put(messageBytes);
buffer.flip();
while (buffer.hasRemaining()) {
channel.write(buffer);
}
logger.debug("Response written successfully");
}
} }
} }

View File

@ -0,0 +1,19 @@
syntax = "proto3";
option java_package = "com.mediamanager.protocol";
option java_outer_classname = "TransportProtocol";
package mediamanager;
message Request {
string request_id = 1;
bytes payload = 2;
map<string, string> headers = 3;
}
message Response {
string request_id = 1;
int32 status_code = 2;
bytes payload = 3;
map<string, string> headers = 4;
}