com.github.longdt.vertxservice.processor.SingleParamMethodProxyHandlerWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vertx-service4j Show documentation

A lightweight replacement for vertx-service-proxy. This library is highly optimized for vertx java

There is a newer version: 1.1.1

package com.github.longdt.vertxservice.processor;

import com.squareup.javapoet.CodeBlock;
import io.vertx.serviceproxy.HelperUtils;

import javax.lang.model.element.ExecutableElement;
import javax.lang.model.type.DeclaredType;
import javax.lang.model.util.Types;

public class SingleParamMethodProxyHandlerWriter extends MethodProxyHandlerWriter {
    public SingleParamMethodProxyHandlerWriter(Types types) {
        super(types);
    }

    @Override
    public void writeInvocation(CodeBlock.Builder codeBlockBuilder, ExecutableElement element) {
        codeBlockBuilder.beginControlFlow("case $S:", element.getSimpleName().toString());
        var param = element.getParameters().get(0);
        var paramType = param.asType();
        boolean needUnwrap = SupportedTypes.isCollectionType(types, paramType);
        if (needUnwrap) {
            codeBlockBuilder.addStatement("var $L = (($T) $L.body()).unwrap()", param.getSimpleName(),
                    MethodProxyWriter.getRequestParamType(types, paramType),
                    Constant.MESSAGE_VARIABLE);
        } else {
            codeBlockBuilder.addStatement("var $L = ($T) $L.body()", param.getSimpleName(), paramType, Constant.MESSAGE_VARIABLE);
        }
        codeBlockBuilder.add("$L.$L($L)\n", Constant.SERVICE_VARIABLE, element.getSimpleName().toString(), param.getSimpleName());
        codeBlockBuilder.indent().indent()
                .beginControlFlow(".onComplete(res ->")
                .beginControlFlow("if (res.failed())")
                .addStatement("$T.manageFailure($L, res.cause(), $L)", HelperUtils.class, Constant.MESSAGE_VARIABLE, Constant.INCLUDE_DEBUG_INFO_VARIABLE)
                .nextControlFlow("else");

        var resultType = ((DeclaredType) element.getReturnType()).getTypeArguments().get(0);
        addReplyHandler(codeBlockBuilder, resultType);

        codeBlockBuilder.endControlFlow()
                .unindent().add("});\n")
                .unindent().unindent();
        codeBlockBuilder.addStatement("break");
        codeBlockBuilder.endControlFlow();
    }
}