Golang使用ReverseProxy实现反向代理

目录

1.源码结构体 

2.官方单机示例

3.使用示例

4.简单的http服务(用于测试)


1.源码结构体 

type ReverseProxy struct {// Rewrite 必须是一个函数,用于将请求修改为要使用 Transport 发送的新请求。然后,其响应将原封不动地复制回原始客户端。返回后,Rewrite 不得访问提供的 ProxyRequest 或其内容。// 在调用 Rewrite 之前,将从出站请求中删除 Forwarded、X-Forwarded、X-Forwarded-Host 和 X-Forwarded-Proto 标头。另请参阅 ProxyRequest.SetXForwarded 方法。// 在调用 Rewrite 之前,将从出站请求中删除不可解析的查询参数。Rewrite 函数可以将入站 URL 的 RawQuery 复制到出站 URL 以保留原始参数字符串。请注意,如果代理对查询参数的解释与下游服务器的解释不匹配,这可能会导致安全问题。// 最多可以设置 Rewrite 或 Director 中的一个。Rewrite func(*ProxyRequest)// Director 是一种将请求修改为要使用 Transport 发送的新请求的功能。然后,其响应将原封不动地复制回原始客户端。Director 在返回后不得访问提供的请求。//默认情况下,X-Forwarded-For 标头设置为客户端 IP 地址的值。如果 X-Forwarded-For 标头已存在,则客户端 IP 将附加到现有值。作为特殊情况,如果标头存在于 Request.Header 映射中,但具有 nil 值(例如由 Director func 设置时),则不会修改 X-Forwarded-For 标头。// 为防止 IP 欺骗,请务必删除来自客户端或不受信任的代理的任何预先存在的 X-Forwarded-For 标头。// 在 Director 返回后,将从请求中删除逐跳标头,这可以删除 Director 添加的标头。请改用 Rewrite 函数来确保保留对请求的修改。// 如果在 Director 返回后设置 Request.Form,则会从出站请求中删除不可解析的查询参数。// 最多可以设置 Rewrite 或 Director 中的一个。Director func(*http.Request)// 用于执行代理请求的传输。如果为 nil,则使用为 http.DefaultTransportTransport http.RoundTripper// FlushInterval 指定在复制响应正文时要刷新到客户端的刷新间隔。如果为零,则不执行定期刷新。负值表示在每次写入 Client 端后立即刷新。当 ReverseProxy 将响应识别为流式响应或其 ContentLength 为 -1 时,将忽略 FlushInterval;对于此类响应,写入会立即刷新到客户端。FlushInterval time.Duration// ErrorLog 为尝试代理请求时发生的错误指定可选记录器。如果为 nil,则通过 log 包的标准 logger 完成日志记录。ErrorLog *log.Logger// BufferPool 可以选择指定一个缓冲池,以获取 io 使用的字节切片。CopyBuffer 在复制 HTTP 响应正文时。   BufferPool BufferPool// ModifyResponse 是一个可选函数,用于修改来自后端的 Response。如果后端返回带有任何 HTTP 状态代码的响应,则调用它。如果无法访问后端,则调用可选的 ErrorHandler,而不调用 ModifyResponse。如果 ModifyResponse 返回错误,则调用 ErrorHandler 及其错误值。如果 ErrorHandler 为 nil,则使用其默认实现。ModifyResponse func(*http.Response) error// ErrorHandler 是一个可选函数,用于处理到达后端的错误或来自 ModifyResponse 的错误。如果为 nil,则默认记录提供的错误并返回 502 Status Bad Gateway 响应。    ErrorHandler func(http.ResponseWriter, *http.Request, error)
}

2.官方单机示例

NewSingleHostReverseProxy是官方给的示例,代理单机服务,如果想实现负载均衡,需自己实现。

源码:

3.使用示例

package mainimport ("log""net""net/http""net/http/httputil""net/url""os""sync""time"
)// 从Go 1.18版本开始,httputil.ReverseProxy 的 BufferPool 字段期望的是一个实现了 BufferPool 接口的对象,该接口要求有一个返回 []byte 的 Get 方法和一个接受 []byte 的 Put 方法。
// 定义一个实现了 BufferPool 接口的结构体
// 定义一个实现了 BufferPool 接口的结构体
type byteBufferPool struct {sync.Pool
}// 实现 BufferPool 接口的 Get 方法
func (b *byteBufferPool) Get() []byte {v := b.Pool.Get()if v == nil {return make([]byte, 1024*1024) // 分配 1MB 的缓冲区}return v.([]byte)
}// 实现 BufferPool 接口的 Put 方法
func (b *byteBufferPool) Put(bts []byte) {b.Pool.Put(bts)
}// 定义一个实现了 ErrorHandler 接口的函数
func customErrorHandler(w http.ResponseWriter, r *http.Request, err error) {// 记录错误log.Printf("Error occurred: %v", err)// 返回自定义的状态码和错误消息w.WriteHeader(http.StatusInternalServerError)w.Write([]byte("An internal error occurred."))
}func main() {// 目标服务器的 URLtargetURL := "http://localhost:8081"// 解析目标 URLtarget, err := url.Parse(targetURL)if err != nil {log.Fatalf("Failed to parse target URL: %v", err)}// 创建反向代理proxy := httputil.NewSingleHostReverseProxy(target)// 设置 FlushInterval// FlushInterval 是一个 time.Duration 类型的字段,它指定了代理服务器将缓冲的数据写入客户端的间隔时间。这对于实现实时应用(如实时聊天、日志流等)非常重要,因为它可以减少延迟,让客户端更快地接收到数据。proxy.FlushInterval = 100 * time.Millisecond// 设置反向代理的 Transportproxy.Transport = &http.Transport{DialContext: (&net.Dialer{Timeout:   30 * time.Second, // 建立连接的最大等待时间KeepAlive: 30 * time.Second, // 保持连接活跃的时间间隔DualStack: true,             // 尝试同时使用 IPv4 和 IPv6 地址}).DialContext,MaxIdleConns:          100,              // 最大空闲连接数IdleConnTimeout:       90 * time.Second, // 空闲连接超时时间TLSHandshakeTimeout:   10 * time.Second, // TLS握手超时时间ExpectContinueTimeout: 1 * time.Second,  // Expect: 100-continue 超时时间}// 创建自定义的日志器proxy.ErrorLog = log.New(os.Stderr, "ERROR: ", log.LstdFlags)// 设置反向代理的 BufferPool 缓冲池proxy.BufferPool = &byteBufferPool{Pool: sync.Pool{New: func() interface{} {return make([]byte, 1024*1024) // 分配 1MB 的缓冲区},},}// 设置反向代理的 ModifyResponseproxy.ModifyResponse = func(res *http.Response) error {// 修改响应头res.Header.Set("X-Modified-By", "ReverseProxy")// 修改状态码(示例)res.StatusCode = 200// 返回 nil 表示继续处理return nil}// 设置反向代理的 ErrorHandlerproxy.ErrorHandler = customErrorHandler// 创建 HTTP 服务器http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {// 调用反向代理proxy.ServeHTTP(w, r)})// 启动 HTTP 服务器log.Println("Starting server on :8080")log.Fatal(http.ListenAndServe(":8080", nil))
}

4.简单的http服务(用于测试)

package mainimport ("fmt""log""net/http"
)func helloWorldHandler(w http.ResponseWriter, r *http.Request) {fmt.Fprintf(w, "Hello, World!")
}func main() {http.HandleFunc("/", helloWorldHandler)log.Println("Starting server on :8081")log.Fatal(http.ListenAndServe(":8081", nil))
}

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.rhkb.cn/news/422487.html

如若内容造成侵权/违法违规/事实不符,请联系长河编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

微软数据库的SQL注入漏洞解析——Microsoft Access、SQLServer与SQL注入防御

说明:本文仅是用于学习分析自己搭建的SQL漏洞内容和原理,请勿用在非法途径上,违者后果自负,与笔者无关;本文开始前请认真详细学习《‌中华人民共和国网络安全法》‌及其相关法规内容【学法时习之丨网络安全在身边一图了解网络安全法_中央网络安全和信息化委员会办公室】 。…

Numba加速计算:最近邻插值(CPU+ GPU + Z轴切块 + XYZ轴切块 + 多线程)

文章目录 最近邻插值(加速方法)(1)scipy.ndimage.zoom(2)Numba-CPU加速(3)Numba-GPU加速(4)Numba-CPU加速(Z轴切块)(5&…

分类预测|基于贝叶斯优化长短期记忆网络的数据分类预测Matlab程序 多特征输入多类别输出 BO-LSTM 附赠预测新数据

分类预测|基于贝叶斯优化长短期记忆网络的数据分类预测Matlab程序 多特征输入多类别输出 BO-LSTM 附赠预测新数据 文章目录 一、基本原理BO-LSTM分类预测原理和流程总结 二、实验结果三、核心代码四、代码获取五、总结 分类预测|基于贝叶斯优化长短期记忆网络的数据分类预测Mat…

利用zabbix监控ogg进程(Windows平台)

💝💝💝欢迎来到我的博客,很高兴能够在这里和您见面!希望您在这里可以感受到一份轻松愉快的氛围,不仅可以获得有趣的内容和知识,也可以畅所欲言、分享您的想法和见解。 推荐:Linux运维老纪的首页…

QT:音视频播放器

目录 一.播放器设计 二.需要使用的控件 三.选择视频 四.播放视频 五.暂停视频 六.关闭视频 七.播放状态设置 八.切换视频(上一首) 九.切换视频(下一首) 十.设置视频滑块 十一.更新滑块显示 十二.实现效果 十三.代码设计 1.mainwindow.h 2.mainwindow.cpp 一.播放…

Windows上安装RabbitMQ

rabbitmq是干嘛的我就不介绍了,直接开始安装教程。 搭建成功演示图 下载安装包 https://pan.baidu.com/s/1ZlCFxh9Q00ynSU3ZCpTC9Q?pwdry51​pan.baidu.com/s/1ZlCFxh9Q00ynSU3ZCpTC9Q?pwdry51 下载完后有两个包(erlang和rabbitmq) 先安装otp_win64_24.1.7.exe…

wifiip地址可以随便改吗?wifi的ip地址怎么改变

对于普通用户来说,WiFi IP地址的管理和修改往往显得神秘而复杂。本文旨在深入探讨WiFi IP地址是否可以随意更改,以及如何正确地改变WiFi的IP地址。虎观代理小二将详细解释WiFi IP地址的基本概念、作用以及更改时需要注意的事项,帮助用户更好地…

使用ShardingSphere实现MySql的分库分表

目录 一 什么是ShardingSphere分库分表 二 代码实现 1.导入相关依赖 2.配置相关参数 3.创建学生类以及mapper接口 4.实现 StandardShardingAlgorithm接口自定义分片算法 唐洋洋我知道你在看!!!嘿嘿 一 什么是ShardingSphere分库分表 我们平时在设计数据库的时候&#xf…

2-92 基于matlab的KPCA的TE过程的故障监测

基于matlab的KPCA的TE过程的故障监测,利用核主元分析法(KPCA)来进行故障检测的思想,将输入空间中复杂的非线性问题转化为特征空间中的线性问题,计算步骤:(1) 选择监控变量,收集正常工况下的各变量的样本&am…

移动订货小程序哪个好 批发订货系统源码哪个好

订货小程序就是依托微信小程序的订货系统,微信小程序订货系统相较于其他终端的订货方式,能够更快进入商城,对经销商而言更为方便。今天,我们一起盘点三个主流的移动订货小程序,看看哪个移动订货小程序好。 第一、核货宝…

无线麦克风哪款好用,手机领夹麦克风哪个牌子好,麦克风推荐

随着短视频与直播行业的蓬勃发展,无线领夹麦克风市场迎来了前所未有的繁荣。品牌如罗德、大疆、西圣等麦克风品牌凭借卓越的技术实力与品牌影响力占据了市场的主导地位,其中西圣更是凭借其高性价比和用户口碑,稳居行业口碑品牌前列。但在这光…

线性规划及其MATLAB实现

目录 线性规划及其MATLAB实现 引言 线性规划的基本模型 线性规划求解方法 MATLAB中的线性规划求解 MATLAB线性规划应用实例 1. 生产计划问题 模型建立: 2. 运输问题 2. 运输问题 MATLAB实现: 线性规划在实际中的应用 结论 线性规划及其MATLA…

路基边坡自动化监测解决方案

物联网云平台 平台登录--用户登录 输入网址:http://yun.sj2000.org.cn,进入系统登录界面,输入用户名及密码后进入系统平台。 设备详情--设备概览 登录系统平台后,用户可在界面左侧看到系统项目栏和子项目选项,登陆的…

LSS可视化分析

1 完整 2 去掉plt.imshow(img_show) 3 去掉plt.axis(‘off’) 4 去掉plt.annotate(cams_text[img_id].replace(‘_’, ’ ), (0.01, 0.92), xycoords=‘axes fraction’)

多线程与并发区别

在Java中,多线程与并发是两个既相关又有所区别的概念。我们可以这样来理解它们: 多线程(Multi-threading): 多线程是指程序能够同时执行多个线程。每个线程都是一个独立的执行流,它们共享程序的内存空间&a…

【隐私计算】Paillier半同态加密算法

一、何为同态加密(HE)? HE是一种特殊的加密方法,它允许直接对加密数据执行计算,如加法和乘法,而计算过程不会泄露原文的任何信息。计算的结果仍然是加密的,拥有密钥的用户对处理过的密文数据进…

基于 SpringBoot 的车辆充电桩管理系统

专业团队,咨询就送开题报告 摘 要 随着信息化时代的到来,管理系统都趋向于智能化、系统化,车辆充电桩管理系统也不例外,但目前国内仍都使用人工管理,市场规模越来越大,同时信息量也越来越庞大,…

江协科技stm32————11-5 硬件SPI读写W25Q64

一、开启时钟,开启SPI和GPIO的时钟 二、初始化GPIO口,其中SCK和MOSI是由硬件外设控制的输出信号,配置为复用推挽输出 MISO是硬件外设的输入信号,配置为上拉输入,SS是软件控制的输出信号,配置为通用推挽输出…

十,Spring Boot 的内容协商的详细剖析(附+Debug调试说明)

十,Spring Boot 的内容协商的详细剖析(附Debug调试说明) 文章目录 十,Spring Boot 的内容协商的详细剖析(附Debug调试说明)1. 基本介绍2. 准备工作3. 内容协商的本质4. 内容协商:注意事项和使用细节5. 总结:6. 最后: 1…

数据库安全性控制

‍ 在当今信息化时代,数据库安全性 对于保护数据免受非法访问和损害至关重要。无论是个人数据还是企业机密,数据库安全性控制都能有效地防范潜在的威胁。本文将为你深入浅出地介绍数据库安全性控制的关键方法和机制,帮助你轻松掌握这一重要概…