コンテキストを使用して Go でリクエストの同時実行制御を実装する方法

高性能プログラミング言語として、Go 言語は常に同時プログラミング能力において優れています。同時リクエストをより適切に管理および制御するために、Go 言語ではコンテキストを使用してリクエストの同時実行制御を実装できます。この記事では、コンテキストを使用して Go 言語でリクエストの同時実行制御を実装する方法を紹介し、コード例を通じてそれを示します。

コンテキストの概要

Go 言語では、コンテキスト パッケージは衝突リクエストを管理する方法を提供します。期限、タイムアウト、キャンセル信号などを含むリクエスト全体のデータをゴルーチン間で受け渡すことができます。コンテキストを使用することで、同時リクエストをより適切に管理および制御できます。

コンテキストを使用してリクエストの同時実行制御を実装する手順

コンテキストを使用してリクエストの同時実行制御を実装する手順は次のとおりです:

1. ルート コンテキストを作成します。ルートコンテキストを作成する必要があります。ルート コンテキストは、すべての子コンテキストの親コンテキストです。 context.TODO() または context.Background() を使用して、空のルート コンテキストを作成できます。

ctx := context.TODO()

2. サブコンテキストの作成: 次に、context.WithXXX() 関数を使用してサブコンテキストを作成します。この関数は、親コンテキストをパラメータとして受け取り、新しい子コンテキストを返します。

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

上記のコードでは、context.WithTimeout() 関数を使用して、タイムアウトが 10 秒のサブコンテキストを作成します。

3. リクエストの処理: 特定のリクエストを処理するとき、context.Context 型パラメータを使用して、渡されたコンテキストを受け取ることができます。リクエストの処理プロセス中に、context.Context メソッドを使用してコンテキスト関連の情報を取得できます。

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

4. 同時実行性の制御: 同時リクエストを行う場合、context.Context の Done() メソッドを使用して、リクエストがキャンセルされたかタイムアウトしたかを判断できます。 。 Done() メソッドが呼び出されると、読み取り専用の Channel が返されます。リクエストがキャンセルされるかタイムアウトになると、Done() メソッドはチャネルを閉じます。

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

上記のコードでは、for ループと select ステートメントを使用して、コンテキストの Done() メソッドを監視します。 Done() メソッドが返されると、リクエストがキャンセルされたかタイムアウトになったことがわかり、ループを終了できます。

5. リクエストのキャンセル: 場合によっては、処理中のリクエストをキャンセルしたいことがあります。この時点で、context.CancelFunc 型の cancel() メソッドを呼び出してリクエストをキャンセルできます。

cancel()

上記のコードでは、cancel() メソッドを呼び出してリクエストをキャンセルします。

サンプル コード

次に、コンテキストを使用してリクエストの同時実行制御を実装する方法を示す簡単なサンプル コードを示します。

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")
}

上記のコードでは、タイムアウトが 5 秒のルート コンテキストと子コンテキストを作成します。次に、リクエストを処理するための 5 つの goroutine を作成しました。リクエストの処理プロセス中に、コンテキストの Done() メソッドを使用して、リクエストがキャンセルされたかタイムアウトしたかを判断します。最後に、sync.WaitGroup を使用して、すべてのリクエストが処理されるのを待ちます。

概要

コンテキストを使用すると、同時リクエストをより適切に管理および制御できます。この記事では、コンテキストを使用してリクエストの同時実行制御を実装する手順を紹介し、サンプル コードを使用してそれを示します。読者は、この記事の紹介とサンプル コードを通じて、リクエストの同時実行制御を実現するためのコンテキストをよりよく理解し、適用できると信じています。

以上がコンテキストを使用して Go でリクエストの同時実行制御を実装する方法の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。