golang http请求转发-Golang-PHP中文网

golang http请求转发

王林

发布： 2023-05-13 11:43:36

原创

563人浏览过

在前后端分离的开发模式中，经常需要在前端发送 http 请求，并在后端进行响应，然而由于某些原因，有时候又需要前端直接与外部系统进行通信，因此就需要在后端进行 http 请求转发。本文将介绍如何使用 golang 实现 http 请求转发。

首先，我们需要安装 Golang 环境，并安装好相应的依赖包。接下来，我们就可以开始编写代码了。

package main

import (
    "log"
    "net/http"
)

func main() {
    http.HandleFunc("/", handleRequestAndRedirect)
    log.Fatal(http.ListenAndServe(":8080", nil))
}

func handleRequestAndRedirect(w http.ResponseWriter, r *http.Request) {
    url := "http://example.com" + r.URL.Path
    log.Printf("Redirecting to %s", url)

    req, err := http.NewRequest(r.Method, url, r.Body)
    if err != nil {
        http.Error(w, err.Error(), http.StatusInternalServerError)
        return
    }

    // 设置请求头
    for name, values := range r.Header {
        for _, value := range values {
            req.Header.Add(name, value)
        }
    }

    client := http.Client{}
    resp, err := client.Do(req)
    if err != nil {
        http.Error(w, err.Error(), http.StatusBadGateway)
        return
    }
    defer resp.Body.Close()

    // 将响应转发到客户端
    for name, values := range resp.Header {
        for _, value := range values {
            w.Header().Add(name, value)
        }
    }
    w.WriteHeader(resp.StatusCode)
    if _, err := io.Copy(w, resp.Body); err != nil {
        log.Printf("Error copying to client: %v", err)
    }
}

登录后复制

以上代码实现了一个 HTTP 请求转发的基础功能。代码中的 handleRequestAndRedirect 函数接收客户端发来的请求，并根据请求路径去请求指定的外部系统。代码通过读取 HTTP 请求的 header，在请求头中添加相关信息，并使用 Go 的标准库中的 http 包发送 HTTP 请求。随后，接收到外部系统的响应，并将其转发到客户端。值得注意的是，本代码还处理了错误情况，并将错误信息返回给客户端。

如果我们需要实现更多的功能，比如：请求重试、请求日志记录等，我们可以对以上代码进行适当地修改。在请求重试方面，我们可以添加请求失败后，对请求重新发送的逻辑；在请求日志记录方面，则可以添加记录请求日志的代码。