Go语言中并发运行多个HTTP服务器与路由管理

// 错误示例：导致路由冲突
http.Handle("/", myMux) // 注册到DefaultServeMux
// ...
http.Handle("/", gorillaMux) // 再次注册到DefaultServeMux，引发panic

package main

import (
    "fmt"
    "log"
    "net/http"
    "time"
)

// heartBeatHandler 用于第一个服务器
func heartBeatHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "心跳检测：服务器1正常运行于 %s\n", time.Now().Format(time.RFC3339))
}

// rpcStatusHandler 用于第一个服务器的RPC状态
func rpcStatusHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "RPC服务状态：OK\n")
}

// indexHandler 用于第二个服务器的根路径
func indexHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "欢迎访问服务器2：这是一个常规Web服务\n")
}

// bookHandler 用于第二个服务器的/book路径
func bookHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "书籍信息页面\n")
}

func main() {
    // --- 服务器1：作为RPC/内部通信服务器 ---
    // 创建一个独立的多路复用器用于服务器1
    rpcMux := http.NewServeMux()
    rpcMux.HandleFunc("/heartbeat", heartBeatHandler)
    rpcMux.HandleFunc("/rpc/status", rpcStatusHandler)

    // 配置服务器1
    rpcServer := &http.Server{
        Addr:    ":3400",          // 监听端口3400
        Handler: rpcMux,           // 将自定义的多路复用器赋值给Handler字段
        // ReadTimeout:  5 * time.Second,
        // WriteTimeout: 10 * time.Second,
        // IdleTimeout:  120 * time.Second,
    }

    // 在一个goroutine中启动服务器1，使其非阻塞
    go func() {
        log.Printf("RPC服务器在端口 %s 启动...", rpcServer.Addr)
        if err := rpcServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
            log.Fatalf("RPC服务器启动失败: %v", err)
        }
        log.Printf("RPC服务器已关闭。")
    }()

    // --- 服务器2：作为常规Web服务器 ---
    // 创建一个独立的多路复用器用于服务器2
    webMux := http.NewServeMux()
    webMux.HandleFunc("/", indexHandler)
    webMux.HandleFunc("/book", bookHandler)
    webMux.HandleFunc("/api/data", func(w http.ResponseWriter, r *http.Request) {
        fmt.Fprintf(w, "这是来自服务器2的API数据\n")
    })

    // 配置服务器2
    webServer := &http.Server{
        Addr:    ":1234",          // 监听端口1234
        Handler: webMux,           // 将自定义的多路复用器赋值给Handler字段
        // ReadTimeout:  5 * time.Second,
        // WriteTimeout: 10 * time.Second,
        // IdleTimeout:  120 * time.Second,
    }

    // 启动服务器2（通常是主服务，阻塞主goroutine）
    log.Printf("Web服务器在端口 %s 启动...", webServer.Addr)
    if err := webServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
        log.Fatalf("Web服务器启动失败: %v", err)
    }
    log.Printf("Web服务器已关闭。")

    // 在实际应用中，您可能需要在此处添加优雅关闭逻辑，例如监听操作系统信号。
}

go get github.com/gorilla/mux

package main

import (
    "fmt"
    "log"
    "net/http"
    "time"

    "github.com/gorilla/mux" // 导入 gorilla/mux
)

// heartBeatHandler 用于第一个服务器
func heartBeatHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "心跳检测：RPC服务器正常运行于 %s\n", time.Now().Format(time.RFC3339))
}

// indexHandler 用于第二个服务器的根路径
func indexHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "欢迎访问Web服务器：这是一个常规Web服务\n")
}

// bookHandler 用于第二个服务器的/book路径
func bookHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "书籍信息页面\n")
}

// productHandler 用于演示gorilla/mux的路径参数
func productHandler(w http.ResponseWriter, r *http.Request) {
    vars := mux.Vars(r)
    fmt.Fprintf(w, "您正在查看产品ID：%s\n", vars["id"])
}

func main() {
    // --- 服务器1：RPC服务器（使用标准库ServeMux） ---
    rpcMux := http.NewServeMux()
    rpcMux.HandleFunc("/heartbeat", heartBeatHandler)
    rpcMux.HandleFunc("/rpc/status", func(w http.ResponseWriter, r *http.Request) {
        fmt.Fprintf(w, "RPC服务状态：OK at %s\n", time.Now().Format(time.RFC3339))
    })

    rpcServer := &http.Server{
        Addr:    ":3400",
        Handler: rpcMux,
    }

    go func() {
        log.Printf("RPC服务器在端口 %s 启动...", rpcServer.Addr)
        if err := rpcServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
            log.Fatalf("RPC服务器启动失败: %v", err)
        }
        log.Printf("RPC服务器已关闭。")
    }()

    // --- 服务器2：常规Web服务器（使用gorilla/mux） ---
    webRouter := mux.NewRouter() // 创建gorilla/mux路由器
    webRouter.HandleFunc("/", indexHandler).Methods("GET")
    webRouter.HandleFunc("/book", bookHandler).Methods("GET", "POST")
    webRouter.HandleFunc("/products/{id}", productHandler).Methods("GET") // 演示路径参数

    webServer := &http.Server{
        Addr:    ":1234",
        Handler: webRouter, // 将gorilla/mux路由器赋值给Handler字段
    }

    log.Printf("Web服务器在端口 %s 启动...", webServer.Addr)
    if err := webServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
        log.Fatalf("Web服务器启动失败: %v", err)
    }
    log.Printf("Web服务器已关闭。")
}