超高同時接続数RPC：GoのWaitGroupを活用した分散呼び出し

Goでは、syncパッケージのWaitGroupを使用して高並行RPCの分散呼び出しを実行できます。

WaitGroupはgoroutineのグループが終了するのを待つためのカウンターです。カウンタの値が0になると、すべてのgoroutineが完了したことを意味します。

次に、WaitGroupを使用して分散呼び出しを実装するサンプルコードを示します。

package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
// 设置WaitGroup的计数器为2，表示有两个RPC需要调用
wg.Add(2)
// 并发调用两个RPC
go func() {
defer wg.Done() // 调用完成时减少计数器
rpcCall("RPC1")
}()
go func() {
defer wg.Done()
rpcCall("RPC2")
}()
// 等待所有的RPC调用完成
wg.Wait()
}
func rpcCall(rpcName string) {
// 实际的RPC调用逻辑
fmt.Println("Calling", rpcName)
}

上の例では、まずWaitGroupオブジェクトを作成し、そのカウンターを2に設定しています。 これは、呼び出しが必要なRPCが2つあることを意味します。

続いて２つのgoroutineから２つのRPCを並列に呼び出し、各goroutineは呼び出し完了後にwg.Done()を呼び出してカウンタを減らします。

RPC 呼び出しが完了するまで wg.Wait() メソッドが待機し、カウンターが 0 になればメソッドは戻り、プログラムは続行します。

WaitGroupを使うことで、高並行な分散呼び出しを容易に実現でき、すべての呼び出しの完了後に他のロジックを実行できます。