131 lines
1.9 KiB
Go
131 lines
1.9 KiB
Go
/*
|
|
* @Author: your name
|
|
* @Date: 2021-08-02 00:02:17
|
|
* @LastEditTime: 2021-08-15 23:15:46
|
|
* @LastEditors: Please set LastEditors
|
|
* @Description: In User Settings Edit
|
|
* @FilePath: \background\test\goroutine_test.go
|
|
*/
|
|
package test
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
|
|
func TestWaitGroup(t *testing.T) {
|
|
var wg sync.WaitGroup
|
|
wg.Add(2)
|
|
|
|
go func () {
|
|
time.Sleep(3*time.Second)
|
|
t.Logf("job 1 done")
|
|
wg.Done()
|
|
}()
|
|
go func () {
|
|
time.Sleep(2 * time.Second)
|
|
t.Logf("job 2 done")
|
|
wg.Done()
|
|
}()
|
|
wg.Wait()
|
|
|
|
t.Log("all finish")
|
|
|
|
|
|
}
|
|
|
|
func TestChanel(t *testing.T){
|
|
exit := make(chan bool)
|
|
go func () {
|
|
for {
|
|
select {
|
|
case <-exit:
|
|
t.Logf("exit 1")
|
|
return
|
|
case <- time.After(1 * time.Second) :
|
|
t.Logf("time out ")
|
|
}
|
|
}
|
|
}()
|
|
go func () {
|
|
for {
|
|
select {
|
|
case <-exit:
|
|
t.Logf("exit 2")
|
|
return
|
|
case <- time.After(1 * time.Second) :
|
|
t.Logf("time out ")
|
|
}
|
|
}
|
|
}()
|
|
time.Sleep(5 * time.Second)
|
|
exit <- true
|
|
t.Logf("finish")
|
|
|
|
|
|
}
|
|
|
|
func TestTimeoutContext(t * testing.T){
|
|
ctx,_ := context.WithTimeout(context.Background(), time.Second)
|
|
|
|
go func () {
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
t.Logf("exit")
|
|
return
|
|
case <- time.After(2 * time.Second) :
|
|
t.Logf("time out ")
|
|
}
|
|
}
|
|
}()
|
|
|
|
time.Sleep(5 * time.Second)
|
|
}
|
|
func TestContext(t *testing.T){
|
|
ctx,cancel := context.WithCancel(context.Background())
|
|
|
|
go func () {
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
t.Logf("exit")
|
|
return
|
|
case <- time.After(1 * time.Second) :
|
|
t.Logf("time out ")
|
|
}
|
|
}
|
|
}()
|
|
|
|
go func () {
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
t.Logf("exit")
|
|
return
|
|
case <- time.After(1 * time.Second) :
|
|
t.Logf("time out ")
|
|
}
|
|
}
|
|
}()
|
|
|
|
go func () {
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
t.Logf("exit")
|
|
return
|
|
case <- time.After(1 * time.Second) :
|
|
t.Logf("time out ")
|
|
}
|
|
}
|
|
}()
|
|
time.Sleep(5 * time.Second)
|
|
cancel()
|
|
|
|
|
|
t.Logf("finish")
|
|
} |