从零手写实现 nginx-35-proxy_pass netty 如何实现？

农民 · 2024-7-27 23:18:37

前言

各人好，我是老马。很高兴遇到你。
我们为 java 开发者实现了 java 版本的 nginx

https://github.com/houbb/nginx4j

假如你想知道 servlet 如那里理的，可以参考我的另一个项目：

手写从零实现简易版 tomcat minicat

根据原始的 request 请求，构建新的请求对象 forwardedRequest
根据指定的路由计谋，获取一个目标服务器。
根据目标服务器的 host+port，用 netty 直接模拟 http 客户端，直接访问长途服务端，然后把长途的响应写回到当前的客户端 resp

实现代码

核心实现如下：

/**
* netty 实现反向代理
*
* @since 0.27.0
* @author 老马啸西风
*/
public class NginxRequestDispatchProxyPass extends AbstractNginxRequestDispatch {
private static final Log logger = LogFactory.getLog(NginxRequestDispatchProxyPass.class);
@Override
public void doDispatch(NginxRequestDispatchContext context) {
// 原始的请求
final FullHttpRequest request = context.getRequest();
final ChannelHandlerContext ctx = context.getCtx();
// 创建一个新的 FullHttpRequest 转发到目标服务器
FullHttpRequest forwardedRequest = new DefaultFullHttpRequest(
request.protocolVersion(), request.method(), request.uri(), request.content().retainedDuplicate());
forwardedRequest.headers().set(request.headers());
final NginxLoadBalanceConfig nginxLoadBalanceConfig = context.getBalanceConfig();
// 创建一个新的 Bootstrap 进行 HTTP 请求
Bootstrap b = new Bootstrap();
b.group(ctx.channel().eventLoop())
.channel(NioSocketChannel.class)
.handler(new ChannelInitializer<Channel>() {
@Override
protected void initChannel(Channel ch) throws Exception {
//...
ch.pipeline().addLast(new SimpleChannelInboundHandler<FullHttpResponse>() {
@Override
protected void channelRead0(ChannelHandlerContext clientCtx, FullHttpResponse response) throws Exception {
// 将目标服务器的响应写回到客户端
FullHttpResponse clientResponse = new DefaultFullHttpResponse(
response.protocolVersion(), response.status(), response.content().retainedDuplicate());
clientResponse.headers().set(response.headers());
ctx.writeAndFlush(clientResponse).addListener(ChannelFutureListener.CLOSE);
clientCtx.close();
}
@Override
public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
logger.error("exceptionCaught meet ex", cause);
ctx.close();
}
});
}
});
// 连接到目标服务器并发送请求
final IServer server = getActualServer(nginxLoadBalanceConfig);
b.connect(server.host(), server.port()).addListener((ChannelFutureListener) future -> {
if (future.isSuccess()) {
future.channel().writeAndFlush(forwardedRequest);
} else {
ctx.close();
}
});
}
}

复制代码

负载平衡

负载平衡计谋，可以看我以前的文章：

从零手写实现 nginx-32-load balance 负载平衡算法 java 实现

小结

到这里开始，我们基本实现了反向代理。
当然，其中尚有许多细节必要处置惩罚。

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。

		自动登录	找回密码
密码			立即注册

从零手写实现 nginx-35-proxy_pass netty 如何实现？

0 个回复

快速回复

楼主热帖

标签云