偷偷努力，悄无声息地变强，然后惊艳所有人！哈哈，小伙伴们又来学习啦~今天我将给大家介绍《有什么办法可以阻止默认的golang程序完成》，这篇文章主要会讲到等等知识点，不知道大家对其都有多少了解，下面我们就一起来看一吧！当然，非常希望大家能多多评论，给出合理的建议，我们一起学习，一起进步！
问题内容我有一个使用 websocket 连接和数据库的服务器。有些用户可以通过套接字连接，所以我需要在db中增加他们的“在线”；在他们断开连接的那一刻，我还减少了他们在数据库中的“在线”字段。但为了防止服务器崩溃，我使用在线用户的本地变量副本map[string]int。因此，我需要推迟服务器关闭，直到它完成一个数据库请求，该请求根据我的变量副本减少所有“在线”用户，因为这样套接字连接不会发送默认的“关闭”事件。
我找到了一个包 github.com/xlab/closer，它可以处理一些系统调用，并且可以在程序完成之前执行一些操作，但我的数据库请求不能以这种方式工作（代码如下）
func main() {
...
// trying to handle program finish event
closer.Bind(cleanupSocketConnections(&pageHandler))
...
}
// function that handles program finish event
func cleanupSocketConnections(p *controllers.PageHandler) func() {
return func() {
p.PageService.ResetOnlineUsers()
}
}
// this map[string]int contains key=userId, value=count of socket connections
type PageService struct {
Users map[string]int
}
func (p *PageService) ResetOnlineUsers() {
for userId, count := range p.Users {
// decrease online of every user in program variable
InfoService{}.DecreaseInfoOnline(userId, count)
}
}也许我使用不正确，或者可能有更好的方法来防止默认程序完成？
解决方案
首先，正如您所说，当服务器“崩溃”时执行任务非常复杂，因为崩溃可能意味着很多事情，并且当您的服务器出现严重问题时，没有什么可以保证清理功能的执行。
从工程角度来看（如果在故障时将用户设置为离线如此重要），最好是在另一台服务器上有一个辅助服务，该服务接收用户连接和断开连接事件以及 ping 事件（如果它没有收到任何消息）在设定的超时时间内更新，该服务会认为您的服务器已关闭，并继续将每个用户设置为离线。
回到你的问题，使用 defer 和等待终止信号应该可以覆盖 99% 的情况。我对代码进行了注释以解释逻辑。
// AllUsersOffline is called when the program is terminated, it takes a *sync.Once to make sure this function is performed only
// one time, since it might be called from different goroutines.
func AllUsersOffline(once *sync.Once) {
once.Do(func() {
fmt.Print("setting all users offline...")
// logic to set all users offline
})
}
// CatchSigs catches termination signals and executes f function at the end
func CatchSigs(f func()) {
cSig := make(chan os.Signal, 1)
// watch for these signals
signal.Notify(cSig, syscall.SIGKILL, syscall.SIGTERM, syscall.SIGINT, syscall.SIGQUIT, syscall.SIGHUP) // these are the termination signals in GNU => https://www.gnu.org/software/libc/manual/html_node/Termination-Signals.html
// wait for them
sig := <- cSig
fmt.Printf("received signal: %s", sig)
// execute f
f()
}
func main() {
// the once is used to make sure AllUsersOffline is performed ONE TIME.
usersOfflineOnce := &sync.Once{}
// catch termination signals
go CatchSigs(func() {
// when a termination signal is caught execute AllUsersOffline function
AllUsersOffline(usersOfflineOnce)
})
// deferred functions are called even in case of panic events, although execution is not to take for granted (OOM errors etc)
defer AllUsersOffline(usersOfflineOnce)
// run server
err := server.Run()
if err != nil {
// error logic here
}
// bla bla bla
}我认为你需要看看goroutine和频道。
这里有一些（也许）有用的东西：
https://nathanleclaire.com/blog/2014/02/15/how-to-wait-for-all-goroutines-to-finish-executing-before-continuing/
今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注编程网公众号，一起学习编程~
