WebSSH进阶之实时监控和强制下线，互联网公司面试流程&面试技巧

最新推荐文章于 2024-07-28 22:05:31 发布

2401_84023710

最新推荐文章于 2024-07-28 22:05:31 发布

阅读量836

点赞数 15

分类专栏： 2024年程序员学习文章标签：面试职场和发展

本文链接：https://blog.csdn.net/2401_84023710/article/details/137254181

版权

2024年程序员学习专栏收录该内容

244 篇文章 0 订阅

订阅专栏

is_connecting=True

)

async_to_sync(self.channel_layer.group_add)(

self.group_name,

self.channel_name

)

self.accept()

WebSocket连接成功后，连接ssh

self.ssh = SSHBridge(self.therecord, websocket=self)

self.ssh.connect(**ssh_connect_args)

def disconnect(self, close_code):

将连接状态置为False

self.therecord.is_connecting = False

self.therecord.save()

async_to_sync(self.channel_layer.group_discard)(

self.group_name,

self.channel_name

)

self.ssh.close()

def receive(self, text_data=None):

text_data = json.loads(text_data)

if text_data.get(‘flag’) == ‘resize’:

self.ssh.resize_pty(cols=text_data[‘cols’], rows=text_data[‘rows’])

else:

self.ssh.shell(data=text_data.get(‘data’, ‘’))

def ssh_message(self, event):

self.send(text_data=json.dumps(

event[‘message’]

))

在connect连接建立时新建一条记录，存储主机、用户、group_name、channel_name以及初始窗口的cols、rows信息，同时标记is_connecting为True，这里的group_name命名与文章『堡垒机的核心武器：WebSSH录像实现』中我们定义的录像文件名规则一致，另外将这篇文章中新建录像记录的操作从SSHBridge.record中给转到了连接建立的connect中来，更合理也更方便

在disconnect连接关闭时，将is_connecting标记为False，这样我们在前端页面上就可以根据这个标记来判断WebSSH是否正在连接，如果连接则展示监控和强制结束按钮，否则展示播放和命令提取按钮

同时添加个ssh_message方法，用来接收发送到组的数据

到这里，我们已经将WebSSH改造成了支持layer的模式，那么接下来就是要在用户点击监控的时候将用户与服务端建立的连接channel加入到上述group中

新建一个名为MonitorConsumer的consumer，主要用来处理监控连接

class MonitorConsumer(WebsocketConsumer):

def connect(self):

pk = self.scope[‘url_route’][‘kwargs’].get(‘id’)

self.group_name = Record.objects.get(id=pk).group

async_to_sync(self.channel_layer.group_add)(

self.group_name,

self.channel_name

)

self.accept()

判断用户已经结束了这个webssh连接时就关闭监控

self.connecting = Record.objects.get(id=pk).is_connecting

if not self.connecting:

self.close()

def disconnect(self, close_code):

async_to_sync(self.channel_layer.group_discard)(

self.group_name,

self.channel_name

)

self.close()

def receive(self, text_data=None):

pass

def ssh_message(self, event):

self.send(text_data=json.dumps(

event[‘message’]

))

MonitorConsumer与SSHConsumer有两个地方不一样，其一是SSHConsumer中我们直接新生成了个group_name，而MonitorConsumer中需要在connect时获取到要监控的ID，然后通过ID拿到group_name，将monitor连接加入到这个group，其二是监控只能看，不能操作，所以也不需要前端发送数据的term.on和Consumer的receive处理数据

最后需要修改SSHBridge方法中发送给websocket的指令，从self.websocket.send改为发送到group的模式，如下

async_to_sync(self.websocket.channel_layer.group_send)(

self.group_name,

{

‘type’: ‘ssh.message’,

‘message’: message

}

)

至此，监控功能就算完成了，什么？前端页面怎么弄？参考下之前的WebSSH界面，几乎可以完全复制

踢用户下线

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数前端工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！