如何编写一个自删除的Go程序？

Golang小白一枚，正在不断学习积累知识，现将学习到的知识记录一下，也是将我的所得分享给大家！而今天这篇文章《如何编写一个自删除的Go程序？》带大家来了解一下##content_title##，希望对大家的知识积累有所帮助，从而弥补自己的不足，助力实战开发！

这一切都始于我想要制作一个可以自我更新的程序。我想我需要该程序来下载新版本并运行一个函数来复制新程序并用下载的版本替换原始程序。

我试图让这个问题尽可能小，我怎样才能制作一个调用另一个程序来删除自身的程序，这是我的尝试：

package main

import (
    "flag"
    "fmt"
    "log"
    "os"
    "os/exec"
    "time"
)

func main()  {
    fmt.println("program started")
    remove := flag.bool("rm", false, "removes test")
    flag.parse()

    if *remove {
        // wait 5 seconds to let other program finish
        time.sleep(5 * time.second)
        // try to remove program that started this program
        fmt.println("running rm")
        err := os.remove("./test")
        if err != nil {
            log.fatalf("os.remove() failed with %s\n", err)
        }
    } else {
        // call the second program which will remove ./test which is currently running
        fmt.println("running remove program")
        cmd := exec.command("./remove", "-rm")
        err := cmd.start()
        if err != nil {
            log.fatalf("cmd.run() failed with %s\n", err)
        }
    }
}

这是我通过 cli 调用它的方式。

uberswe$ go build -o test
uberswe$ go build -o remove
uberswe$ ./test
program started
running remove program
uberswe$ ls -la
total 9048
drwxr-xr-x@  6 uberswe  staff      192 Apr 14 15:55 .
drwxr-xr-x@ 56 uberswe  staff     1792 Apr 14 15:36 ..
drwxr-xr-x@  6 uberswe  staff      192 Apr 14 15:55 .idea
-rw-r--r--@  1 uberswe  staff      680 Apr 14 15:55 main.go
-rwxr-xr-x@  1 uberswe  staff  2311528 Apr 14 15:55 remove
-rwxr-xr-x@  1 uberswe  staff  2311528 Apr 14 15:55 test

总而言之：如何制作一个可以自行删除或通过第二个命令/程序删除自身的程序？

如果它可以移植到不同的操作系统，那就更好了。

解决方案

下面的代码是我如何解决这个问题的示例，它适用于 mac os、linux 和 windows。

package main

import (
    "flag"
    "fmt"
    "log"
    "os"
    "os/exec"
    "runtime"
    "time"
)

func main()  {
    fmt.println("program started")
    remove := flag.bool("rm", false, "removes test")
    flag.parse()

    if *remove {
        var err error
        // wait 5 seconds to let other program finish
        time.sleep(5 * time.second)
        // try to remove program that started this program
        fmt.println("running rm")
        if runtime.goos == "windows" {
            err = os.remove("./test.exe")
        } else {
            err = os.remove("./test")
        }
        if err != nil {
            log.fatalf("os.remove() failed with %s\n", err)
        }
    } else {
        var cmd *exec.cmd
        // call the second remove program which will remove ./test which is currently running
        fmt.println("running remove program")
        if runtime.goos == "windows" {
            cmd = exec.command("./remove.exe", "-rm")
        } else {
            cmd = exec.command("./remove", "-rm")
        }
        cmd.stdout = os.stdout
        cmd.stderr = os.stderr
        err := cmd.start()
        if err != nil {
            log.fatalf("cmd.run() failed with %s\n", err)
        }
    }
    fmt.println("finished running program")
}

如果您像这样运行该程序，它将自行删除

uberswe$ go build -o test
uberswe$ go build -o remove
uberswe$ ./test

在 windows 上，我必须进行一项更改，即添加 .exe 扩展名，以便 windows cmd 将二进制文件识别为可执行文件。我能够使应用程序 test 调用 remove 来删除自身。

由于您的目标是应用程序进行自我更新，因此我会将此功能移至第二个“更新程序”应用程序中。否则，可执行文件（取决于操作系统）可能会被锁定，而且您仍然会遇到重新启动应用程序的问题。流程如下：

• 主程序生成更新程序并自行终止
• 更新程序等待主程序终止
• 更新程序替换可执行文件并重新启动应用程序（如果需要）

到这里，我们也就讲完了《如何编写一个自删除的Go程序？》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！