com.liubs.shadowrpcfly.server.module.ServerModule Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of shadowrpcfly-server Show documentation
There is a newer version: 1.0.3
package com.liubs.shadowrpcfly.server.module;

import com.liubs.shadowrpcfly.annotation.ShadowInterface;
import com.liubs.shadowrpcfly.server.annotation.ModuleInject;
import com.liubs.shadowrpcfly.server.annotation.ShadowModule;
import com.liubs.shadowrpcfly.server.annotation.ShadowService;
import com.liubs.shadowrpcfly.server.annotation.ShadowServiceHolder;
import com.liubs.shadowrpcfly.config.ServerConfig;
import com.liubs.shadowrpcfly.logging.Logger;
import com.liubs.shadowrpcfly.server.util.AnnotationScanner;
import com.liubs.shadowrpcfly.server.service.ServiceLookUp;
import com.liubs.shadowrpcfly.server.service.ServiceTarget;


import java.io.IOException;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.stream.Collectors;

/**
 * @author Liubsyy
 * @date 2024/1/16
 */
@ShadowModule
public class ServerModule implements IModule {
    private static final Logger logger = Logger.getLogger(ServerModule.class);

    @ModuleInject
    private SerializeModule serializeModule;

    private ServerConfig serverConfig;

    //所有服务
    private Map allRPC = new ConcurrentHashMap<>();

    public void init(ServerConfig serverConfig,List packages) {
        this.serverConfig = serverConfig;
        //初始化服务
        List> shadowServices = new ArrayList<>();

        for(String packageName : packages) {
            try {
                shadowServices.addAll(AnnotationScanner.scanAnnotations(packageName, ShadowService.class));
            } catch (IOException e) {
                logger.error("scanService err",e);
            }
        }

        for(ShadowServiceHolder ShadowServiceHolder : shadowServices) {
            ShadowService serviceAnnotation = ShadowServiceHolder.getAnnotation();
            Class serviceClass = ShadowServiceHolder.getClassz();
            try {
                Object o = serviceClass.newInstance();

                List> shadowInterfaces = Arrays.stream(serviceClass.getInterfaces())
                        .filter(c -> c.getAnnotation(ShadowInterface.class) != null)
                        .collect(Collectors.toList());

                for(Method method : serviceClass.getMethods()) {

                    if(Modifier.isStatic(method.getModifiers()) || !Modifier.isPublic(method.getModifiers())){
                        continue;
                    }
                    boolean existInterface = false;
                    for(Class shadowInterface : shadowInterfaces) {
                        try {
                            shadowInterface.getMethod(method.getName(),method.getParameterTypes());
                            existInterface = true;
                        } catch (NoSuchMethodException e) {}
                    }
                    if(!existInterface) {
                        continue;
                    }

                    ServiceLookUp serviceLookUp = new ServiceLookUp();
                    serviceLookUp.setServiceName(serviceAnnotation.serviceName());
                    serviceLookUp.setMethodName(method.getName());
                    serviceLookUp.setParamTypes(method.getParameterTypes());

                    ServiceTarget serviceTarget = new ServiceTarget(o,method);
                    this.addRPCInterface(serviceLookUp,serviceTarget);
                }

            } catch (InstantiationException | IllegalAccessException e) {
                throw new RuntimeException(e);
            }
        }
    }

    public void addRPCInterface(ServiceLookUp lookUp,ServiceTarget obj) {
        allRPC.put(lookUp,obj);
    }

    public ServiceTarget getRPC(ServiceLookUp lookUp) {
        return allRPC.get(lookUp);
    }


    public ServerConfig getServerConfig() {
        return serverConfig;
    }
}