test feature

alterway · Sep 20, 2023 · f146cce · f146cce
1 parent bce0a6a
commit f146cce
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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() {