新增: 频道同步输入

Author: TheBlindM

server/src/main/java/com/tshell/controller/SyncChannelController.java

@@ -0,0 +1,45 @@
+package com.tshell.controller;
+
+import com.tshell.common.response.BaseResponse;
+import com.tshell.service.SyncChannelService;
+import io.smallrye.common.annotation.RunOnVirtualThread;
+
+import javax.inject.Inject;
+import javax.ws.rs.*;
+import javax.ws.rs.core.MediaType;
+
+/**
+ *
+ * 频道同步
+ * @author TheBlind
+ */
+@Path("/syncChannel")
+@Produces(MediaType.APPLICATION_JSON)
+@RunOnVirtualThread
+public class SyncChannelController {
+
+    @Inject
+    SyncChannelService syncChannelService;
+
+    @GET
+    public BaseResponse<String> get(@QueryParam("sshChannelId")String sshChannelId){
+        return BaseResponse.ok("success", syncChannelService.get(sshChannelId));
+    }
+
+    @POST
+    @Path("/bind")
+    public BaseResponse bind(@FormParam("syncChannelId")String syncChannelId,@FormParam("sshChannelId") String sshChannelId) {
+        syncChannelService.bind(syncChannelId,sshChannelId);
+        return BaseResponse.ok();
+    }
+
+    @POST
+    @Path("/removeBind")
+    public BaseResponse remove(@FormParam("sshChannelId") String sshChannelId) {
+        syncChannelService.remove(sshChannelId);
+        return BaseResponse.ok();
+    }
+
+
+
+}

server/src/main/java/com/tshell/core/TerminalService.java

@@ -7,6 +7,7 @@
 import com.tshell.module.dto.terminal.SshInitConnectDTO;
 import com.tshell.module.entity.Session;
 import com.tshell.module.entity.SshSession;
+import com.tshell.service.SyncChannelService;
 import com.tshell.service.ConnectionLogService;
 import com.tshell.socket.WebSocket;
 import lombok.extern.slf4j.Slf4j;
@@ -15,7 +16,6 @@
 import javax.inject.Singleton;
 import javax.ws.rs.WebApplicationException;
 import java.io.IOException;
-import java.util.Optional;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
 import java.util.function.Consumer;
@@ -40,6 +40,9 @@ public class TerminalService {
     @Inject
     ConnectionLogService connectionLogService;
 
+    @Inject
+    SyncChannelService syncChannelService;
+
 
     private Parameter.SshParameter convertToSshParameter(SshInitConnectDTO sshInitConnectDTO) {
         String sessionId = sshInitConnectDTO.sessionId();
@@ -141,14 +144,30 @@ private void receiveMsg(Consumer<String> msgHandle, TtyConnector ttyConnector) {
         });
     }
 
+
+
     public void executeCmd(String key, String cmd) {
-        ttyConnectorManager.getConnector(key).ifPresent((ttyConnector -> {
-            try {
-                ttyConnector.write(cmd);
-            } catch (IOException e) {
-                throw new RuntimeException(e);
-            }
-        }));
+        syncChannelService.getSshChannelIds(key).ifPresentOrElse((sshChannelIds) -> {
+            sshChannelIds.forEach((sshChannelId) -> {
+                ttyConnectorManager.getConnector(sshChannelId).ifPresent((ttyConnector -> {
+                    try {
+                        ttyConnector.write(cmd);
+                    } catch (IOException e) {
+                        throw new RuntimeException(e);
+                    }
+                }));
+            });
+
+        }, () -> {
+            ttyConnectorManager.getConnector(key).ifPresent((ttyConnector -> {
+                try {
+                    ttyConnector.write(cmd);
+                } catch (IOException e) {
+                    throw new RuntimeException(e);
+                }
+            }));
+        });
+
     }
 
 
@@ -159,6 +178,7 @@ public void changeWindow(String key, TtySize ttySize) throws IOException {
 
     public void close(String channelId) {
         ttyConnectorManager.getConnector(channelId).ifPresent(ttyConnector -> {
+            syncChannelService.remove(channelId);
             int sessionCount = ttyConnectorManager.getSessionCount(ttyConnector.getSessionId());
             if (sessionCount == 1) {
                 fileManagerService.pauseTransfer(channelId);

server/src/main/java/com/tshell/service/SyncChannelService.java

@@ -0,0 +1,48 @@
+package com.tshell.service;
+
+import cn.hutool.core.collection.CollUtil;
+
+import javax.enterprise.context.ApplicationScoped;
+import java.util.*;
+
+/**
+ * @author TheBlind
+ */
+@ApplicationScoped
+public class SyncChannelService {
+
+
+    public static final Map<String, List<String>> syncChannels = HashMap.newHashMap(5);
+    public static final Map<String, String> sshChannelIdMapChannels = HashMap.newHashMap(5);
+
+
+    public void bind(String channelId, String sshChannelId) {
+        List<String> list = syncChannels.computeIfAbsent(channelId, (k)->new ArrayList<>(3));
+        if (sshChannelIdMapChannels.containsKey(sshChannelId)) {
+            String s = sshChannelIdMapChannels.get(sshChannelId);
+            remove(s);
+        }
+        sshChannelIdMapChannels.put(sshChannelId, channelId);
+        list.add(sshChannelId);
+    }
+
+    public void remove(String sshChannelId) {
+        String channelId = sshChannelIdMapChannels.get(sshChannelId);
+        List<String> list = syncChannels.get(channelId);
+        if(CollUtil.isNotEmpty(list)){
+            list.remove(sshChannelId);
+            sshChannelIdMapChannels.remove(sshChannelId);
+        }
+    }
+
+
+    public Optional<List<String>> getSshChannelIds(String sshChannelId) {
+        String channelId = sshChannelIdMapChannels.get(sshChannelId);
+        return Optional.ofNullable(syncChannels.get(channelId));
+    }
+
+
+    public String get(String sshChannelId) {
+        return sshChannelIdMapChannels.get(sshChannelId);
+    }
+}