Setup Job To Cleanup Empty Sessions

internal/daemon/worker/status.go

@@ -7,6 +7,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"log"
 	"math/rand"
 	"time"
 
@@ -50,7 +51,7 @@ func (w *Worker) startStatusTicking(cancelCtx context.Context, sessionManager se
 		}
 		return common.StatusInterval + time.Duration(f*float64(time.Second))
 	}
-
+	w.startBackgroundCleanup(1*time.Minute, sessionManager)
 	timer := time.NewTimer(0)
 	for {
 		select {
@@ -516,3 +517,36 @@ func (w *Worker) isPastGrace() (bool, time.Time, time.Duration) {
 	v := time.Since(t)
 	return v > u, t, u
 }
+
+func (w *Worker) cleanupPendingSessions(sessionManager session.Manager) {
+	const op = "worker.(Worker).cleanupPendingSessions"
+	closeInfo := make(map[string]*session.ConnectionCloseData)
+	cleanSessionIds := make([]string, 0)
+
+	sessionManager.ForEachLocalSession(func(s session.Session) bool {
+		if len(s.GetLocalConnections()) == 0 {
+			cleanSessionIds = append(cleanSessionIds, s.GetId())
+			event.WriteSysEvent(context.Background(), op, "terminated session due to no active channels", "session_id", s.GetId())
+		}
+		return true
+	})
+
+	if len(closeInfo) > 0 {
+		_ = sessionManager.RequestCloseConnections(context.Background(), closeInfo)
+	}
+	sessionManager.DeleteLocalSession(cleanSessionIds)
+}
+
+func (w *Worker) startBackgroundCleanup(interval time.Duration, sessionManager session.Manager) {
+	go func() {
+		ticker := time.NewTicker(interval)
+		defer ticker.Stop()
+		for {
+			select {
+			case <-ticker.C:
+				log.Println("Cleaning up pending sessions")
+				w.cleanupPendingSessions(sessionManager)
+			}
+		}
+	}()
+}