Go에서 컨텍스트를 사용하여 요청 동시성 제어를 구현하는 방법-Golang-php.cn

Go에서 컨텍스트를 사용하여 요청 동시성 제어를 구현하는 방법

PHPz

풀어 주다： 2023-07-22 09:03:44

원래의

1532명이 탐색했습니다.

고성능 프로그래밍 언어인 Go 언어는 동시 프로그래밍 기능 면에서 항상 뛰어났습니다. 동시 요청을 더 잘 관리하고 제어하기 위해 Go 언어에서는 컨텍스트를 사용하여 요청 동시성 제어를 구현할 수 있습니다. 이 기사에서는 컨텍스트를 사용하여 Go 언어에서 요청 동시성 제어를 구현하는 방법을 소개하고 코드 예제를 통해 이를 보여줍니다.

컨텍스트 소개

Go 언어에서 컨텍스트 패키지는 충돌 요청을 관리하는 방법을 제공합니다. 마감일, 시간 초과, 취소 신호 등을 포함하여 고루틴 간에 요청 전체 데이터를 전달할 수 있습니다. 컨텍스트를 사용하면 동시 요청을 더 잘 관리하고 제어할 수 있습니다.

컨텍스트를 사용하여 요청 동시성 제어를 구현하는 단계

다음은 컨텍스트를 사용하여 요청 동시성 제어를 구현하는 단계입니다.

루트 컨텍스트 만들기: 먼저 루트 컨텍스트를 만들어야 합니다. 루트 컨텍스트는 모든 하위 컨텍스트의 상위 컨텍스트입니다. context.TODO() 또는 context.Background()를 사용하여 빈 루트 컨텍스트를 만들 수 있습니다. context.TODO()或context.Background()来创建一个空的根context。

ctx := context.TODO()

로그인 후 복사

创建子context：接下来，我们可以使用context.WithXXX()函数创建一个子context。这个函数接收一个父context作为参数，并返回一个新的子context。

childCtx := context.WithTimeout(ctx, time.Second*10)

로그인 후 복사

在上述代码中，我们使用context.WithTimeout()函数创建了一个超时为10秒的子context。

处理请求：在处理具体的请求时，我们可以使用context.Context类型的参数接收传递过来的context。在处理请求的过程中，我们可以使用context.Context的方法来获取context的相关信息。

func HandleRequest(ctx context.Context) {
    // 处理请求
}

로그인 후 복사

控制并发：在进行并发请求时，我们可以使用context.Context的Done()方法来判断请求是否被取消或超时。当调用Done()方法时，它会返回一个只读的Channel。如果请求被取消或超时，Done()方法会关闭这个Channel。

go func() {
    for {
        select {
        case <-ctx.Done():
            return
        default:
            // 发起请求
            HandleRequest(ctx)
        }
    }
}()

로그인 후 복사

在上述代码中，我们使用for循环和select语句来监听context的Done()方法。当Done()方法返回时，我们就知道请求已经被取消或超时，可以退出循环。

取消请求：在某些情况下，我们希望取消正在处理的请求。这时，我们可以调用context.CancelFunc类型的cancel()

cancel()

로그인 후 복사

context.WithXXX()

package main

import (
    "context"
    "fmt"
    "sync"
    "time"
)

func Worker(ctx context.Context, wg *sync.WaitGroup) {
    defer wg.Done()
 
    for {
        select {
        case <-ctx.Done():
            // 请求已被取消或超时
            return
        default:
            // 处理请求
            fmt.Println("Handling request...")
            time.Sleep(time.Second * 1)
        }
    }
}

func main() {
    ctx := context.TODO()
    childCtx, cancel := context.WithTimeout(ctx, time.Second*5)
    defer cancel()

    var wg sync.WaitGroup
    for i := 0; i < 5; i++ {
        wg.Add(1)
        go Worker(childCtx, &wg)
    }

    wg.Wait()
    fmt.Println("All requests processed")
}

로그인 후 복사