background/test/goroutine_test.go

/*
 * @Author: your name
 * @Date: 2021-08-02 00:02:17
 * @LastEditTime: 2021-08-24 23:18:40
 * @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")
	
	
}

type TestInherit struct{
	context.Context
	x int32
	
}
func TestTimeoutContext(t * testing.T){
	ctx,_ := context.WithTimeout(context.Background(), time.Second * 1)
	aa := TestInherit{ctx,1}

	go func (aa *TestInherit)  {
		for {
			select {
			case <-aa.Done():
				t.Logf("exit")
				return
			case <- time.After(5 * time.Second)	:
				t.Logf("time out ")
			}
		}	
	}(&aa)
	
	time.Sleep(10 * 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")
}