标签:wg 执行 task string Golang 并发 管道 理解 taskChan
程序代码:
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)。
可以比喻成一个简单的例子:搬砖
更深层次的内容还有更优美的代码还得学学嘿嘿
标签:wg,执行,task,string,Golang,并发,管道,理解,taskChan 来源: https://www.cnblogs.com/xiaoxiaosen/p/15015983.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。