go提供了sync包和channel來解決協(xié)程同步和通訊。
方式1:
sync.WaitGroup是等待一組協(xié)程結(jié)束，sync.WaitGroup只有3個(gè)方法，Add()添加一個(gè)計(jì)數(shù)，Done()減去一個(gè)計(jì)數(shù)，Wait()阻塞直到所有任務(wù)完成。
package main
import (
"fmt"
"sync"
"time"
)
var wg sync.WaitGroup //定義一個(gè)同步等待的組
func task(i int){
fmt.Println("task...",i)
//耗時(shí)操作任務(wù)，網(wǎng)絡(luò)請(qǐng)求，讀取文件
time.Sleep(time.Second)
wg.Done() //減去一個(gè)計(jì)數(shù)
}
func main(){
for i:= 0;i10;i++{
wg.Add(1) //添加一個(gè)計(jì)數(shù)
go task(i)
}
wg.Wait() //阻塞直到所有任務(wù)完成
fmt.Println("over")
}
運(yùn)行結(jié)果：
task... 9
task... 4
task... 6
task... 0
task... 7
task... 5
task... 1
task... 2
task... 8
task... 3
over
方式2:
利用緩沖信道channel協(xié)程之間通訊，其阻塞等待功能實(shí)現(xiàn)等待一組協(xié)程結(jié)束，不能保證其goroutine按照順序執(zhí)行
package main
import (
"fmt"
)
var ch = make(chan int,10)
func task(i int){
fmt.Println("task...",i)
ch - i
}
func main(){
for i:= 0;i10;i++{
go task(i)
}
for i:= 0;i10;i++{
- ch
}
fmt.Println("over")
}
運(yùn)行結(jié)果：
task... 9
task... 0
task... 1
task... 2
task... 6
task... 7
task... 3
task... 4
task... 8
task... 5
over
方式3:
利用無緩沖的信道channel協(xié)程之間通訊，其阻塞等待功能實(shí)現(xiàn)等待一組協(xié)程結(jié)束，保證了其goroutine按照順序執(zhí)行
package main
import (
"fmt"
"time"
)
var ch = make(chan int)
func task(i int){
fmt.Println("task...",i)
time.Sleep(time.Second)
- ch
}
func main(){
for i:= 0;i10;i++{
go task(i)
ch - i
}
fmt.Println("over")
}
運(yùn)行結(jié)果：
task... 0
task... 1
task... 2
task... 3
task... 4
task... 5
task... 6
task... 7
task... 8
task... 9
over
補(bǔ)充：Go中使用Channel等待所有協(xié)程結(jié)束
讓main方法等待所有協(xié)程執(zhí)行完畢再退出。可能一般思路是設(shè)置一個(gè)共有變量，然后通過修改這個(gè)變量的狀態(tài)。這是通過共享變量來通信的方式，而go要做的是，通過通信來共享內(nèi)存。
1. 按順序執(zhí)行
每次通信進(jìn)行成對(duì)通信，當(dāng)main向協(xié)程發(fā)送一個(gè)寫channel時(shí)，同時(shí)也等待協(xié)程返回一個(gè)讀channel。
這兩個(gè)channel一定是成對(duì)的，所以構(gòu)造一個(gè)結(jié)構(gòu)體
type worker struct {
in chan int
done chan bool
}
func chanDemo1(){
var workers [10]worker
for i := 0; i 10; i++ {
workers[i] = createWorker1(i)
}
for i := 0; i 10; i++ {
workers[i].in - 'a' + i
- workers[i].done
}
for i := 0; i 10; i++ {
workers[i].in - 'A' + i
- workers[i].done
}
}
func createWorker1(id int) worker {
work := worker{
in: make(chan int),
done: make(chan bool),
}
go func() {
for {
fmt.Printf("Work %d receiverd %c\n", id, - work.in)
work.done - true
}
}()
return work
}
func main(){
chanDemo1()
fmt.Println("over")
}
這個(gè)執(zhí)行結(jié)果完全是按照0-9，先小寫再大寫的順序
如果這樣順序執(zhí)行，還要協(xié)程干啥
2. 批量處理
type worker struct {
in chan int
done chan bool
}
func chanDemo1(){
var workers [10]worker
for i := 0; i 10; i++ {
workers[i] = createWorker1(i)
}
for i := 0; i 10; i++ {
workers[i].in - 'a' + i
}
for _, worker := range workers {
- worker.done
}
for i := 0; i 10; i++ {
workers[i].in - 'A' + i
}
for _, worker := range workers {
- worker.done
}
}
func createWorker1(id int) worker {
work := worker{
in: make(chan int),
done: make(chan bool),
}
go func() {
for {
fmt.Printf("Work %d receiverd %c\n", id, - work.in)
work.done - true
}
}()
return work
}
這樣的話，先打印小寫，再打印大寫，但是大小寫時(shí)順序不固定
3. 完全隨機(jī)
func chanDemo1(){
var workers [10]worker
for i := 0; i 10; i++ {
workers[i] = createWorker1(i)
}
for i := 0; i 10; i++ {
workers[i].in - 'a' + i
}
for i := 0; i 10; i++ {
workers[i].in - 'A' + i
}
for _, worker := range workers {
- worker.done
- worker.done
}
}
func createWorker1(id int) worker {
work := worker{
in: make(chan int),
done: make(chan bool),
}
go func() {
for {
fmt.Printf("Work %d receiverd %c\n", id, - work.in)
// 再開一個(gè)協(xié)程
go func() { work.done - true}()
}
}()
return work
}
這種方式就是完全隨機(jī)了
使用channel進(jìn)行樹的遍歷
func (node *Node) TraverseFunc(f func(*Node)){
if node == nil{
return
}
node.Left.TraverseFunc(f)
f(node)
node.Right.TraverseFunc(f)
}
func (node *Node) TraverseWithChannel() chan *Node{
out := make(chan *Node)
go func() {
node.TraverseFunc(func(node *Node) {
out - node
})
close(out)
}()
return out
}
func main(){
var root Node
root = Node{Value:3}
root.Left = Node{}
root.Right = Node{5,nil,nil}
root.Right.Left = new(Node)
root.Left.Right =Node{6,nil,nil}
root.Traverse()
c:=root.TraverseWithChannel()
maxNode := 0
for node := range c{
if node.Value > maxNode{
maxNode = node.Value
}
}
fmt.Println("max node value:", maxNode)
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教。
您可能感興趣的文章:- Go遞歸修改文件名的實(shí)例代碼
- go build 通過文件名后綴實(shí)現(xiàn)不同平臺(tái)的條件編譯操作
- Go并發(fā):使用sync.WaitGroup實(shí)現(xiàn)協(xié)程同步方式
- SpringBoot整合MongoDB實(shí)現(xiàn)文件上傳下載刪除
- golang協(xié)程池模擬實(shí)現(xiàn)群發(fā)郵件功能
- Go 如何批量修改文件名
