erik/converge
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Much imporoved websocket connection closure will now detect closing of

Browse Source 
websockets immediately.
This commit is contained in:
Erik Brakkee 2024-08-13 10:57:05 +02:00
parent cb03b79630
commit bac7a9a6f1
2 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cmd/converge/converge.go
View File

@ -136,13 +136,13 @@ func main() {
sessionService := websocketutil.WebSocketService{
Handler: func(w http.ResponseWriter, r *http.Request, conn net.Conn) {
ctx, cancel := context.WithCancel(context.Background())
websession := websessions.NewSession(conn, ctx)
websession := websessions.NewSession(conn, ctx, cancel)
defer websessions.SessionClosed(websession)
location, err := matchmaker.GetUserLocation(r)
if err != nil {
panic(err)
}
websession.WriteNotifications(location, cancel)
websession.WriteNotifications(location, ctx, cancel)
},
Text: true,
}

9
pkg/server/matchmaker/websessions.go
View File

@ -51,7 +51,8 @@ func (sessions *WebSessions) notifyWebSessions(notification *models.State) {
}
}
func (sessions *WebSessions) NewSession(wsConnection net.Conn, ctx context.Context) *WebSession {
func (sessions *WebSessions) NewSession(wsConnection net.Conn, ctx context.Context,
cancel context.CancelFunc) *WebSession {
sessions.mutex.Lock()
defer sessions.mutex.Unlock()
session := &WebSession{
@ -70,6 +71,7 @@ func (sessions *WebSessions) NewSession(wsConnection net.Conn, ctx context.Conte
session.notifications <- sessions.lastNotification
} else {
log.Printf("Got error reading %v", err)
cancel()
return
}
}
@ -84,19 +86,22 @@ func GetUserLocation(r *http.Request) (*time.Location, error) {
if tzName == "" {
tzName = r.Header.Get("X-Timezone")
}
//log.Printf("Got timezone from request %v %v", tzName, r.URL.Path)
if tzName == "" {
tzName = "UTC"
}
return time.LoadLocation(tzName)
}
func (session *WebSession) WriteNotifications(location *time.Location, cancel context.CancelFunc) {
func (session *WebSession) WriteNotifications(location *time.Location, ctx context.Context, cancel context.CancelFunc) {
timer := time.NewTicker(10 * time.Second)
defer timer.Stop()
// if for some reason we cannot send notifications to the web client then the context is canceled.
defer cancel()
for {
select {
case <-ctx.Done():
return
case notification, ok := <-session.notifications:
if !ok {
log.Println("channel closed")