自己发出的websocket信息不再发送请求了(因为会造成报错),现处理峰值为1w/min消息

This commit is contained in:
Jerry Yan 2022-06-03 23:56:33 +08:00
parent 09da646b99
commit 1cd43be0c5

View File

@ -4,7 +4,6 @@ import time
from mitmproxy import http
import re
import requests
import base64
session = requests.session()
@ -13,9 +12,12 @@ class Writer:
def websocket_message(self, flow: http.HTTPFlow):
re_c = re.search('webcast\d-ws-web-.*\.douyin\.com', flow.request.host)
if re_c:
message = flow.websocket.messages[-1].content
session.post("http://127.0.0.1:5000/message", data=message, headers={
"X-MITM_TS": str(time.time()),
message = flow.websocket.messages[-1]
if message.from_client:
return
content = message.content
session.post("http://127.0.0.1:5000/message", data=content, headers={
"X-MITM-TS": str(time.time()),
"X_REFERER": flow.request.host
}, timeout=(1, 1))