# -*- coding: utf-8 -*-
|
"""
|
@File : record.py
|
@Time : 2023/9/7 19:10
|
@Author : geekbing
|
@LastEditTime : -
|
@LastEditors : -
|
@Description : 流量录制->生成case功能
|
"""
|
import json
|
import re
|
|
from backend.utils.redis_manager import RedisHelper
|
from apps.schema.request import RequestInfo
|
|
|
class Recorder:
|
def request(self, flow):
|
flow.request.headers["X-Forwarded-For"] = flow.client_conn.address[0]
|
|
async def response(self, flow):
|
if (
|
"47.119.28.171" in flow.request.url
|
or flow.request.method.lower() == "options"
|
or flow.request.url.endswith(("js", "css", "ttf", "jpg", "svg", "gif"))
|
):
|
# 如果是options请求，js等url直接拒绝
|
return
|
addr = flow.client_conn.address[0]
|
record = RedisHelper.get_address_record(addr)
|
if not record:
|
return
|
data = json.loads(record)
|
user_id = data.get("user_id", "")
|
pattern = re.compile(data.get("regex"))
|
if re.findall(pattern, flow.request.url):
|
# 说明已开启录制开关，记录状态
|
request_data = RequestInfo(flow)
|
dump_data = request_data.dumps()
|
await RedisHelper.cache_record(user_id=user_id, request=dump_data)
|
# TODO: 下个版本需要加入ws协议的支持
|
