Comment implémenter la fonction hot switch du programme Go de manière plus cool

Pendant le développement, nous avons souvent besoin d'interrupteurs thermiques, c'est-à-dire que des fonctions spécifiques peuvent être activées ou désactivées à des moments appropriés pendant l'exécution du programme. Par exemple, l'échantillonnage pprof utilisé dans l'analyse des performances est un interrupteur thermique typique. Cet article explique comment rendre cet interrupteur thermique plus froid.

Avant de présenter la nouvelle solution, passons en revue comment pprof est réalisé dans les programmes Go.

Appel d'interface

L'échantillonnage des performances du programme peut affecter ses capacités de service. Par conséquent, l’échantillonnage en ligne est généralement effectué dans une courte plage de temps spécifiée et nécessite un contrôle efficace des commutateurs.

Pour ce faire, nous introduisons généralement le package net/http/pprof (sa fonction init est liée à la fonction de fonction de routage) dans le code. La fonction d'échantillonnage est activée lors d'un accès externe au service HTTP du port spécifié. Une fois la période d'échantillonnage terminée, la collecte est désactivée.

Le code d'implémentation est le suivant

package main

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

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

Il n'y a bien sûr aucun problème avec cette approche. Nous pouvons l'apprendre et transformer d'autres fonctions de commutation en services HTTP. Mais existe-t-il un autre moyen plus cool ?

信号通知

在信号处理与 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
   }
  }
 }()
}

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

在<span style="font-size: 15px;">main</span>函数中，就可以这样替代<span style="font-size: 15px;">http.ListenAndServe(":8080", nil)</span>了。

package main

import (
  "syscall"
  ...
)

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

在 linux 系统，可以通过<span style="font-size: 15px;">kill -signal_number pid</span>命令向程序发送指定信号。

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

依葫芦画瓢，我们再来一个打印 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))
  }
 }()
}

总结

热开关是一个很简单常用的功能，无非是选择何种触发与等待方式。基于接口的调用更适合于远程控制，基于信号则便于本地控制。

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!