Golang之context包

àì夳堔傛蜴生んèń 2022-10-15 10:50 214阅读 0赞

context经常发挥的作用：

*  多个goroutine之间的信息交互
 *  进行超时控制
 *  上下文控制  
    大概说明就是，进行进程之间的控制。  
    常用的方法有：Background()，WithCancel()，WithDeadline()，Background()
 *  Background()方法，初始化一个空的context，没有过期时间，一般用在多线程的根部。
 *  WithCancel()  
    会继承一个context，如果父context被结束，或者超时，则子context也会被结束。  
    和Background()不同之处在于，这个方法在继承父context之后，会返回两个变量，第一个变量是context.Context，第二个变量是cancel函数，如果调用这个函数，就会导致这个context进程结束。

func main () {
        
    	ctx , cancel:= context.WithCancel(context.Background())
    	defer cancel()
    	t := gen(ctx)
    	for y := range t {
        
    		fmt.Println(y)
    		if y == 10 {
        
    			break
    		}
    	}
    }
    
    func gen(ctx context.Context) chan int{
        
    	context.Background()
    	dst := make(chan int , 1)
    	n := 1
    	go func() {
        
    		for  {
        
    			select {
        
    			case <-ctx.Done():
    				return
    			case dst <- n:
    				n++
    			}
    		}
    	}()
    	return dst
    }

*  WithDeadline()  
    设置一个过期的时间，如果超过这个时间，此进程就会自动停止。

func main () {
        
    	ctx , cancel:= context.WithDeadline(context.Background() , time.Now().Add(time.Second))
    	defer cancel()
    	t := gen(ctx)
    	for y := range t {
        
    		fmt.Println(y)
    	}
    }
    
    func gen(ctx context.Context) chan int{
        
    	context.Background()
    	dst := make(chan int , 1)
    	n := 1
    	go func() {
        
    		for  {
        
    			select {
        
    			case <-ctx.Done():
    				return
    			case dst <- n:
    				n++
    			}
    		}
    	}()
    	return dst
    }

你会发现只执行1s就会结束程序

*  WithTimeout()  
    这个方法与iWithDeadline不同的是此方法，第二个参数是Time.Duration，作用相似，会有一个计时器，如果超过这个时间，会自动停止

func main () {
        
    	ctx , cancel:= context.WithTimeout(context.Background() , time.Second)
    	defer cancel()
    	t := gen(ctx)
    	for y := range t {
        
    		fmt.Println(y)
    	}
    }
    
    func gen(ctx context.Context) chan int{
        
    	context.Background()
    	dst := make(chan int , 1)
    	n := 1
    	go func() {
        
    		for  {
        
    			select {
        
    			case <-ctx.Done():
    				return
    			case dst <- n:
    				n++
    			}
    		}
    	}()
    	return dst
    }

和上面的WithDeadline()相似，也是只执行1s中结束

*  WithValue()  
    此方法是创建一个context会携带一个key，value值，并用于上下文的分享，也就是线程之间的通信，获得这个值，需要调用context.Value()根据key值返回value，如果没有返回nil

func main () {
        
    	ctx:= context.WithValue(context.Background() , "name" , "test")
    	fmt.Println(ctx.Value("name"))
    
    }