com.github.netty.springboot.server.NRpcProtocolSpringAdapter Maven / Gradle / Ivy

Go to download
package com.github.netty.springboot.server;

import com.github.netty.annotation.NRpcParam;
import com.github.netty.annotation.NRpcService;
import com.github.netty.core.AbstractNettyServer;
import com.github.netty.core.util.*;
import com.github.netty.protocol.NRpcProtocol;
import com.github.netty.springboot.SpringUtil;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.beans.factory.config.ConfigurableBeanFactory;
import org.springframework.web.bind.annotation.*;

import java.lang.reflect.Method;
import java.util.Collection;
import java.util.function.Function;

/**
 * Internal RPC protocol registry (spring adapter)
 *
 * @author wangzihao
 */
public class NRpcProtocolSpringAdapter extends NRpcProtocol implements BeanPostProcessor {
    private final ClassFileMethodToParameterNamesFunction classFileMethodToParameterNamesFunction = new ClassFileMethodToParameterNamesFunction();
    private final AnnotationMethodToParameterNamesFunction annotationMethodToParameterNamesFunction = new AnnotationMethodToParameterNamesFunction(
            NRpcParam.class, RequestParam.class, RequestBody.class, RequestHeader.class,
            PathVariable.class, CookieValue.class, RequestPart.class);
    private final ConfigurableBeanFactory beanFactory;

    public NRpcProtocolSpringAdapter(ConfigurableBeanFactory beanFactory, ApplicationX application) {
        super(application);
        this.beanFactory = beanFactory;
        beanFactory.addBeanPostProcessor(this);
    }

    @Override
    public  void onServerStart(T server) throws Exception {
        Collection list = super.getApplication().getBeanForAnnotation(NRpcService.class);
        getAnnotationMethodToMethodNameFunction().getMethodNameAnnotationClasses().add(RequestMapping.class);

        for (Object serviceImpl : list) {
            if (super.existInstance(serviceImpl)) {
                continue;
            }

            RequestMapping requestMapping = getRequestMapping(serviceImpl);
            if (requestMapping != null) {
                String requestMappingName = getRequestMappingName(requestMapping);
                Function methodToParameterNamesFunction = getMethodToParameterNamesFunction(serviceImpl);
                super.addInstance(serviceImpl, requestMappingName, methodToParameterNamesFunction);
            } else {
                super.addInstance(serviceImpl);
            }
        }
        super.onServerStart(server);
    }

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        if (SpringUtil.isSingletonBean(beanFactory, beanName)) {
            getApplication().addSingletonBean(bean, beanName, false);
        }
        return bean;
    }

    protected Function getMethodToParameterNamesFunction(Object serviceImpl) {
        if (ReflectUtil.hasParameterAnnotation(serviceImpl.getClass(), annotationMethodToParameterNamesFunction.getParameterAnnotationClasses())) {
            return annotationMethodToParameterNamesFunction;
        } else {
            return classFileMethodToParameterNamesFunction;
        }
    }

    protected String getRequestMappingName(RequestMapping requestMapping) {
        //Get the service name
        String requestMappingName = requestMapping.name();
        String[] values = requestMapping.value();
        String[] paths = requestMapping.path();
        if (StringUtil.isEmpty(requestMappingName) && values.length > 0) {
            requestMappingName = values[0];
        }
        if (StringUtil.isEmpty(requestMappingName) && paths.length > 0) {
            requestMappingName = paths[0];
        }
        if (StringUtil.isEmpty(requestMappingName)) {
            throw new IllegalArgumentException("RequestMapping isEmpty!");
        }
        return requestMappingName;
    }

    public AnnotationMethodToParameterNamesFunction getAnnotationMethodToParameterNamesFunction() {
        return annotationMethodToParameterNamesFunction;
    }

    protected RequestMapping getRequestMapping(Object serviceImpl) {
        Class annotationOnClass = ReflectUtil.findClassByAnnotation(serviceImpl.getClass(), RequestMapping.class);
        if (annotationOnClass == null) {
            return null;
        }
        //In the case of server-side annotations, the information is taken from the annotations
        return (RequestMapping) annotationOnClass.getDeclaredAnnotation(RequestMapping.class);
    }

}