程序代码:
func RunTasks(task []string) {
var wg sync.WaitGroup
taskChan := make(chan string,vars.Threads) //创建一个管道
for _,v := range task{ //生产者
wg.Add(1)
taskChan <- v
}
for i:=0;i <vars.Threads;i++{ //消费者
go usescan.ScanPort(taskChan,&wg)
}
close(taskChan) //关闭管道
wg.Wait()
}
func ScanPort(tasks chan string,wg *sync.WaitGroup) {
for task := range tasks{ //遍历管道
_, err := net.DialTimeout("tcp",task,time.Duration(time.Second * 2))
if err != nil{
fmt.Println("[-]",task,"close")
}else {
fmt.Println("[+]",task,"open")
}
wg.Done()
}
}
当主进程执行完后,不管子线程是否执行完,都会被关闭,这在实际的程序中就会造成很多问题。一开始会用sleeptime之类的函数,让主程序执行完后等待一段时间,等到子程序执行完后再结束掉主进程,这个时候问题又出现了,sleep的时间是多久呢?这是不太好把控的,那么就使用wg.wait()了。
wg.wait()写在主程序的最后,相当于一把锁,主程序执行完了是结束不了的。在并发模块中,生产者发布一个任务,执行wg.add(1),而消费者完成一个任务,执行wg.Done()。当生产者发布的所有任务都被消费者完成后,wg.wait()这把锁就开了,程序结束。所以在如果我们写的时候忘记写wg.Done()的话,程序就会一直阻塞。
在这里有一个管道的概念,Channel。生产者发布的任务通过管道传给消费者。当管道里的东西都被拿走了,记得关闭管道,close(chan)。
可以比喻成一个简单的例子:搬砖
更深层次的内容还有更优美的代码还得学学嘿嘿