对Golang并发的一些理解

程序代码:


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)。

可以比喻成一个简单的例子:搬砖

 

更深层次的内容还有更优美的代码还得学学嘿嘿

 

posted @ 2021-07-15 15:58  xiaoxiaosen  阅读(63)  评论(0编辑  收藏  举报