Goプログラムのホットスイッチ機能の実装方法-Golang-php.cn

Goプログラムのホットスイッチ機能をよりかっこよく実装する方法

リリース： 2023-07-21 12:00:03

転載

1121 人が閲覧しました

開発中、サーマルスイッチが必要になることがよくあります。つまり、プログラムの実行中に適切なタイミングで特定の機能をオンまたはオフにすることができます。たとえば、パフォーマンス分析で使用される pprof サンプリングは、典型的なサーマルスイッチです。この記事では、このサーマルスイッチを冷却する方法について説明します。

新しいソリューションを紹介する前に、Go プログラムで pprof がどのように実行されるかを確認しましょう。

インターフェイス呼び出し

プログラムのパフォーマンスサンプリングは、そのサービス機能に影響を与える可能性があります。したがって、オンラインサンプリングは通常、指定された短い時間範囲内で実行され、効果的なスイッチ制御が必要です。

これを実現するために、通常、net/http/pprof パッケージ (その init 関数はルーティング関数関数にバインドされています) をコードに導入します。指定ポートのHTTPサービスに外部からアクセスするとサンプリング機能がオンになり、サンプリング時間が経過すると収集がオフになります。

実装コードは以下の通りです

package main

import (
 "net/http"
 _ "net/http/pprof"
)

func main() {
 go func() {
  _ = http.ListenAndServe(":8080", nil)
 }()
 ...
}

ログイン後にコピー

もちろんこのやり方でも問題ありません。これを学習して、他のスイッチ機能を HTTP サービスに組み込むことができます。しかし、他にもっとクールな方法はあるでしょうか?

信号通知

在信号处理与 Go 程序的优雅退出一文中，我们谈论过信号机制，它用以向应用程序发送某种事件通知。

我们可以将基于接口触发的方式改为信号通知。

首先，构造采样功能函数（对应于 net/http/pprof 包下 init 函数中绑定的路由功能函数）。

func RegisterSignalForProfiling(sig os.Signal) {
 ch := make(chan os.Signal)
 started := false
 signal.Notify(ch, sig)

 go func() {
  var memoryProfile, cpuProfile, traceProfile *os.File
  for range ch {
   if started {
    pprof.StopCPUProfile()
    trace.Stop()
    pprof.WriteHeapProfile(memoryProfile)
    memoryProfile.Close()
    cpuProfile.Close()
    traceProfile.Close()
    started = false
   } else {
    cpuProfile, _ = os.Create("cpu.pprof")
    memoryProfile, _ = os.Create("memory.pprof")
    traceProfile, _ = os.Create("runtime.trace")
    pprof.StartCPUProfile(cpuProfile)
    trace.Start(traceProfile)
    started = true
   }
  }
 }()
}

ログイン後にコピー

在上述函数中，我们定义了接收信号通道ch，通过signal.Notify(ch, sig)将指定的通知信号sig与ch进行绑定。for range ch 将阻塞等待外部信号sig，随着sig信号的到来，交替进入开启或关闭采样的逻辑。

在main函数中，就可以这样替代http.ListenAndServe(":8080", nil)了。

package main

import (
  "syscall"
  ...
)

func main() {
  RegisterSignalForProfiling(syscall.Signal(31))
 ...
}

ログイン後にコピー

在 linux 系统，可以通过kill -signal_number pid命令向程序发送指定信号。

如上代码所示，我们硬编码指定的采样开关信号值是 31。因此，当程序运行起来后，我们在控制台输入kill -31 pid 命令，即可开启采样，再次输入kill -31 pid命令，就关闭了采样。

依葫芦画瓢，我们再来一个打印 goroutine 堆栈信息的热开关函数，是不是很酷？

func RegisterSignalForPrintStack(sig os.Signal) {
 ch := make(chan os.Signal)
 signal.Notify(ch, sig)

 go func() {
  for range ch {
   buffer := make([]byte, 1024*1024*4)
   runtime.Stack(buffer, true)
   fmt.Println(string(buffer))
  }
 }()
}

ログイン後にコピー