org.enodeframework.messaging.impl.MessageHandlerProxy2.kt 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.messaging.IMessage
import org.enodeframework.messaging.IMessageHandlerProxy2
import java.lang.invoke.MethodHandle
import java.lang.reflect.Method
import kotlin.coroutines.intrinsics.suspendCoroutineUninterceptedOrReturn
import kotlin.reflect.jvm.kotlinFunction

/**
 * @author [email protected]
 */
class MessageHandlerProxy2 : IMessageHandlerProxy2 {
    private lateinit var innerObject: Any
    private lateinit var methodHandle: MethodHandle
    private lateinit var method: Method

    override suspend fun handleAsync(message1: IMessage, message2: IMessage) {
        val methodParameterTypes = method.parameterTypes
        if (method.kotlinFunction?.isSuspend == true) {
            if (methodParameterTypes[0].isAssignableFrom(message1.javaClass)) {
                invokeSuspend(getInnerObject(), message1, message2)
            } else {
                invokeSuspend(getInnerObject(), message2, message1)
            }
            return
        }
        if (methodParameterTypes[0].isAssignableFrom(message1.javaClass)) {
            methodHandle.invoke(getInnerObject(), message1, message2)
        } else {
            methodHandle.invoke(getInnerObject(), message2, message1)
        }
    }

    private suspend fun invokeSuspend(obj: Any, message1: IMessage, message2: IMessage): Any? = suspendCoroutineUninterceptedOrReturn { continuation ->
        methodHandle.invoke(obj, message1, message2, continuation)
    }

    override fun getInnerObject(): Any {
        return innerObject
    }

    override fun setInnerObject(innerObject: Any) {
        this.innerObject = innerObject;
    }

    override fun setMethodHandle(methodHandle: MethodHandle) {
        this.methodHandle = methodHandle
    }

    override fun getMethod(): Method {
        return method
    }

    override fun setMethod(method: Method) {
        this.method = method
    }
}