消费者有一个需要注意的地方,就是调用需要透明,也就是说,框架使用者不用关心底层的网络实现。这里我们可以使用 JDK 的动态代理来实现这个目的。
思路:客户端调用代理方法,返回一个实现了 HelloService 接口的代理对象,调用代理对象的方法,返回结果。我们需要在代理中做手脚,当调用代理方法的时候,我们需要初始化 Netty 客户端,还需要向服务端请求数据,并返回数据。首先创建代理相关的类:
RPCConsumer
package com.lwl.client;
import com.lwl.common.JSONSerializer;
import com.lwl.common.RpcEncoder;
import com.lwl.common.RpcRequest;
import com.lwl.handler.UserClientHandler;
import io.netty.bootstrap.Bootstrap;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelOption;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.codec.string.StringDecoder;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.util.UUID;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
/**
* 消费者
*/
public class RPCConsumer {
//1、创建线程池对象 处理自定义事件
//并且当前线程池的线程数以当前系统的cpu核数
private static ExecutorService executorService =
Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());
//2、声明一个自定义事件处理器 UserClientHandler
public static UserClientHandler userClientHandler;
//初始化客户端 创建连接池 bootStrap 甚至bootStrap 连接服务器
public static void initClient() throws InterruptedException {
//初始化 UserClientHandler
userClientHandler = new UserClientHandler();
//创建连接池对象
NioEventLoopGroup group = new NioEventLoopGroup();
//创建客户端引导对象
Bootstrap bootstrap = new Bootstrap();
//配置引导对象
bootstrap.group(group)
//设置对象为nio
.channel(NioSocketChannel.class)
//设置请求协议为tcp
.option(ChannelOption.TCP_NODELAY,true)
//监听handler 并初始化
.handler(new ChannelInitializer<SocketChannel>() {
@Override
protected void initChannel(SocketChannel socketChannel) throws Exception {
//获取channelPipeline
ChannelPipeline pipeline = socketChannel.pipeline();
//设置编码
pipeline.addLast(new StringDecoder());
pipeline.addLast( new RpcEncoder(RpcRequest.class, new JSONSerializer()));
//添加自定义事件处理器
pipeline.addLast(userClientHandler);
}
});
bootstrap.connect("127.0.0.1",8999).sync();
}
//定义一个方法 使用jdk动态代理创建对象
//serverClass 接口类型 根据哪个接口生成子类代理对象
public static Object createProxy(final Class<?> serviceClass){
return Proxy.newProxyInstance(Thread.currentThread().getContextClassLoader(),
new Class[]{serviceClass}, new InvocationHandler() {
public Object invoke(Object o, Method method, Object[] objects) throws Throwable {
//1、初始化客户端client
if (null == userClientHandler){
initClient();
}
//2、给userClientHandler 设置参数
RpcRequest rpcRequest = new RpcRequest();
rpcRequest.setClassName(serviceClass.getName());
rpcRequest.setMethodName(method.getName());
rpcRequest.setParameters(objects);
rpcRequest.setParameterTypes(method.getParameterTypes());
rpcRequest.setRequestId(UUID.randomUUID().toString());
userClientHandler.setParam(rpcRequest);
//3、使用线程池 开启一个线程处理call() 写操作 并返回结果
Object result = executorService.submit(userClientHandler).get();
//返回结果
return result;
}
}
);
}
}
自定义事件处理器 UserClientHandler
package com.lwl.handler;
import com.lwl.common.RpcRequest;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import java.util.concurrent.Callable;
/**
* 自定义事件处理器
*/
public class UserClientHandler extends ChannelInboundHandlerAdapter implements Callable {
//定义成员变量
//事件处理器上下文对象 (存储handler信息 写操作)
private ChannelHandlerContext context;
//记录服务器返回的数据
private String result;
//记录发送给服务器的数据
private RpcRequest rpcRequest;
//事件channelActive 客户端和服务器连接时,该方法自动执行
@Override
public void channelActive(ChannelHandlerContext ctx) throws Exception {
//初始化
this.context = ctx;
}
//实现channelRead 当服务器督导数据时 方法自动执行
@Override
public synchronized void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
//将读到的数据msg 设置为成员变量的值
result = msg.toString();
//唤醒 写操作
notify();
}
//将客户端的数据写到服务器
public synchronized Object call() throws Exception {
context.writeAndFlush(rpcRequest);
wait();
return result;
}
//设置参数的方法
public void setParam(RpcRequest rpcRequest){
this.rpcRequest =rpcRequest;
}
}
启动引导类ConsumerBoot
package com.lwl.boot;
import com.lwl.client.RPCConsumer;
import com.lwl.service.IUserService;
public class ConsumerBoot {
//参数定义
private static final String PROVIDER_NAME = "UserService#sayHello#";
public static void main(String[] args) throws InterruptedException {
//1、创建代理对象
IUserService service = (IUserService) RPCConsumer.createProxy(IUserService.class);
//2、循环给服务器写数据
while (true){
String result = service.sayHello("are you ok ----");
System.out.println("-->{} "+result);
Thread.sleep(2000);
}
}
}