基于Springboot+Netty实现rpc的方法 附demo

<dependency>

     <groupId>org.springframework.boot</groupId>

     <artifactId>spring-boot-starter-web</artifactId>

     <version> 2.3 . 2 .RELEASE</version>

</dependency>

<dependency>

     <groupId>io.netty</groupId>

     <artifactId>netty-all</artifactId>

     <version> 4.1 . 48 .Final</version>

</dependency>

<dependency>

     <groupId>com.alibaba</groupId>

     <artifactId>fastjson</artifactId>

     <version> 1.2 . 58 </version>

</dependency>

<dependency>

     <groupId>org.slf4j</groupId>

     <artifactId>slf4j-log4j12</artifactId>

     <version> 2.0 . 0 -alpha1</version>

</dependency>

@Data

@AllArgsConstructor

public class RequestModel {

     private String requestId;

     private String serviceName;

     private String methodName;

     private Class[] paramTypes;

     private Object[] paramValues;

}

@Data

@AllArgsConstructor

public class ResponseModel {

     private String responseId;

     private String serviceName;

     private String methodName;

     private String code;

     private String data;

}

@Override

protected void channelRead0(ChannelHandlerContext ctx, String msg) {

     StringBuilder sb = null ;

     RequestModel result = null ;

     try {

         // 报文解析处理

         sb = new StringBuilder();

         result = JSON.parseObject(msg, RequestModel. class );

         requestId = result.getRequestId();

         String serviceName = result.getServiceName();

         String methodName = result.getMethodName();

         Class[] paramType = result.getParamTypes();

         Object[] paramValue = result.getParamValues();

         System.out.println(serviceName + "  " + methodName);

         String substring = serviceName.substring(serviceName.lastIndexOf( "." ) + 1 );

         String s = substring.substring( 0 , 1 ).toLowerCase() + substring.substring( 1 );

         Object serviceObject = applicationContext.getBean(s);

         Method method = Class.forName(serviceName).getMethod(methodName, paramType);

         Object returnValue = method.invoke(serviceObject, paramValue);

         ResponseModel responseModel = new ResponseModel(requestId,serviceName,methodName, "200" ,JSON.toJSONString(returnValue));

         sb.append(JSON.toJSONString(responseModel));

         sb.append( "\n" );

         System.out.println(sb.toString());

         ctx.writeAndFlush(sb);

     } catch (Exception e) {

         ResponseModel responseModel = new ResponseModel(requestId, "" , "" , "500" ,e.getMessage());

         String errorCode = JSON.toJSONString(responseModel)+ "\n" ;

         log.error(errorCode);

         ctx.writeAndFlush(errorCode);

         log.error( "报文解析失败: " + e.getMessage());

     }

}

@Override

    protected void channelRead0(ChannelHandlerContext ctx, String msg) {

        System.out.println( "收到服务端消息: " + msg);

        ResponseModel responseModel = JSON.parseObject(msg,ResponseModel. class );

        String responseId = responseModel.getResponseId();

        Promise promise = LocalPromise.promiseMap.remove(responseId);

        if (promise != null ){

            String code = responseModel.getCode();

            if (code.equals( "200" )){

                promise.setSuccess(responseModel.getData());

            } else {

                promise.setFailure( new RuntimeException(responseModel.getData()));

            }

        }

    }

private <T> T getProxyService(Class<T> serviceClass) {

         Object service = Proxy.newProxyInstance(serviceClass.getClassLoader(), new Class[]{serviceClass}, new InvocationHandler() {

             @Override

             public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {

                 Channel channel = NettyClient.getChannel(host, port);

                 RequestModel requestModel = new RequestModel( "100001" , method.getDeclaringClass().getName(), method.getName(), method.getParameterTypes(), args);

                 channel.writeAndFlush(JSON.toJSONString(requestModel) + "\n" );

                 Promise promise = new DefaultPromise(channel.eventLoop());

                 LocalPromise.promiseMap.put(requestModel.getRequestId(), promise);

                 System.out.println(LocalPromise.promiseMap+ ">>>>>>>>>>>>" );

                 promise.await();

                 if (promise.isSuccess()) {

                     Class<?> returnType = method.getReturnType();

                     return JSON.toJavaObject(JSON.parseObject(promise.getNow()+ "" ),returnType);

                 } else {

                     System.out.println(promise.cause());

                     return promise.cause();

                 }

             }

         });

         return (T) service;

     }