Move server shutdown into group.go.

This commit is contained in:
Juliusz Chroboczek
2021-09-11 15:42:14 +02:00
parent 184bd30077
commit c860fb96a3
2 changed files with 7 additions and 6 deletions
+6 -2
View File
@@ -701,8 +701,12 @@ func kickall(g *Group, message string) {
})
}
func (g *Group) Shutdown(message string) {
kickall(g, message)
func Shutdown(message string) {
Range(func(g *Group) bool {
g.SetLocked(true, message)
kickall(g, message)
return true
})
}
type warner interface {
+1 -4
View File
@@ -70,10 +70,7 @@ func Serve(address string, dataDir string) error {
}
}
s.RegisterOnShutdown(func() {
group.Range(func(g *group.Group) bool {
go g.Shutdown("server is shutting down")
return true
})
group.Shutdown("server is shutting down")
})
server.Store(s)