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.mzph.cn/news/879718.shtml

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

相关文章

解决mybatis plus 中 FastjsonTypeHandler无法正确反序列化List类型的问题

由于是根据自动映射类型,我们设置的字段类型是List 也就是反序列化的时候也只是用 FastjsonTypeHandler中的 Override protected Object parse(String json) { return JSON.parseObject(json, type); } 反序列化方法,这是type为List 反序列后我们并没…

C++11: 声明和定义

声明与定义是C/C中两个核心的概念,也是C/C区别于其他语言独有的特性。它们对程序的编译和链接过程起着至关重要的作用。 一、C标准的描述 声明(Declaration):声明告诉编译器某个实体(如变量、函数、类等)…

用ASR PRO离线语音芯片和月饼盒做一个会跑会跳会说话的机器狗

中秋节刚过,大家月饼盒应该还有,不要扔,可以做点小玩意。 机器狗的创意来自B站石桥北的视频,他使用了一块ESP32芯片和打印件加四个舵机实现,应该说是比较复杂的,需要有3D打印机打印外壳,还得会…

Linux标准IO-系统调用详解

1.1 系统调用 系统调用(system call)其实是 Linux 内核提供给应用层的应用编程接口(API),是 Linux 应用层进入内核的入口。不止 Linux 系统,所有的操作系统都会向应用层提供系统调用,应用程序通…

【Harmony】轮播图特效,持续更新中。。。。

效果预览 swiper官网例子 Swiper 高度可变化 两边等长露出,跟随手指滑动 Swiper 指示器导航点位于 Swiper 下方 一、官网 例子 参考代码: // xxx.ets class MyDataSource implements IDataSource {private list: number[] []constructor(list: nu…

软考高级:嵌入式系统调度算法 AI 解读

嵌入式系统中的调度算法用于管理任务的执行顺序,确保系统资源能够有效分配。以下是几种常见的调度算法的通俗讲解。 生活化例子 想象你是一位超市收银员,有很多顾客排队,每位顾客都可以看作一个任务,收银台就是你的处理器。你需…

PostgreSQL技术内幕10:PostgreSQL事务原理解析-日志模块介绍

文章目录 0.简介1.PG日志介绍2.事务日志介绍3.WAL分析3.1 WAL概述3.2 WAL设计考虑3.2.1 存储格式3.2.2 实现方式3.2.3 数据完整性校验3.3 check ponit 4.事务提交日志(CLOG)4.1 clog存储使用介绍4.2 slru缓冲池并发控制 0.简介 本文将延续上一篇文章内容…

vue常用业务场景

vue3: 使用本地存储 import { useStorage } from /hooks/web/useStorage const { getStorage, setStorage, removeStorage } useStorage()setStorage("data", data); // 可为对象使用pinia (1)state: ({isDisableSwitch: false}) (2) getters: getIs…

【无标题】Java_Se 数据变量与运算符

标识符、变量、常量、数据类型、运算符、基本数据类型的类型转换等。这些是编程中的“砖块”,是编程的基础。要想开始正式编程,还需要再学“控制语句”,控制语句就像“水泥”,可以把“砖块”粘到一起,最终形成“一座大…

华为OD机试 - 二维伞的雨滴效应(Python/JS/C/C++ 2024 E卷 200分)

华为OD机试 2024E卷题库疯狂收录中,刷题点这里 专栏导读 本专栏收录于《华为OD机试真题(Python/JS/C/C)》。 刷的越多,抽中的概率越大,私信哪吒,备注华为OD,加入华为OD刷题交流群,…

多模态AI技术详解:跨越数据边界的智能未来

1. 引言 在当今快速发展的人工智能(AI)领域,多模态AI技术正逐渐成为研究热点。它通过结合不同类型的数据(如文本、图像、音频等),为机器提供更全面的理解能力。本文将深入探讨多模态AI的基本概念、应用场景…

ClickHouse-Kafka Engine 正确的使用方式

Kafka 是大数据领域非常流行的一款分布式消息中间件,是实时计算中必不可少的一环,同时一款 OLAP 系统能否对接 Kafka 也算是考量是否具备流批一体的衡量指标之一。ClickHouse 的 Kafka 表引擎能够直接与 Kafka 系统对接,进而订阅 Kafka 中的 …

PHP 数组排序类型介绍

在PHP中,数组排序是一项常见且重要的操作,它允许开发者根据一定的规则对数组中的元素进行排序。PHP提供了多种数组排序函数,以适应不同的排序需求。这些函数包括基本的升序和降序排序,以及基于特定键值、自定义排序逻辑等的复杂排…

镀金引线---

一、沉金和镀金 沉金和镀金都是常见的PCB金手指处理方式,它们各有优劣势,选择哪种方式取决于具体的应用需求和预算。 沉金(ENIG)是一种常用的金手指处理方式,它通过在金手指表面沉积一层金层来提高接触性能和耐腐蚀性…

【C++】模拟实现vector

在上篇中我们已经了解过的vector各种接口的功能使用,接下来我们就试着模拟实现一下吧! 注意:我们在此实现的和C标准库中实现的有所不同,其目的主要是帮助大家大概理解底层原理。 我们模拟vector容器的大致框架是: t…

2024年【四川省安全员B证】新版试题及四川省安全员B证考试试卷

题库来源:安全生产模拟考试一点通公众号小程序 四川省安全员B证新版试题参考答案及四川省安全员B证考试试题解析是安全生产模拟考试一点通题库老师及四川省安全员B证操作证已考过的学员汇总,相对有效帮助四川省安全员B证考试试卷学员顺利通过考试。 1、…

什么是注解?

1.是什么 注解(Annotation)是Java语言中一种元数据(metadata)的形式,它用于提供关于代码的额外信息,这些信息不会影响代码的编译或执行,但可以被工具用来进行编译时的检查或者在运行时进行处理。…

【webpack4系列】webpack基础用法(二)

文章目录 entryoutputloaderpluginmode前端构建基础配置关联HTML插件html-webpack-plugin构建 CSS 解析 ES6和React JSX解析 ES6解析 React JSX 解析CSS、Less和Sass解析CSS解析Less解析sass 解析图片和字体资源解析:解析图片资源解析:解析字体资源解析&…

Flutter 优化技巧分享

Flutter 作为一个跨平台框架,凭借其高效的开发体验和丰富的生态,受到越来越多开发者的青睐。然而,在开发过程中,随着项目的复杂度增加,性能问题和代码优化需求也逐渐显现。为了确保应用在各种场景下保持流畅的用户体验…

JS - 获取剪切板内容 Clipboard API

目录 1,需求最终效果 2,实现示例 3,注意点1,只支持安全上下文环境2,只能读取当前页面的剪切板3,权限获取问题4,获取内容的 MIME_TYPE 问题1,文本内容2,图片内容 5&#x…