基于Go语言的微服务请求转发与代理工具

package main

import (
    "fmt"
    "io/ioutil"
    "log"
    "net/http"
    "net/url"
    "strings"
)

func main() {
    // 加载配置文件
    config := loadConfig("config.ini")

    // 监听指定端口
    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        // 获取请求路径
        path := r.URL.Path

        // 查找对应的微服务地址
        serviceURL, ok := config[path]
        if !ok {
            http.Error(w, "Invalid request path", http.StatusBadRequest)
            return
        }

        // 构建目标URL
        targetURL := fmt.Sprintf("%s%s", serviceURL, r.URL.RawQuery)

        // 发起请求
        resp, err := http.Get(targetURL)
        if err != nil {
            log.Printf("Failed to proxy request: %v", err)
            http.Error(w, "Failed to proxy request", http.StatusInternalServerError)
            return
        }

        // 将微服务的响应返回给客户端
        defer resp.Body.Close()
        body, err := ioutil.ReadAll(resp.Body)
        if err != nil {
            log.Printf("Failed to read response body: %v", err)
            http.Error(w, "Failed to read response body", http.StatusInternalServerError)
            return
        }
        w.Write(body)
    })

    log.Println("Proxy server started on port 8080")
    log.Fatal(http.ListenAndServe(":8080", nil))
}

func loadConfig(filename string) map[string]string {
    // 解析配置文件
    // 例如：
    // /user -> http://user-service:8000
    // /order -> http://order-service:8000
    config := make(map[string]string)
    config["/user"] = "http://user-service:8000"
    config["/order"] = "http://order-service:8000"
    return config
}