如何在程序退出时使用golang删除文件？

一尘不染

如何在程序退出时使用golang删除文件？

我制作了一个命令行应用程序，在其中压缩文件夹并在本地服务器上共享供他人下载。我想做的是关闭服务器后立即删除压缩文件夹的副本。这是我的代码：

func main() {
    //flag to specify whether we will be uploading folder or a single file
    zipped := flag.Bool("z",false,"Use for zipping folders and serving them as a single file on the server.(Deletes the zipped file once the server closes.)")
    save := flag.Bool("s",false,"Use with -z for saving the zipped files locally even after the server closes.")
    flag.Parse()


    if len(flag.Args())>0{

        if *zipped{
            fmt.Println("zipping...")
            flag.Args()[0]=ZipFile()

            if !(*save){
              //I expect this to remove the file when I hit ctrl+c on cmd  
              defer os.Remove(flag.Args()[0])   
                 }
        }
        http.HandleFunc("/",ShareFile)
        fmt.Printf("Sharing file on %s:8080\n",GetOutboundIP())

        log.Fatal(http.ListenAndServe(":8080",nil))
    }else{
        fmt.Println("Invalid usage. No file mentioned. Use wshare -h for help.")
    }

}

当我按下ctrl-
c时，程序退出并且main函数关闭，结果，是否应该执行os.Remove（xyz）？去的游览说，推迟执行该表达式时，函数返回。在这里，我不觉得main有机会退还任何东西。

有什么解决方法可以实现我要完成的任务？我脑子里有一些解决方案，例如等待按键等。但是我希望这个程序超级简单，所以有一种方法可以在服务器关闭/程序退出后立即删除文件，而无需我做任何进一步的输入？

阅读 461

2020-07-02

共1个答案

一尘不染

评论中已经回答了这个问题，但是为了完整起见，我将在此处进行记录。

defer仅当您在其中使用的程序和代码正常运行时才起作用。另一方面，使用命令停止程序或将其杀死会向该程序发送信号，然后异常终止该程序，这不允许该程序defer干净地运行所有语句。

如果要清除OS终止，则可以监听OS信号-基于此处示例的代码：

sigs := make(chan os.Signal, 1)
signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
go func() {
  <- sigs
  cleanupAllTheThings()
  os.Exit(0)
}()

如果您从调用此程序main，它将在您的程序生命周期中侦听OS信号，从而使goroutine一直运行。而且，该cleanupAllTheThings()功能需要编写为尽可能快地运行，而不会受到阻碍，从而使该功能不起作用-
您永远都不知道操作系统何时会因偏见而终止您。

同样，这也不会保护您免受拔出插头或死机的恐慌-因此在启动时或在单独的清理脚本中对旧程序状态进行某种清理通常是有意义的。

2020-07-02