From f146cce40e08115dc479d9e9aab8bdd737870c59 Mon Sep 17 00:00:00 2001
From: libvoid <135131094+libvoid@users.noreply.github.com>
Date: Wed, 20 Sep 2023 17:19:05 +0200
Subject: [PATCH] test feature

---
 pkg/bastion/shell.go | 2 +-
 pkg/bastion/ssh.go   | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/pkg/bastion/shell.go b/pkg/bastion/shell.go
index 3825ded3..7bb84cd1 100644
--- a/pkg/bastion/shell.go
+++ b/pkg/bastion/shell.go
@@ -1668,7 +1668,7 @@ GLOBAL OPTIONS:
 							if err := db.Where("user_id = ?", user.ID).Delete(&dbmodels.UserKey{}).Error; err != nil {
 								return err
 							}
-							if err := db.Model(&dbmodels.Session{}).Where(&dbmodels.Session{User: user, Status: string(dbmodels.SessionStatusActive)}).Update("status", "closed").Error; err != nil {
+							if err := db.Model(&dbmodels.Session{}).Where("user_id = ?", user.ID).Update("status", "closed").Error; err != nil {
 								return err
 							}
 						}
diff --git a/pkg/bastion/ssh.go b/pkg/bastion/ssh.go
index c78ebbb0..4b99a1d8 100644
--- a/pkg/bastion/ssh.go
+++ b/pkg/bastion/ssh.go
@@ -175,13 +175,13 @@ func ChannelHandler(srv *ssh.Server, conn *gossh.ServerConn, newChan gossh.NewCh
 						Model:  gorm.Model{ID: sessionID},
 						Status: string(dbmodels.SessionStatusActive),
 					}
-
+					log.Println(sess.Status)
 					if sess.Status != string(dbmodels.SessionStatusActive) {
 						log.Println("Session", sessionID, "should be closed : closing connection")
 						conn.Close()
 						break
 					}
-					time.Sleep(60 * time.Second)
+					time.Sleep(5 * time.Second)
 				}
 			}(conn, sess.ID)
 			go func() {