Target

pixelib · Aug 24, 2023 · d64ebc8 · d64ebc8
1 parent c04219c
commit d64ebc8
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 11 deletions.
diff --git a/rpcnis-common/src/main/java/com/rpcnis/base/RpcTransport.java b/rpcnis-common/src/main/java/com/rpcnis/base/RpcTransport.java
@@ -23,5 +23,5 @@ public interface RpcTransport extends Closeable {
      *                  if the packet was not handled, the transport implementation should continue processing the packet.
      *                  the transport implementation should call the onReceive function, regardless of the ReadStatus.
      */
-    void onReceive(Function<byte[], ReadStatus> onReceive);
+    void onReceive(Direction target, Function<byte[], ReadStatus> onReceive);
 }
diff --git a/rpcnis-core/src/main/java/com/rpcnis/core/transport/TransportHandler.java b/rpcnis-core/src/main/java/com/rpcnis/core/transport/TransportHandler.java
@@ -4,6 +4,7 @@
 import com.rpcnis.base.RpcTransport;
 import com.rpcnis.base.enums.ReadStatus;
 import com.rpcnis.core.executor.ImplementationWrapper;
+import com.rpcnis.core.models.InvocationDescriptor;
 import com.rpcnis.core.trackers.IncomingInvocationTracker;
 import com.rpcnis.core.transport.packets.InvocationPacket;
 
@@ -25,7 +26,7 @@ public TransportHandler(RpcSerializer serializer, RpcTransport transport, Incomi
 
     private ReadStatus onReceive(byte[] bytes) {
         // deserialize the packet
-        InvocationPacket packet = serializer.deserialize(bytes, InvocationPacket.class);
+        InvocationDescriptor invocationDescriptor = serializer.deserialize(bytes, InvocationDescriptor.class);
 
         // get the invocation handler for this packet
         Collection<ImplementationWrapper> implementations = incomingInvocationTracker.getImplementations().get(packet.getProcedure());

diff --git a/rpcnis-core/src/main/java/com/rpcnis/core/transport/packets/InvocationPacket.java b/rpcnis-core/src/main/java/com/rpcnis/core/transport/packets/InvocationPacket.java