org.enodeframework.messaging.impl.MessageHandlerProxy3 Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of enode Show documentation
The enodeframework core implementation.
There is a newer version: 1.1.10
package org.enodeframework.messaging.impl;

import org.enodeframework.common.container.ObjectContainer;
import org.enodeframework.messaging.IMessage;
import org.enodeframework.messaging.IMessageHandlerProxy3;

import java.lang.invoke.MethodHandle;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;
import java.util.concurrent.CompletableFuture;

/**
 * @author [email protected]
 */
public class MessageHandlerProxy3 implements IMessageHandlerProxy3 {

    private Class handlerType;
    private Object handler;
    private MethodHandle methodHandle;
    private Method method;
    private Class[] methodParameterTypes;

    @Override
    public CompletableFuture handleAsync(IMessage message1, IMessage message2, IMessage message3) {
        CompletableFuture future = new CompletableFuture<>();
        List> parameterTypes = Arrays.asList(methodParameterTypes);
        List params = new ArrayList<>();
        params.add(message1);
        params.add(message2);
        params.add(message3);
        //排序参数
        params.sort(Comparator.comparingInt(m -> getMessageParameterIndex(parameterTypes, m))
        );
        try {
            //参数按照方法定义参数类型列表传递
            Object result = methodHandle.invoke(getInnerObject(), params.get(0), params.get(1), params.get(2));
            if (result instanceof CompletableFuture) {
                return (CompletableFuture) result;
            }
            future.complete(true);
        } catch (Throwable throwable) {
            future.completeExceptionally(throwable);
        }
        return future;
    }

    private int getMessageParameterIndex(List> methodParameterTypes, IMessage message) {
        int i = 0;
        for (Class paramType : methodParameterTypes) {
            if (paramType.isAssignableFrom(message.getClass())) {
                return i;
            }
            i++;
        }
        return i;
    }

    @Override
    public Object getInnerObject() {
        if (handler != null) {
            return handler;
        }
        handler = ObjectContainer.INSTANCE.resolve(handlerType);
        return handler;
    }

    @Override
    public void setHandlerType(Class handlerType) {
        this.handlerType = handlerType;
    }

    @Override
    public void setMethodHandle(MethodHandle methodHandle) {
        this.methodHandle = methodHandle;
    }

    @Override
    public Method getMethod() {
        return method;
    }

    @Override
    public void setMethod(Method method) {
        this.method = method;
        methodParameterTypes = method.getParameterTypes();
    }
}