package main

import (
    "fmt"
    "net"
    "os"
    "os/exec"
    "strconv"
    "time"
)

// main 函數(shù)根據(jù)命令行參數(shù)決定運行父進(jìn)程還是子進(jìn)程邏輯
func main() {
    if len(os.Args) > 1 && os.Args[1] == "child" {
        runChildProcess()
        os.Exit(0)
    } else {
        runParentProcess()
    }
}

// runParentProcess 包含父進(jìn)程的邏輯
func runParentProcess() {
    fmt.Printf("父進(jìn)程 (PID: %d)：開始運行...\n", os.Getpid())

    // 1. 在父進(jìn)程中創(chuàng)建一個TCP監(jiān)聽器
    addr := "127.0.0.1:8080"
    listener, err := net.Listen("tcp", addr)
    if err != nil {
        fmt.Printf("父進(jìn)程：創(chuàng)建監(jiān)聽器失敗: %v\n", err)
        return
    }
    fmt.Printf("父進(jìn)程：在 %s 上監(jiān)聽。\n", addr)

    // 2. 從 net.Listener 獲取底層的 *os.File
    // 需要類型斷言，因為 File() 方法是 *net.TCPListener 或 *net.UnixListener 特有的
    tcpListener, ok := listener.(*net.TCPListener)
    if !ok {
        fmt.Printf("父進(jìn)程：監(jiān)聽器不是 *net.TCPListener 類型，無法獲取文件描述符。\n")
        listener.Close()
        return
    }

    file, err := tcpListener.File() // 此操作會復(fù)制文件描述符
    if err != nil {
        fmt.Printf("父進(jìn)程：獲取文件描述符失敗: %v\n", err)
        listener.Close()
        return
    }
    // 確保這個 *os.File 在子進(jìn)程啟動后被父進(jìn)程關(guān)閉，以釋放資源
    // 注意：這里關(guān)閉的是 file 副本，原始 listener 可以選擇繼續(xù)使用或關(guān)閉
    defer file.Close()

    // 3. 準(zhǔn)備子進(jìn)程命令，并將文件描述符添加到 ExtraFiles
    // 假設(shè)子進(jìn)程是當(dāng)前可執(zhí)行文件，通過命令行參數(shù) "child" 區(qū)分
    cmd := exec.Command(os.Args[0], "child")
    cmd.ExtraFiles = []*os.File{file} // 第一個 ExtraFile 將在子進(jìn)程中對應(yīng) FD 3

    // 4. (可選但推薦) 通過環(huán)境變量告知子進(jìn)程文件描述符的索引
    // 這提高了代碼的可讀性和健壯性，特別是有多個 ExtraFiles 時
    cmd.Env = os.Environ()
    cmd.Env = append(cmd.Env, "LISTENER_FD="+strconv.Itoa(3)) // 告知子進(jìn)程監(jiān)聽器是 FD 3

    // 5. 配置子進(jìn)程的輸出，并啟動子進(jìn)程
    cmd.Stdout = os.Stdout
    cmd.Stderr = os.Stderr
    fmt.Printf("父進(jìn)程：啟動子進(jìn)程，傳遞FD %d...\n", file.Fd())
    if err := cmd.Start(); err != nil {
        fmt.Printf("父進(jìn)程：啟動子進(jìn)程失敗: %v\n", err)
        listener.Close() // 如果子進(jìn)程啟動失敗，父進(jìn)程關(guān)閉原始監(jiān)聽器
        return
    }

    fmt.Printf("父進(jìn)程：子進(jìn)程已啟動 (PID: %d)。父進(jìn)程繼續(xù)執(zhí)行...\n", cmd.Process.Pid)

    // 父進(jìn)程可以選擇在此處關(guān)閉自己的監(jiān)聽器，將監(jiān)聽任務(wù)完全交給子進(jìn)程
    // listener.Close()
    // 為了演示，父進(jìn)程保持監(jiān)聽器打開一段時間，模擬父進(jìn)程繼續(xù)處理其他任務(wù)
    time.Sleep(5 * time.Second)
    fmt.Printf("父進(jìn)程：等待子進(jìn)程退出...\n")
    cmd.Wait() // 等待子進(jìn)程退出
    fmt.Printf("父進(jìn)程：子進(jìn)程已退出。父進(jìn)程關(guān)閉原始監(jiān)聽器。\n")
    listener.Close()
}

// runChildProcess 包含子進(jìn)程的邏輯
func runChildProcess() {
    fmt.Printf("子進(jìn)程 (PID: %d)：開始運行...\n", os.Getpid())

    // 1. 從環(huán)境變量獲取文件描述符的索引（如果父進(jìn)程提供了）
    fdStr := os.Getenv("LISTENER_FD")
    fdNum := 3 // ExtraFiles 默認(rèn)從 FD 3 開始