Skip to content

C
coco
Unverified Commit 7a54acaf authored by liuzheng712's avatar liuzheng712
Browse files
Options

feat: update

parent 10228864
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 3 deletions
coco/httpd.py
View file @ 7a54acaf
...@@ -94,11 +94,13 @@ class SSHws(Namespace, BaseWebSocketHandler): ...@@ -94,11 +94,13 @@ class SSHws(Namespace, BaseWebSocketHandler):
child, parent = socket.socketpair() child, parent = socket.socketpair()
self.clients[request.sid]["client"][connection] = Client(parent, self.clients[request.sid]["request"]) self.clients[request.sid]["client"][connection] = Client(parent, self.clients[request.sid]["request"])
self.clients[request.sid]["proxy"][connection] = WSProxy(self, child, self.clients[request.sid]["room"], self.clients[request.sid]["proxy"][connection] = WSProxy(self, child, self.clients[request.sid]["room"],
connection) connection)
self.app.clients.append(self.clients[request.sid]["client"][connection]) self.app.clients.append(self.clients[request.sid]["client"][connection])
self.clients[request.sid]["forwarder"][connection] = ProxyServer(self.app, self.clients[request.sid]["forwarder"][connection] = ProxyServer(self.app,
self.clients[request.sid]["client"][connection]) self.clients[request.sid]["client"][
connection])
self.socketio.start_background_task(self.clients[request.sid]["forwarder"][connection].proxy, asset, self.socketio.start_background_task(self.clients[request.sid]["forwarder"][connection].proxy, asset,
system_user) system_user)
...@@ -135,6 +137,8 @@ class SSHws(Namespace, BaseWebSocketHandler): ...@@ -135,6 +137,8 @@ class SSHws(Namespace, BaseWebSocketHandler):
def on_disconnect(self): def on_disconnect(self):
self.on_leave(self.clients[request.sid]["room"]) self.on_leave(self.clients[request.sid]["room"])
try: try:
for connection in self.clients[request.sid]["client"]:
self.clients[request.sid]["client"][connection].close()
del self.clients[request.sid] del self.clients[request.sid]
except: except:
pass pass
......
coco/recorder.py
View file @ 7a54acaf
...@@ -250,8 +250,8 @@ class S3ReplayRecorder(ReplayRecorder): ...@@ -250,8 +250,8 @@ class S3ReplayRecorder(ReplayRecorder):
def push_to_server(self, session_id): def push_to_server(self, session_id):
self.s3.upload_file( self.s3.upload_file(
filename=os.path.join(self.app.config['LOG_DIR'], session_id + '.replay.gz'), os.path.join(self.app.config['LOG_DIR'], session_id + '.replay.gz'),
bucket=self.bucket) # TODO: 添加日期文件夹及协调一些事宜<liuzheng> self.bucket, self.aws_key) # TODO: 添加日期文件夹及协调一些事宜<liuzheng>
def get_command_recorder_class(config): def get_command_recorder_class(config):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment