golang 协程 tcp 扫描退出问题
问题描述
当尝试使用协程扫描主机范围内的开放端口时,主协程会提前退出,没有将 net.dial 的结果插入结果管道中。
解决方案
要解决此问题,需要使用 sync.waitgroup 来确保在关闭结果管道之前,所有协程都已完成扫描任务。
修改后的代码
import ( "fmt" "net" "sync" "time" ) func scanPort(jobChan <-chan int, retChan chan<- string, wg *sync.WaitGroup) { defer wg.Done() for port := range jobChan { target := fmt.Sprintf("149.129.68.235:%d", port) _, err := net.DialTimeout("tcp", target, 100*time.Millisecond) if err == nil { retChan <- fmt.Sprintf("%s is open", target) } } } func main() { jobChan := make(chan int, 5) retChan := make(chan string, 5) var wg sync.WaitGroup go func() { for i := 0; i < 1024; i++ { jobChan <- i } close(jobChan) }() for i := 0; i < 10; i++ { wg.Add(1) go scanPort(jobChan, retChan, &wg) } go func() { for ret := range retChan { fmt.Println(ret) } }() wg.Wait() close(retChan) }
修改说明