Fix. Input type

Author: sleipnir

src/main/java/io/eigr/spawn/internal/Entity.java

@@ -1,6 +1,8 @@
 package io.eigr.spawn.internal;
 
 import io.eigr.functions.protocol.actors.ActorOuterClass;
+import io.eigr.spawn.api.Value;
+import io.eigr.spawn.api.actors.ActorContext;
 import io.eigr.spawn.api.annotations.*;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -158,6 +160,15 @@ public Method getMethod() {
         }
 
         public Class getInputType() {
+            int arity = method.getParameterTypes().length;
+
+            if (arity == 2 && Objects.isNull(inputType)) {
+                for (Class<?> parameterType : method.getParameterTypes()) {
+                    if (!inputType.isAssignableFrom(ActorContext.class)) {
+                        return parameterType;
+                    }
+                }
+            }
             return inputType;
         }
 

src/main/java/io/eigr/spawn/internal/handlers/ActorServiceHandler.java

@@ -135,6 +135,7 @@ private Optional<Value> callAction(String system, String actor, String commandNa
 
                 final Method actorMethod = entityMethod.getMethod();
                 Class inputType = entityMethod.getInputType();
+                log.debug("Action input type is: {}", inputType);
 
                 ActorContext actorContext;
                 if (context.hasState()) {
@@ -147,7 +148,7 @@ private Optional<Value> callAction(String system, String actor, String commandNa
                 if (inputType.isAssignableFrom(ActorContext.class)) {
                     return Optional.of((Value) actorMethod.invoke(actorRef, actorContext));
                 } else {
-                    final Object unpack = value.unpack(entityMethod.getInputType());
+                    final Object unpack = value.unpack(inputType);
                     return Optional.of((Value) actorMethod.invoke(actorRef, unpack, actorContext));
                 }
             } catch (IllegalAccessException e) {