我寫了一個 golang 腳本來掃描開放端口并使用 sync.WaitGourp 來控制 goroutine 的數(shù)量。當(dāng) goroutine 太大時，比如 2000，結(jié)果與 1000 不同。類似于提前退出。代碼顯示如下func worker(wg *sync.WaitGroup) { for job := range jobs { _, err := net.DialTimeout("tcp", fmt.Sprintf("%s:%d", job.host, job.port), time.Millisecond*1500) if err != nil { results <- Result{job, false} } else { results <- Result{job, true} } } wg.Done()}func main() { go func() { for i := 1; i < 65535; i++ { jobs <- Job{host, i} } close(jobs) }() go func() { for result := range results { if result.status { fmt.Println(result.job, "open") } } }() wg := sync.WaitGroup{} for i := 1; i < 1000; i++ { wg.Add(1) go worker(&wg) } wg.Wait()}當(dāng) 1000{127.0.0.1 80} 打開 {127.0.0.1 631} 打開 {127.0.0.1 3306} 打開 {127.0.0.1 6379} 打開 {127.0.0.1 33060} 打開當(dāng) 2000 年{127.0.0.1 80} 打開 {127.0.0.1 631} 打開我希望 2000 輸出所有端口，如 1000
2 回答
泛舟湖上清波郎朗
TA貢獻(xiàn)1818條經(jīng)驗(yàn) 獲得超3個贊
您無需等待 中的兩個“非工作”goroutine main，因此一旦wg.Wait()返回，進(jìn)程就會關(guān)閉，并拆除所有未完成的 goroutine。
由于其中一個正在處理結(jié)果，因此在您看來，好像并非所有任務(wù)都已處理（這是真的）。
慕的地6264312
TA貢獻(xiàn)1817條經(jīng)驗(yàn) 獲得超6個贊
工作人員完成后關(guān)閉結(jié)果通道。在主 goroutine 中處理結(jié)果。
wg := sync.WaitGroup{}
for i := 1; i < 1000; i++ {
wg.Add(1)
go worker(&wg)
}
go func() {
for i := 1; i < 65535; i++ {
jobs <- Job{host, i}
}
// No more jobs, exit from worker loops.
close(jobs)
// Wait for workers to write all results and exit.
wg.Wait()
// No more results, exit from main loop.
close(results)
}()
for result := range results {
if result.status {
fmt.Println(result.job, "open")
}
}
- 2 回答
- 0 關(guān)注
- 131 瀏覽
添加回答
舉報(bào)
0/150
提交
取消