Nginx反向代理下Go应用重定向路径错误解决方案

location /bar/ {
  proxy_pass http://localhost:8088/;
  proxy_redirect off;
  proxy_set_header  Host               $host;
  proxy_set_header  X-Real-IP          $remote_addr;
  proxy_set_header  X-Forwarded-For    $proxy_add_x_forwarded_for;
  proxy_set_header  X-Forwarded-Proto  https;
}

package main

import (
    "fmt"
    "net/http"
)

func root(w http.ResponseWriter, r *http.Request) {
    fmt.Fprint(w, "You reached root")
}

func foo(w http.ResponseWriter, r *http.Request) {
    http.Redirect(w, r, "/", http.StatusFound) // 问题所在：硬编码重定向到根路径
}

func main() {
    http.HandleFunc("/", root)
    http.HandleFunc("/foo", foo)
    http.ListenAndServe("localhost:8088", nil)
}

package main

import (
    "fmt"
    "net/http"
    "os"
    "strings"
)

// 定义一个全局变量或通过结构体传递，表示应用的外部基础路径
var appBasePath string

func init() {
    // 示例：从环境变量获取基础路径，如果没有则默认为空
    // 实际项目中可以从配置文件或命令行参数获取
    appBasePath = os.Getenv("APP_BASE_PATH")
    if appBasePath == "" {
        appBasePath = "/" // 默认为根路径
    }
    // 确保基础路径以斜杠结尾，并以斜杠开头（如果不是根路径）
    if !strings.HasPrefix(appBasePath, "/") {
        appBasePath = "/" + appBasePath
    }
    if !strings.HasSuffix(appBasePath, "/") {
        appBasePath = appBasePath + "/"
    }
    fmt.Printf("Application Base Path: %s\n", appBasePath)
}

// ... (其他函数)

// redirectWithBasePath 根据应用的基础路径进行重定向
func redirectWithBasePath(w http.ResponseWriter, r *http.Request, path string, code int) {
    // 如果目标路径是相对路径且以"/"开头，则拼接基础路径
    if strings.HasPrefix(path, "/") && !strings.HasPrefix(path, "//") { // 避免处理绝对URL或协议相对URL
        // 移除基础路径末尾的斜杠，避免双斜杠
        targetPath := strings.TrimSuffix(appBasePath, "/") + path
        http.Redirect(w, r, targetPath, code)
        return
    }
    // 对于非"/"开头的路径（如相对路径或完整URL），直接重定向
    http.Redirect(w, r, path, code)
}

// 修改 foo 处理器以使用自定义重定向函数
func foo(w http.ResponseWriter, r *http.Request) {
    // 现在重定向到应用的“内部根路径”时，会被自动转换为外部的 /bar/
    redirectWithBasePath(w, r, "/", http.StatusFound)
}

func root(w http.ResponseWriter, r *http.Request) {
    fmt.Fprint(w, "You reached root")
}

func main() {
    http.HandleFunc("/", root)
    http.HandleFunc("/foo", foo)
    http.ListenAndServe("localhost:8088", nil)
}

export APP_BASE_PATH="/bar"
go run your_app.go