コンテキストを使用して Go でリクエストサーキットブレーカーを実装する方法-Golang-php.cn

コンテキストを使用して Go でリクエストサーキットブレーカーを実装する方法

WBOY

リリース： 2023-07-25 09:04:54

オリジナル

1376 人が閲覧しました

コンテキストを使用して Go でリクエストサーキットブレーカーを実装する方法

マイクロサービスアーキテクチャの普及により、さまざまなサービス間の通信がますます頻繁になってきました。サービス間通信では、呼び出しチェーンが非常に長くなる可能性があり、1 つのリクエストの失敗またはタイムアウトによって呼び出しチェーン全体が失敗し、システム全体の可用性に影響を与える可能性があります。単一サービスの障害からシステム全体を保護するために、リクエストサーキットブレークを使用して、特定のサービスへのアクセスを制御および制限できます。この記事では、コンテキストを使用して Go でリクエストサーキットブレーカーを実装する方法を紹介します。

リクエストサーキットブレーカーとは何ですか?

リクエストサーキットブレークは、システム全体を保護するために使用される戦略です。サービスのリクエスト失敗率が所定のしきい値を超えると、リクエストサーキットブレーカーはサービスへのアクセスを即座に拒否し、連鎖的な失敗の発生を回避します。リクエストブレーカーモードは、通常、サーキットブレーカーモード (サーキットブレーカーパターン) と組み合わせて使用され、リクエストが失敗した場合にサーキットブレーカーがすぐにオープンされるため、サービスへのリクエストが拒否され、大量のリクエストが蓄積してサービスが中断されるのを防ぎます。システムリソースの枯渇を引き起こします。

コンテキストを使用して Go でリクエストサーキットブレーカーを実装するサンプルコードは次のとおりです。

package main

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

type CircuitBreaker struct {
    context    context.Context
    cancel     context.CancelFunc
    maxFail    int
    fail       int
    breaker    bool
    resetTime  time.Duration
    breakerMux sync.Mutex
}

func NewCircuitBreaker(maxFail int, resetTime time.Duration) *CircuitBreaker {
    ctx, cancel := context.WithCancel(context.Background())

    circuitBreaker := &CircuitBreaker{
        context:    ctx,
        cancel:     cancel,
        maxFail:    maxFail,
        fail:       0,
        breaker:    false,
        resetTime:  resetTime,
        breakerMux: sync.Mutex{},
    }

    return circuitBreaker
}

func (c *CircuitBreaker) Do(req func() error) error {
    select {
    case <-c.context.Done():
        return fmt.Errorf("circuit breaker is open")
    default:
        if !c.breaker {
            err := req()
            if err == nil {
                c.reset()
            } else {
                c.fail++
                if c.fail >= c.maxFail {
                    c.breakerMux.Lock()
                    c.breaker = true
                    c.breakerMux.Unlock()
                    go time.AfterFunc(c.resetTime, c.reset)
                }
            }
            return err
        } else {
            return fmt.Errorf("circuit breaker is open") 
        }
    }
}

func (c *CircuitBreaker) reset() {
    c.fail = 0
    c.breakerMux.Lock()
    c.breaker = false
    c.breakerMux.Unlock()
    c.cancel()
}

func main() {
    circuitBreaker := NewCircuitBreaker(3, 2*time.Minute)

    // 进行模拟请求
    for i := 0; i < 10; i++ {
        err := circuitBreaker.Do(func() error {
            // 这里执行实际的请求操作，此处只是模拟
            fmt.Println("执行请求...")
            if i%5 == 0 {
                return fmt.Errorf("request failed")
            }
            return nil
        })

        if err != nil {
            fmt.Printf("请求失败: %v
", err)
        } else {
            fmt.Println("请求成功")
        }
    }
}

ログイン後にコピー

上記のサンプルコードでは、CircuitBreaker 構造体を通じて単純なリクエストサーキットブレーカーを実装しました。 CircuitBreaker 構造体には次の属性があります: