Merge pull request #136 from jumpserver/pr@dev@feat_locked_session

feat: locked session can not download

Author: Aaron3S

backend/framework/src/main/java/org/jumpserver/chen/framework/console/dataview/DataView.java

@@ -141,6 +141,12 @@ public void export(String scope, String format) throws SQLException {
                 SessionManager.getCurrentSession().getController().showMessage(MessageLevel.ERROR, MessageUtils.get("DownloadNotAllowed"));
                 return;
             }
+
+            if (SessionManager.getCurrentSession().isLocked()) {
+                SessionManager.getCurrentSession().getController().showMessage(MessageLevel.ERROR, MessageUtils.get("SessionLockedMessage", SessionManager.getCurrentSession().getLockCreator()));
+                return;
+            }
+
             File f = null;
             switch (scope) {
                 case "current":

backend/framework/src/main/java/org/jumpserver/chen/framework/session/Session.java

@@ -21,6 +21,10 @@
 
 public interface Session {
 
+    String getLockCreator();
+
+    boolean isLocked();
+
     List<Common.DataMaskingRule> getDataMaskingRules();
 
     void refreshLastActiveTime();

backend/framework/src/main/java/org/jumpserver/chen/framework/session/impl/BaseSession.java

@@ -63,6 +63,12 @@ public class BaseSession implements Session {
     @Getter
     private LocalDateTime lastActiveTime;
 
+    @Getter
+    private boolean locked = false;
+
+    @Getter
+    private String lockCreator;
+
 
     public BaseSession(Datasource datasource, String remoteAddr) {
         this.datasource = datasource;

backend/framework/src/main/java/org/jumpserver/chen/framework/session/impl/JMSSession.java

@@ -62,8 +62,13 @@ public class JMSSession extends BaseSession {
     @Setter
     private String gatewayId;
 
+
+    @Getter
     private boolean locked = false;
 
+    @Getter
+    private String lockCreator;
+
     private boolean canUpload = false;
     private boolean canDownload = false;
 
@@ -72,16 +77,21 @@ public class JMSSession extends BaseSession {
 
     private boolean closed = false;
 
+
     public void lockSession(String creator) {
         SessionManager.setContext(this.getWebToken());
-        this.getController().showMessage(MessageLevel.ERROR, MessageUtils.get("SessionLockedMessage", creator));
+
+        this.lockCreator = creator;
         this.locked = true;
+        this.getController().showMessage(MessageLevel.ERROR, MessageUtils.get("SessionLockedMessage", this.lockCreator));
     }
 
     public void unloadSession(String creator) {
         SessionManager.setContext(this.getWebToken());
+
         this.getController().showMessage(MessageLevel.SUCCESS, MessageUtils.get("SessionUnlockedMessage", creator));
         this.locked = false;
+        this.lockCreator = null;
     }