Go语言中并发网络地址反向解析的同步机制与常见陷阱

package main

import (
    "fmt"
    "net"
    "strconv"
    "strings"
    // "sync" // 稍后会用到
)

func getHostName(h chan string, ipAdresse string, n int) {
    ip := ipAdresse + strconv.Itoa(n)

    // net.LookupAddr 返回 []string, error
    addr, err := net.LookupAddr(ip) // 修正：第二个返回值是error
    // fmt.Println(err) // 原始代码打印ok，这里应打印err

    if err == nil { // 检查错误是否为nil
        // 确保addr切片不为空，否则可能引发panic
        if len(addr) > 0 {
            h <- ip + " - " + addr[0]
        } else {
            h <- ip + " - No hostname found" // 没有找到主机名
        }
    } else {
        // fmt.Println(err) // 原始代码在这里打印ok，应打印具体的错误
        h <- ip + " - Error: " + err.Error() // 发送错误信息到通道
    }
}

func printer(n chan string) {
    msg := <-n
    fmt.Println(msg)
}

func main() {
    fmt.Println("Please enter your local IP-Adresse e.g 192.168.1.1")

    var ipAdresse_user string
    fmt.Scanln(&ipAdresse_user)

    ipsegment := strings.SplitAfter(ipAdresse_user, ".")
    // 确保ipsegment至少有3个元素，否则可能导致panic
    if len(ipsegment) < 3 {
        fmt.Println("Invalid IP address format. Please enter an address like 192.168.1.1")
        return
    }
    ipadresse_3 := ipsegment[0] + ipsegment[1] + ipsegment[2]

    host := make(chan string)

    for i := 0; i < 55; i++ {
        go getHostName(host, ipadresse_3, i)
        // go printer(host) // 原始代码：这里启动了55个printer协程，不推荐
    }

    // 原始代码的问题在于：主协程在此处直接输出"Finish - Network Scan"并退出
    // 而没有等待之前启动的55个getHostName协程完成
    fmt.Println("Finish - Network Scan")
}

package main

import (
    "fmt"
    "net"
    "strconv"
    "strings"
    "sync" // 引入sync包
    "time" // 用于演示超时或等待
)

// getHostName 函数现在接收一个 WaitGroup 指针
func getHostName(wg *sync.WaitGroup, h chan string, ipAdresse string, n int) {
    defer wg.Done() // 协程结束时调用 Done()

    ip := ipAdresse + strconv.Itoa(n)

    addr, err := net.LookupAddr(ip) // 正确处理 error

    if err == nil {
        if len(addr) > 0 {
            h <- ip + " - " + addr[0]
        } else {
            h <- ip + " - No hostname found"
        }
    } else {
        h <- ip + " - Error: " + err.Error()
    }
}

func main() {
    fmt.Println("Please enter your local IP-Adresse e.g 192.168.1.1")

    var ipAdresse_user string
    fmt.Scanln(&ipAdresse_user)

    ipsegment := strings.SplitAfter(ipAdresse_user, ".")
    if len(ipsegment) < 3 {
        fmt.Println("Invalid IP address format. Please enter an address like 192.168.1.1")
        return
    }
    ipadresse_3 := ipsegment[0] + ipsegment[1] + ipsegment[2]

    var wg sync.WaitGroup // 声明一个 WaitGroup
    host := make(chan string, 55) // 使用带缓冲的通道，避免阻塞生产者

    // 启动一个单独的消费者协程来处理所有结果
    go func() {
        for i := 0; i < 55; i++ {
            msg := <-host
            fmt.Println(msg)
        }
        close(host) // 所有结果消费完毕后关闭通道
    }()

    for i := 0; i < 55; i++ {
        wg.Add(1) // 每启动一个协程，计数器加1
        go getHostName(&wg, host, ipadresse_3, i)
    }

    wg.Wait() // 阻塞主协程，直到所有协程都调用 Done()

    // 给消费者协程一点时间打印最后的输出，或者直接等待通道关闭
    // 由于我们知道有55个结果，消费者会自行消费完
    // 这里可以加一个短暂的等待，或者更严谨地通过另一个WaitGroup同步消费者关闭
    time.Sleep(100 * time.Millisecond) 

    fmt.Println("Finish - Network Scan")
}

// 原始答案的简易解决方案
func main() {
    // ... 其他代码 ...
    for i := 0; i < 55; i++ {
        go getHostName(host, ipadresse_3, i)
        // go printer(host) // 仍然不建议这样启动printer
    }
    // 简单阻塞主协程，等待用户输入
    // 这可以让其他协程有机会运行，但无法保证所有协程都完成
    fmt.Scanln() 
    fmt.Println("Finish - Network Scan")
}