Netty 笔记——手把手教你如何写一个 RPC 程序

2401_84103936

于 2024-04-30 06:37:44 发布

阅读量464

点赞数 9

分类专栏：程序员文章标签：笔记 rpc qt

本文链接：https://blog.csdn.net/2401_84103936/article/details/138332714

版权

程序员专栏收录该内容

222 篇文章 0 订阅

订阅专栏

参数的值

private Object[] paramValues;

/**

参数的类型

private Class<?>[] paramTypes;

}

2、Provider 服务端

==================================================================================

Provider 服务端主要由三个类构成

Provider: 服务端的入口类，启动 Netty 服务
ProviderHandler: 根据请求调用在 Provider 注册的具体服务，并将结果写入 ChannelHandlerContext 返回
ProviderRegister: 使用 ConcurrentHashMap 保存在 Provider 注册的服务和其对象实例

ProviderRegister，主要提供两个方法，addService 和 getService，其作用是在 Provider 注册一个服务和从 HashMap 获取某个服务。

public class ProviderRegister {

/**

服务名称和其对象

private static final Map<String, Object> SERVICE_MAP = new ConcurrentHashMap<>();

/**

添加 RPC Provider 端的服务

public void addService(T service, Class clazz) {

// getCanonicalName() 是获取所传类从java语言规范定义的格式输出

String serviceName = clazz.getCanonicalName();

log.info(“添加服务，名称是 {}”, serviceName);

if (!SERVICE_MAP.containsKey(serviceName)) {

// 将服务名和服务对应的对象添加到 SERVICE_MAP

SERVICE_MAP.put(serviceName, service);

}

/**

获取 RPC Provider 端的服务

public Object getService(String serviceName) {

Object service = SERVICE_MAP.get(serviceName);

if (service == null) {

log.debug(“没有找到该 PRC 服务”);

return null;

}

log.info(“找到服务 {}”, serviceName);

return service;

}

ProviderHandler，根据 RpcProtocol 请求数据中的接口名称，获取其对应的服务，并将结果写入返回。

@Slf4j

public class ProviderHandler extends ChannelInboundHandlerAdapter {

private final ProviderRegister register = new ProviderRegister();

@Override

public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {

Object result; RpcProtocol rpcProtocol = (RpcProtocol) msg; try {

// 从 provider 查找有没有这个服务

Object service = register.getService(rpcProtocol.getInterfaceName());

// 从 service 根据方法名称和传入参数类型获取具体的方法

Method method = service.getClass().getMethod(rpcProtocol.getMethodName(),

rpcProtocol.getParamTypes());

// 执行这个方法

result = method.invoke(service, rpcProtocol.getParamValues());

// 将结果返回

ctx.writeAndFlush(result);

log.info(“服务名称：{}，调用的方法是 {}”, rpcProtocol.getInterfaceName(), rpcProtocol.getMethodName());

} catch (NoSuchMethodException | IllegalArgumentException |

InvocationTargetException | IllegalAccessException e) {

log.error(“服务未找到或者服务发生错误”);

} finally {

ctx.flush();

ctx.close();

}

@Override

public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {

cause.printStackTrace();

ctx.close();

}

Provider，服务端的入口类，启动 Netty 服务，其跟前面两个 demo 中几乎没有区别，主要是更换了编码和解码器，注册 ProviderHandler 处理具体的事件。

@Slf4j

public class Provider {

private final int port;

private final String host;

private final ProviderRegister register = new ProviderRegister();

public Provider(String host, int port) {

this.port = port;

this.host = host;

} /**

启动 Netty 服务，跟前面的 demo 差不多，不同点在于编码器和解码器

public void start() {

log.info(“开始启动 RPC 服务，地址是 {} 端口号是：{}”, host, port);

EventLoopGroup bossGroup = new NioEventLoopGroup();

EventLoopGroup workerGroup = new NioEventLoopGroup();

try {

ServerBootstrap bootstrap = new ServerBootstrap();

bootstrap.group(bossGroup, workerGroup) .channel(NioServerSocketChannel.class)

// 连接的超时时间，超过这个时间还是建立不上的话则代表连接失败

.option(ChannelOption.CONNECT_TIMEOUT_MILLIS, 5000)

// TCP默认开启了 Nagle 算法，该算法的作用是尽可能的发送大数据快，减少网络传输。

// TCP_NODELAY 参数的作用就是控制是否启用 Nagle 算法。

.childOption(ChannelOption.TCP_NODELAY, true)

// 是否开启 TCP 底层心跳机制

.childOption(ChannelOption.SO_KEEPALIVE, true)

//表示系统用于临时存放已完成三次握手的请求的队列的最大长度,如果连接建立频繁，服务器处理创建新连接较慢，

// 可以适当调大这个参数

.option(ChannelOption.SO_BACKLOG, 128)

// Channel 通道的绑定 ChannelPipeline

.childHandler(new ChannelInitializer() {

@Override

protected void initChannel(SocketChannel ch) throws Exception {

ChannelPipeline pipeline = ch.pipeline(); // 使用 JDK 自带的序列化机制 TODO 使用 protobuf 或者 kryo 进行序列化

// 对象的解码器

pipeline.addLast(new ObjectDecoder(Integer.MAX_VALUE,

ClassResolvers.weakCachingConcurrentResolver(this.getClass().getClassLoader())));

// 对象的编码器

pipeline.addLast(new ObjectEncoder());

pipeline.addLast(new ProviderHandler());

}

});

// 使用 bind 监听 host 和 port

ChannelFuture future = bootstrap.bind(host, port).sync();

future.channel().closeFuture().sync();

} catch (InterruptedException e) {

log.error(“开启服务错误:”, e);

} finally {

log.info(“关闭 bossGroup 和 workerGroup”);

bossGroup.shutdownGracefully();

workerGroup.shutdownGracefully();

}

/**

PRC Provider 服务端注册服务
@param service 服务
@param clazz 服务接口定义的类
@param 服务具体的实现类

public void addService(T service, Class clazz) {

}

3、Consumer 客户端

==================================================================================

Consumer：连接 Provider 服务端，发送请求
ConsumerHandler ：将服务端返回的数据进行处理
ConsumerProxy ：使用 InvocationHandler 处理动态代理对象的方法调用

// 将返回的结果提取出来即可

public class ConsumerHandler extends ChannelInboundHandlerAdapter { private Object result; public Object getResult() { return result;

} @Override public void channelRead(ChannelHandlerContext ctx, Object msg) { result = msg; }}

Consumer ,使用 Netty 和 Provider 端通信，基本使用方法和 Netty 客户端方法一样，

public class Consumer {

private final int port;

private final String host;

private final RpcProtocol protocol;

public Consumer(String host, int port, RpcProtocol protocol) {

this.port = port;

this.host = host;

this.protocol = protocol;

} public Object start() throws InterruptedException {

// TODO Netty 连接复用，将这些业务抽取出来

EventLoopGroup group = new NioEventLoopGroup();

ConsumerHandler consumerHandler = new ConsumerHandler();

try {

Bootstrap bootstrap = new Bootstrap();

bootstrap.group(group)

.channel(NioSocketChannel.class)

// 连接的超时时间，超过这个时间还是建立不上的话则代表连接失败

.option(ChannelOption.CONNECT_TIMEOUT_MILLIS, 5000)

// 是否开启 TCP 底层心跳机制

.option(ChannelOption.SO_KEEPALIVE, true)

// TCP默认开启了 Nagle 算法，该算法的作用是尽可能的发送大数据快，减少网络传输。TCP_NODELAY 参数的作用就是控制是否启用 Nagle 算法。

.option(ChannelOption.TCP_NODELAY, true)

// Channel 通道的绑定 ChannelPipeline

.handler(new ChannelInitializer() {

@Override

protected void initChannel(SocketChannel ch) {

ChannelPipeline pipeline = ch.pipeline();

// 对象参数类型解码器

pipeline.addLast(new ObjectDecoder(Integer.MAX_VALUE,

ClassResolvers.cacheDisabled(this.getClass().getClassLoader())));

// 对象参数类型编码器

pipeline.addLast(new ObjectEncoder());

pipeline.addLast(consumerHandler);

}

});

// 链接到服务端和使用 ChannelFuture 接收返回的数据

ChannelFuture future = bootstrap.connect(host, port).sync();

笔者福利

以下是小编自己针对马上即将到来的金九银十准备的一套“面试宝典”，不管是技术还是HR的问题都有针对性的回答。

有了这个，面试踩雷？不存在的！

回馈粉丝，诚意满满！！！

pipeline.addLast(consumerHandler);

}

});

// 链接到服务端和使用 ChannelFuture 接收返回的数据

ChannelFuture future = bootstrap.connect(host, port).sync();

笔者福利

以下是小编自己针对马上即将到来的金九银十准备的一套“面试宝典”，不管是技术还是HR的问题都有针对性的回答。

有了这个，面试踩雷？不存在的！

回馈粉丝，诚意满满！！！

[外链图片转存中…(img-c3ol4X1o-1714430252845)]
[外链图片转存中…(img-M7ffBjtk-1714430252845)]
[外链图片转存中…(img-QDEX22YW-1714430252845)]
[外链图片转存中…(img-f0O5dCJE-1714430252846)]

本文已被CODING开源项目：【一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码】收录

2401_84103936

关注

9
点赞
踩
25

收藏

觉得还不错? 一键收藏
0
评论
Netty 笔记——手把手教你如何写一个 RPC 程序

//***/Provider 服务端主要由三个类构成Provider: 服务端的入口类，启动 Netty 服务ProviderHandler: 根据请求调用在 Provider 注册的具体服务，并将结果写入 ChannelHandlerContext 返回ProviderRegister: 使用 ConcurrentHashMap 保存在 Provider 注册的服务和其对象实例。
复制链接

扫一扫