Go信号处理程序不处理终端窗口关闭事件。

huangapple
go评论126阅读模式
英文:

Go signal handler doesn't handle terminal window closing

问题

我用Go语言编写了一个命令行工具。当通过终端调用该命令行工具时，我希望工具在调用cleanup()函数之前等待Ctrl+C中断(SIGINT)，最后退出。

然而，除了处理SIGINT之外，我还希望该工具能处理用户简单关闭终端窗口的情况。如果窗口关闭，我希望调用相同的cleanup函数。以下是我尝试的示例代码。

  1. package main
  3. import (
  4. 	"fmt"
  5. 	"os"
  6. 	"syscall"
  7. 	"os/signal"
  8. 	"os/exec"
  9. 	"github.com/sirupsen/logrus"
  10. )
  12. func main() {
  13. 	// 解析命令行参数
  14. 	if len(os.Args) < 2 {
  15. 		logrus.Fatalf("缺少必需的参数")
  16. 	}
  17. 	arg := os.Args[1]
  19. 	logrus.Infof("使用参数运行：%s", arg)
  21. 	// 设置信号处理
  22. 	signals := make(chan os.Signal, 1)
  23. 	signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP, syscall.SIGQUIT)
  25. 	done := make(chan bool, 1)
  26. 	go func() {
  27. 		sig := <-signals
  28. 		fmt.Println("")
  29. 		fmt.Println("通过Ctrl+C请求断开连接", sig)
  30. 		done <- true
  31. 	}()
  33. 	logrus.Infof("按下Ctrl+C断开连接。")
  34. 	<-done
  36. 	cleanup()
  38. 	os.Exit(0)
  39. }
  41. func cleanup() {
  42. 	fmt.Println("执行清理任务...")
  43. 	touchFile := exec.Command("touch", "testfile.txt")
  44. 	_, err := touchFile.CombinedOutput()
  46. 	if err != nil {
  47. 		println(err)
  48. 	}
  49. }

这段代码可以处理用户通过Ctrl+C发送中断信号的情况，并调用cleanup函数，但是当用户简单关闭窗口时，cleanup函数不会被调用。我认为包含syscall.SIGHUP应该可以解决这个问题。请指导。

编辑：我在默认的MacOS终端中测试了这段代码，在关闭窗口时会调用cleanup函数，但在使用iTerm时不会调用。

英文:

I have written a cli tool in go. When the cli tool is invoked via the terminal, i want the tool to wait for a Ctrl+C interrupt(SIGINT) before calling a cleanup() function, before finally exiting.

however, as well as handling SIGINT, i also want the tool to handle cases where the terminal window is simply closed by the user. if its closed, i want the same cleanup function to be called. Here is a sample of what i tried.

  1. package main
  3. import (
  4. 	&quot;fmt&quot;
  5. 	&quot;os&quot;
  6. 	&quot;syscall&quot;
  7. 	&quot;os/signal&quot;
  8. 	&quot;os/exec&quot;
  9. 	&quot;github.com/sirupsen/logrus&quot;
  10. )
  12. func main() {
  13. 	// Parse the command-line argument.
  14. 	if len(os.Args) &lt; 2 {
  15. 		logrus.Fatalf(&quot;Missing required argument&quot;)
  16. 	}
  17. 	arg := os.Args[1]
  19. 	logrus.Infof(&quot;Running with argument: %s&quot;, arg)
  21. 	// Set up signal handling.
  22. 	signals := make(chan os.Signal, 1)
  23. 	signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP, syscall.SIGQUIT)
  25. 	done := make(chan bool, 1)
  26. 	go func() {
  27. 		sig := &lt;-signals
  28. 		fmt.Println(&quot;&quot;)
  29. 		fmt.Println(&quot;Disconnection requested via Ctrl+C&quot;, sig)
  30. 		done &lt;- true
  31. 	}()
  33. 	logrus.Infof(&quot;Press Ctrl+C to disconnect.&quot;)
  34. 	&lt;-done
  36. 	cleanup()
  38. 	os.Exit(0)
  39. }
  41. func cleanup() {
  42. 	fmt.Println(&quot;Performing cleanup tasks...&quot;)
  43. 	touchFile := exec.Command(&quot;touch&quot;, &quot;testfile.txt&quot;,)
  44. 	_, err := touchFile.CombinedOutput()
  46. 	if err != nil {
  47. 		println(err)
  48. 	}
  49. }

This works for cases where the user sends an interrupt via Ctrl+C and the cleanup function is called, but the cleanup function is never called when the user simply just closes the window. I though the inclusion of syscall.SIGHUP would work for this. please advise.

Edit: i've tested this in the default MacOS terminal, and the cleanup function is called when closing the window, but not when using iTerm.

答案1

得分: 0

  1. package main
  3. import (
  4. 	redacted
  5. )
  7. func main() {
  8. 	// 解析命令行参数。
  9. 	if len(os.Args) < 2 {
  10. 		logrus.Fatalf("缺少必需的参数")
  11. 	}
  12. 	arg := os.Args[1]
  14. 	log.Infof("使用参数运行: %s", arg)
  16. 	// 设置信号处理。
  17. 	signals := make(chan os.Signal, 1)
  18. 	signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP, syscall.SIGQUIT)
  20. 	done := make(chan bool, 1)
  21. 	go func() {
  22. 		sig := <-signals
  23. 		fmt.Println("")
  24. 		fmt.Println("通过 Ctrl+C 请求断开连接", sig)
  25. 		done <- true
  26. 	}()
  28. 	log.Infof("按下 Ctrl+C 断开连接。")
  29. 	<-done
  31. 	cleanup()
  33. 	os.Exit(0)
  34. }
  36. func cleanup() {
  37. 	fmt.Println("执行清理任务...")
  38. 	touchFile := exec.Command("touch", "testfile.txt")
  39. 	_, err := touchFile.CombinedOutput()
  41. 	if err != nil {
  42. 		println(err)
  43. 	}
  44. }

现在在 iTerm 中添加了 SIGKILL 的功能。

英文: 
  1. package main
  3. import (
  4. redacted
  5. )
  7. func main() {
  8. 	// Parse the command-line argument.
  9. 	if len(os.Args) &lt; 2 {
  10. 		logrus.Fatalf(&quot;Missing required argument&quot;)
  11. 	}
  12. 	arg := os.Args[1]
  14. 	log.Infof(&quot;Running with argument: %s&quot;, arg)
  16. 	// Set up signal handling.
  17. 	signals := make(chan os.Signal, 1)
  18. 	signal.Notify(signals, syscall.SIGINT, syscall.SIGTERM, syscall.SIGHUP, syscall.SIGQUIT)
  20. 	done := make(chan bool, 1)
  21. 	go func() {
  22. 		sig := &lt;-signals
  23. 		fmt.Println(&quot;&quot;)
  24. 		fmt.Println(&quot;Disconnection requested via Ctrl+C&quot;, sig)
  25. 		done &lt;- true
  26. 	}()
  28. 	log.Infof(&quot;Press Ctrl+C to disconnect.&quot;)
  29. 	&lt;-done
  31. 	cleanup()
  33. 	os.Exit(0)
  34. }
  36. func cleanup() {
  37. 	fmt.Println(&quot;Performing cleanup tasks...&quot;)
  38. 	touchFile := exec.Command(&quot;touch&quot;, &quot;testfile.txt&quot;,)
  39. 	_, err := touchFile.CombinedOutput()
  41. 	if err != nil {
  42. 		println(err)
  43. 	}
  44. }

The addition of SIGKILL works now for iTerm.

huangapple
  • 本文由 发表于 2022年12月24日 02:00:06
  • 转载请务必保留本文链接：https://go.coder-hub.com/74902679.html
  • go
  • sigint
  • signals
  • terminal
匿名

发表评论

匿名网友

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen:

确定