Go语言中并发运行多个Web服务器的实践指南

package main

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

    "github.com/gorilla/mux" // 引入gorilla/mux，如果不需要可以只用http.NewServeMux()
)

// heartBeatHandler 处理器用于第一个服务器
func heartBeatHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Heartbeat from regular server at %s\n", time.Now().Format(time.RFC3339))
    log.Printf("Regular server received heartbeat request from %s", r.RemoteAddr)
}

// indexHandler 处理器用于第二个服务器的根路径
func indexHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Welcome to the RPC server!\n")
    log.Printf("RPC server received index request from %s", r.RemoteAddr)
}

// bookHandler 处理器用于第二个服务器的/book路径
func bookHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "This is the book endpoint on the RPC server.\n")
    log.Printf("RPC server received book request from %s", r.RemoteAddr)
}

func main() {
    // --- 第一个服务器：常规Web服务器 (使用http.NewServeMux) ---
    // 1. 创建一个独立的ServeMux实例
    regularMux := http.NewServeMux()
    // 2. 在这个独立的ServeMux上注册处理器
    regularMux.HandleFunc("/heartbeat", heartBeatHandler)
    regularMux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        fmt.Fprintf(w, "Default handler for regular server.\n")
    })

    // 3. 创建http.Server实例，并将独立的ServeMux赋值给Handler字段
    regularServer := &http.Server{
        Addr:    ":3400", // 监听端口3400
        Handler: regularMux,
    }

    // 4. 在一个goroutine中启动第一个服务器，使其并发运行
    go func() {
        log.Printf("Starting regular web server on :3400")
        if err := regularServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
            log.Fatalf("Regular server failed to start: %v", err)
        }
    }()

    // --- 第二个服务器：RPC服务器 (使用gorilla/mux) ---
    // 1. 创建一个独立的gorilla/mux路由器实例
    rpcRouter := mux.NewRouter()
    // 2. 在这个独立的路由器上注册处理器
    rpcRouter.HandleFunc("/", indexHandler)
    rpcRouter.HandleFunc("/book", bookHandler)

    // 3. 创建http.Server实例，并将独立的gorilla/mux路由器赋值给Handler字段
    rpcServer := &http.Server{
        Addr:    ":1234", // 监听端口1234
        Handler: rpcRouter,
    }

    // 4. 启动第二个服务器。通常主goroutine会阻塞在这里，
    //    直到服务器关闭或发生致命错误。
    log.Printf("Starting RPC server on :1234")
    if err := rpcServer.ListenAndServe(); err != nil && err != http.ErrServerClosed {
        log.Fatalf("RPC server failed to start: %v", err)
    }

    // 如果两个服务器都放在goroutine中，需要某种机制保持主goroutine存活，
    // 例如使用select{}阻塞，或者等待信号。
    // select {}
}