org.enodeframework.messaging.impl.DefaultMessageHandlerProxy3.kt Maven / Gradle / Ivy

Go to download
package org.enodeframework.messaging.impl

import kotlinx.coroutines.future.await
import org.enodeframework.messaging.Message
import org.enodeframework.messaging.MessageHandlerProxy3
import java.lang.invoke.MethodHandle
import java.lang.reflect.Method
import java.util.concurrent.CompletionStage
import kotlin.coroutines.intrinsics.suspendCoroutineUninterceptedOrReturn
import kotlin.reflect.jvm.kotlinFunction

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

    override suspend fun handleAsync(message1: Message, message2: Message, message3: Message) {
        //参数按照方法定义参数类型列表传递
        if (method.kotlinFunction?.isSuspend == true) {
            invokeSuspend(innerObject, message1, message2, message3)
            return
        }
        val result = methodHandle.invoke(innerObject, message1, message2, message3)
        if (result is CompletionStage<*>) {
            result.await()
        }
    }

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

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

    override fun getInnerObject(): Any {
        return innerObject
    }

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

    override fun getMethod(): Method {
        return method
    }

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