我们在运行GO开发的桌面端软件时,经常遇到一个需求,就是只允许一个实例运行,比如一个游戏,我们只允许一个实例运行,防止多个实例同时运行,导致游戏卡顿。
然而如果我们不做任何处理,只要我们双击一次可执行文件,它就会运行一个新的实例,这样往往不是我们想要的结果。
怎么处理这个问题呢?
我的想法是:在程序运行时,获取到当前的进程的 PID,并将 PID 写入到用户目录下的一个临时文件里,在程序退出的时候,删除这个临时文件。
处理的逻辑一:保持旧程序运行:先读取当前运行的进程的 PID,然后读取用户目录下的临时文件,如果临时文件存在,则解析临时文件的 PID,并检查那个PID是否在运行,如果在运行,则说明程序已经启动。这时候就退出当前的程序。如果临时文件不存在,或者临时文件的 PID 不在运行,则说明程序没有启动,则将当前的PID 写入到临时文件,然后运行程序。
处理的逻辑二:运行新程序:先读取当前运行的进程的 PID,然后读取用户目录下的临时文件,如果临时文件存在,则解析临时文件的 PID,并检查那个PID是否在运行,如果在运行,则说明程序已经启动,则使用命令退出旧程序。并将当前的PID 写入到临时文件,然后运行程序。如果临时文件不存在,或者临时文件的 PID 不在运行,则说明程序没有启动,则将当前的PID 写入到临时文件,然后运行程序。
你可以根据实际情况选择使用哪种处理方式。下面我们就以逻辑二的方式来处理。
在上一篇文章中,我们用 Golang + Sciter 开发了一个简单的桌面软件,我们这回再在 main.go 中增加一段代码,来控制只运行一个实例。
var tempFile string
func checkPid() {
pid := os.Getpid()
tempFile = strings.TrimRight(strings.ReplaceAll(os.TempDir(), "\\", "/"), "/") + "/.googleindexing.lock"
tmpBuf, err := os.ReadFile(tempFile)
if err == nil {
// 文件已存在
tmpPid, _ := strconv.Atoi(string(tmpBuf))
pro, err := os.FindProcess(tmpPid)
if err == nil {
if tmpPid > 1 {
// 启动新的,结束旧的
_ = pro.Kill()
}
}
}
_ = os.WriteFile(tempFile, []byte(strconv.Itoa(pid)), os.ModePerm)
}
// 在初始化的时候就判断 PID,并做相应的判断
func init() {
checkPid()
}