def get(self, request, *args, **kwargs):
rsp_data = self.retrieve(request, *args, **kwargs)
if rsp_data.status_code == 200:
return APIRsp(data=rsp_data.data)
else:
return APIRsp(code='400',
msg='无数据',
status=rsp_data.status_code,
data=rsp_data.data)
def post(self, request, *args, **kwargs):
rsp_data = self.create(request, *args, **kwargs)
if rsp_data.status_code == 201:
return APIRsp()
else:
return APIRsp(code='400',
msg='创建用户失败',
status=rsp_data.status_code,
data=rsp_data.data)
def get(self, request, flowid):
try:
qs = FlowTaskAggregateReport.objects.all().filter(flow_id=flowid)
serializer = FlowTaskAggregateReportSerializer(qs, many=True)
return APIRsp(data=serializer.data)
except Exception as e:
logger.exception(f'生成聚合报告异常\n{e}')
return APIRsp(code=500,
msg='生成聚合报告异常',
status=status.HTTP_500_INTERNAL_SERVER_ERROR)
def get(self, request, task_id):
try:
qs = TasksDetails.objects.all().filter(task_id=task_id)
serializer = TasksDetailsSerializer(qs, many=True)
return APIRsp(data=serializer.data)
except Exception as e:
logger.exception(f'获取任务列表异常\n{e}')
return APIRsp(code=400,
msg='获取任务列表异常',
status=status.HTTP_400_BAD_REQUEST)
def get(self, request, flow_id, sampler_id):
try:
qs = RspResult.objects.all().filter(flow_id=flow_id,
sampler_id=sampler_id)
serializer = RspResultSerializer(qs, many=True)
return APIRsp(data=serializer.data)
except Exception as e:
logger.exception(f'获取任务列表异常\n{e}')
return APIRsp(code=400,
msg='获取任务列表异常',
status=status.HTTP_400_BAD_REQUEST)
def get(self, request, jmx_id):
try:
qs = self.get_queryset().filter(jmx_id=jmx_id)
qs = self.filter_queryset(qs)
serializer = self.get_serializer(instance=qs, many=True)
return APIRsp(data=serializer.data)
except Exception as e:
logger.exception(f'获取线程组子元素异常\n{e}')
return APIRsp(code=400,
msg='获取线程组子元素异常',
status=status.HTTP_500_INTERNAL_SERVER_ERROR)
def post(self, request, taskid, flowid):
try:
# values中是数据库中的字段
jtl_url = JtlsSummary.objects.values('jtl_url').get(
task_id=taskid, flow_id=flowid)['jtl_url']
except Exception as e:
logger.exception(f'生成报告异常\n{e}')
return APIRsp(code=400,
msg='获取流水任务失败',
status=status.HTTP_400_BAD_REQUEST)
generate_report.delay(taskid, flowid, jtl_url)
return APIRsp()
def get(self, request, *args, **kwargs):
rsp_data = self.list(request, *args, **kwargs)
if rsp_data.status_code == 200:
data = rsp_data.data
del data['next']
del data['previous']
return APIRsp(data=data)
else:
return APIRsp(code=400,
msg='查询失败',
status=rsp_data.status_code,
data=rsp_data.data)
def post(self, request, flowid):
try:
task_info = TaskFlow.objects.values(
'randomstr', 'celery_task_id').get(id=flowid)
task_flow_str = task_info['randomstr']
celery_task_id = task_info['celery_task_id']
except Exception as e:
logger.exception(f'查杀进行异常\n{e}')
return APIRsp(code=500,
msg='查杀进行异常',
status=status.HTTP_500_INTERNAL_SERVER_ERROR)
kill_task.delay(celery_task_id, task_flow_str)
return APIRsp()
def post(self, request, *args, **kwargs):
try:
self.create(request, *args, **kwargs)
return APIRsp()
except UnsupportedMediaType as e:
logger.exception(f'创建失败\n{e}')
return APIRsp(code=400,
msg='不支持的mediaType',
status=status.HTTP_400_BAD_REQUEST)
except Exception as e:
logger.exception(f'创建失败\n{e}')
return APIRsp(code=400,
msg='创建失败',
status=status.HTTP_400_BAD_REQUEST)
def post(self, request, samplerId, childId):
try:
qs = SamplersChildren.objects.get(id=childId)
if not qs:
return APIRsp(code=400, msg='无效的id')
child_xpath = json.loads(qs.child_info)['xpath']
# 获取jmx的信息
jmxInfo = JmxThreadGroup.objects.all().filter(id=samplerId)
jmx_path = jmxInfo[0].jmx
# 删除子元素
op = OperateJmx(str(jmx_path))
op.remove_node_and_next(child_xpath)
op.save_change()
SamplersChildren.objects.filter(id=childId).delete()
return APIRsp()
except:
return APIRsp(code=400, msg='删除失败')
def post(self, request):
samplerId = request.data.get('samplerId')
childId = request.data.get('childId')
name = request.data.get('name')
if not name:
name = "响应断言"
radioStr = request.data.get('radioStr')
checkedFalseStr = request.data.get('checkedFalseStr')
checkedOrStr = request.data.get('checkedOrStr')
assert_content = request.data.get('assertContent')
assert_str = radioStr + checkedFalseStr + checkedOrStr
if len(assert_content
) == 1 and assert_content[0]['key'] == "" and not childId:
return APIRsp(msg='断言参数为空，不创建响应断言')
try:
# 获取jmx的信息
jmxInfo = JmxThreadGroup.objects.all().filter(id=samplerId)
jmx_path = jmxInfo[0].jmx
sampler_xpath = json.loads(jmxInfo[0].child_info)['xpath']
if childId:
child_info = json.loads(
str(SamplersChildren.objects.get(id=childId).child_info))
child_xpath = child_info['xpath']
new_child_info = ModifyJMX(str(jmx_path)).add_rsp_assert(
sampler_xpath,
name,
assert_str=assert_str,
assert_content=assert_content,
rsp_assert_xpath=child_xpath)
SamplersChildren.objects.filter(id=childId).update(
child_name=name, child_info=json.dumps(new_child_info))
else:
child_info = ModifyJMX(str(jmx_path)).add_rsp_assert(
sampler_xpath,
name,
assert_str=assert_str,
assert_content=assert_content)
s = SamplersChildren(sampler_id=samplerId,
child_name=name,
child_type='rsp_assert',
child_info=json.dumps(child_info))
s.save()
return APIRsp()
except:
return APIRsp(code=400, msg='创建响应断言失败，参数错误！')
def post(self, request, userid, jmxid):
# 判断任务是否存在或者是否绑定了jmx
task = TasksDetails.objects.filter(jmx_id=jmxid, task__task_type=1)
if not task:
try:
task_name = Jmxs.objects.get(id=jmxid).jmx_alias
except:
return APIRsp(code=400, msg='无效的jmxid')
# 创建任务
t = Tasks(task_name=task_name, task_type=1, add_user_id=userid)
t.save()
taskid = t.id
# 将jmx加入任务详情
td = TasksDetails(task_id=taskid, jmx_id=jmxid)
td.save()
else:
taskid = task[0].task_id
running = TaskFlow.objects.filter(task_id=taskid, task_status=0)
if running:
return APIRsp(code=400,
msg='存在运行中的流水任务，请稍后重试！',
status=status.HTTP_200_OK)
try:
jmxs_id = TasksDetails.objects.values('jmx').filter(task_id=taskid)
rsp = Jmxs.objects.values('id', 'jmx').filter(id__in=jmxs_id)
jmxs = []
for jmx in rsp:
jmxs.append(jmx)
except Exception as e:
logger.exception(f'获取任务信息失败\n{e}')
return APIRsp(code=500,
msg='获取任务信息失败',
status=status.HTTP_500_INTERNAL_SERVER_ERROR)
task_flow_str = Tools.random_str()
celery_task_id = run_task.delay(taskid, task_flow_str, jmxs)
url = settings.GRAFANA_ENNDPOINT + f'&var-application={task_flow_str}'
flow = TaskFlow(task_id=taskid,
celery_task_id=celery_task_id,
randomstr=task_flow_str,
task_flow_url=url)
flow.save()
return APIRsp()
def post(self, request):
samplerId = request.data.get('samplerId')
childId = request.data.get('childId')
name = request.data.get('name')
jsonPath = request.data.get('jsonPath')
expectedValue = request.data.get('expectedValue')
expectNull = request.data.get('expectNull')
invert = request.data.get('invert')
if not jsonPath or not name:
return APIRsp(msg='存在参数为空，不创建JSON断言')
try:
# 获取jmx的信息
jmxInfo = JmxThreadGroup.objects.all().filter(id=samplerId)
jmx_path = jmxInfo[0].jmx
sampler_xpath = json.loads(jmxInfo[0].child_info)['xpath']
if childId:
child_info = json.loads(
str(SamplersChildren.objects.get(id=childId).child_info))
child_xpath = child_info['xpath']
new_child_info = ModifyJMX(str(jmx_path)).add_json_assert(
sampler_xpath,
name,
jsonPath,
expectedValue,
expectNull,
invert,
json_aassert_xpath=child_xpath)
SamplersChildren.objects.filter(id=childId).update(
child_name=name, child_info=json.dumps(new_child_info))
else:
child_info = ModifyJMX(str(jmx_path)).add_json_assert(
sampler_xpath, name, jsonPath, expectedValue, expectNull,
invert)
s = SamplersChildren(sampler_id=samplerId,
child_name=name,
child_type='json_assert',
child_info=json.dumps(child_info))
s.save()
return APIRsp()
except:
return APIRsp(code=400, msg='创建Json断言失败，参数错误！')
def post(self, request, childId):
try:
qs = JmxThreadGroup.objects.get(id=childId)
if not qs:
return APIRsp(code=400, msg='无效的id')
jmx_path = str(qs.jmx)
child_info = json.loads(qs.child_info)
# 删除sampler或者csv
op = OperateJmx(str(jmx_path))
op.remove_node_and_next(child_info['xpath'])
op.save_change()
JmxThreadGroup.objects.filter(id=childId).delete()
Csvs.objects.filter(jmx_id=int(qs.jmx_id)).delete()
child_type = str(qs.child_type)
if child_type == 'csv':
os.remove(child_info['filename'])
return APIRsp()
except FileNotFoundError:
return APIRsp()
except:
return APIRsp(code=400, msg='删除失败')
def post(self, request):
samplerId = request.data.get('samplerId')
childId = request.data.get('childId')
name = request.data.get('name')
params = request.data.get('params')
express = request.data.get('express')
match_num = request.data.get('match_num')
if not params or not express or not name:
return APIRsp(msg='存在参数为空，不创建Json提取器')
if not match_num:
match_num = ""
try:
# 获取jmx的信息
jmxInfo = JmxThreadGroup.objects.all().filter(id=samplerId)
jmx_path = jmxInfo[0].jmx
sampler_xpath = json.loads(jmxInfo[0].child_info)['xpath']
if childId:
child_info = json.loads(
str(SamplersChildren.objects.get(id=childId).child_info))
child_xpath = child_info['xpath']
new_child_info = ModifyJMX(str(jmx_path)).add_json_extract(
sampler_xpath,
name,
params,
express,
match_num,
json_extract_xpath=child_xpath)
SamplersChildren.objects.filter(id=childId).update(
child_name=name, child_info=json.dumps(new_child_info))
else:
child_info = ModifyJMX(str(jmx_path)).add_json_extract(
sampler_xpath, name, params, express, match_num)
s = SamplersChildren(sampler_id=samplerId,
child_name=name,
child_type='json_extract',
child_info=json.dumps(child_info))
s.save()
return APIRsp()
except:
return APIRsp(code=400, msg='创建Json提取器失败，参数错误！')
def post(self, request):
samplerId = request.data.get('samplerId')
childId = request.data.get('childId')
name = request.data.get('name')
to_JSR223_param = request.data.get('to_JSR223_param')
express = request.data.get('express')
if not express or not name:
return APIRsp(msg='存在参数为空，不创建JSR223!')
if not to_JSR223_param:
to_JSR223_param = ""
try:
# 获取jmx的信息
jmxInfo = JmxThreadGroup.objects.all().filter(id=samplerId)
jmx_path = jmxInfo[0].jmx
sampler_xpath = json.loads(jmxInfo[0].child_info)['xpath']
if childId:
child_info = json.loads(
str(SamplersChildren.objects.get(id=childId).child_info))
child_xpath = child_info['xpath']
new_child_info = ModifyJMX(str(jmx_path)).add_JSR223(
sampler_xpath,
name,
to_JSR223_param,
express,
JSR223_xpath=child_xpath)
SamplersChildren.objects.filter(id=childId).update(
child_name=name, child_info=json.dumps(new_child_info))
else:
child_info = ModifyJMX(str(jmx_path)).add_JSR223(
sampler_xpath, name, to_JSR223_param, express)
s = SamplersChildren(sampler_id=samplerId,
child_name=name,
child_type='JSR223',
child_info=json.dumps(child_info))
s.save()
return APIRsp()
except:
return APIRsp(code=400, msg='创建JSR223，参数错误！')
def get(self, request, *args, **kwargs):
rsp = self.retrieve(request, *args, **kwargs)
if rsp.status_code == 200:
if rsp.data:
id = rsp.data['id']
children = []
qs = JmxThreadGroup.objects.values(
'id', 'child_name', 'child_type',
'child_info').filter(jmx_id=id)
for q in qs:
q['child_info'] = json.loads(q['child_info'])
children.append(q)
rsp.data = {'id': id, 'children': children}
return APIRsp(data=rsp.data)
return APIRsp(code='400',
msg='无数据',
status=rsp.status_code,
data=rsp.data)
else:
return APIRsp(code='400',
msg='无数据',
status=rsp.status_code,
data=rsp.data)
def post(self, request):
samplerId = request.data.get('samplerId')
childId = request.data.get('childId')
name = request.data.get('name')
params = request.data.get('params')
if not name:
name = "HTTP信息头管理器"
if len(params) == 1 and params[0]['key'] == "" and params[0][
'value'] == "" and not childId:
return APIRsp(msg='头参数为空，不创建头信息')
try:
# 获取jmx的信息
jmxInfo = JmxThreadGroup.objects.all().filter(id=samplerId)
jmx_path = jmxInfo[0].jmx
sampler_xpath = json.loads(jmxInfo[0].child_info)['xpath']
if childId:
child_info = json.loads(
str(SamplersChildren.objects.get(id=childId).child_info))
child_xpath = child_info['xpath']
new_child_info = ModifyJMX(str(jmx_path)).add_header(
sampler_xpath,
name,
headers=params,
header_xpath=child_xpath)
SamplersChildren.objects.filter(id=childId).update(
child_name=name, child_info=json.dumps(new_child_info))
else:
child_info = ModifyJMX(str(jmx_path)).add_header(
sampler_xpath, name, headers=params)
s = SamplersChildren(sampler_id=samplerId,
child_name=name,
child_type='header',
child_info=json.dumps(child_info))
s.save()
return APIRsp()
except:
return APIRsp(code=400, msg='创建header失败，参数错误！')
def post(self, request):
"""
修改csv信息
"""
childId = request.data.get('childId')
name = request.data.get('name')
variableNames = request.data.get('variableNames')
delimiter = request.data.get('delimiter')
ignoreFirstLine = request.data.get('ignoreFirstLine')
recycle = request.data.get('recycle')
stopThread = request.data.get('stopThread')
if childId and variableNames:
csv_base_info = JmxThreadGroup.objects.get(id=childId)
jmx_path = csv_base_info.jmx
child_info = json.loads(csv_base_info.child_info)
xpath = child_info['xpath']
threadType = str(csv_base_info.child_thread)
csv_path = child_info['filename']
csv_info = ModifyJMX(str(jmx_path)).add_csv(
name,
csv_path,
variableNames,
ignoreFirstLine=ignoreFirstLine,
delimiter=delimiter,
recycle=recycle,
stopThread=stopThread,
accord=threadType,
xpath=xpath)
# 更新csv信息
JmxThreadGroup.objects.filter(id=childId).update(
child_name=name, child_info=json.dumps(csv_info))
return APIRsp()
else:
return APIRsp(code=400,
msg='修改失败，参数不完整',
status=status.HTTP_400_BAD_REQUEST)
def post(self, request):
"""
:param request: :param request: {'csv': ,'jmx': , 'user': 1}
:return:
"""
data = {}
# request.POST.get适用于form-data请求获取参数
csv = request.FILES.get('csv')
name = request.POST.get('name')
jmx_id = request.POST.get('jmxId')
user = request.POST.get('userId')
variableNames = request.POST.get('variableNames')
delimiter = request.POST.get('delimiter')
# mult-form-data会将json中的true或者false转换为字符串
ignoreFirstLine = request.POST.get('ignoreFirstLine')
recycle = request.POST.get('recycle')
stopThread = request.POST.get('stopThread')
threadType = request.POST.get('threadType')
if csv and jmx_id and user and variableNames and delimiter and ignoreFirstLine and recycle and stopThread and threadType:
csv_name_ext = os.path.splitext(csv.name)
csv_name = csv_name_ext[0]
csv_ext = csv_name_ext[1]
if csv_ext not in settings.CSV_ALLOWED_FILE_TYPE:
return APIRsp(code=205,
msg='无效的格式，请上传.csv格式的文件',
status=status.HTTP_205_RESET_CONTENT)
csvfile = csv_name + "." + Tools.random_str(9) + csv_ext
path = settings.CSV_URL + csvfile
with open(path, 'wb') as f:
for i in csv.chunks():
f.write(i)
data['csv'] = csvfile
# csv不存在时，接口会报错
data['jmx'] = jmx_id
# user不存在时，接口会报错
data['add_user'] = user
obj = CsvSerializer(data=data)
if obj.is_valid():
obj.save()
jmx_path = Jmxs.objects.values('jmx').get(id=jmx_id)['jmx']
csv_info = ModifyJMX(jmx_path).add_csv(
name,
path,
variableNames,
ignoreFirstLine=Tools.strToBool(ignoreFirstLine),
delimiter=delimiter,
recycle=Tools.strToBool(recycle),
stopThread=Tools.strToBool(stopThread),
accord=threadType)
# 保存csv信息
s = JmxThreadGroup(jmx_id=jmx_id,
child_name=name,
child_type='csv',
child_info=json.dumps(csv_info),
child_thread=threadType)
s.save()
return APIRsp()
return APIRsp(code=500,
msg='添加失败，校验未通过',
status=status.HTTP_400_BAD_REQUEST)
else:
return APIRsp(code=400,
msg='添加失败，参数不完整',
status=status.HTTP_400_BAD_REQUEST)
def post(self, request):
jmx_id = request.data.get('jmxId')
# 修改的时候才需要传入childid信息
child_id = request.data.get('childId')
thread_type = request.data.get('threadType')
sampler_name = request.data.get('samplerName')
method = request.data.get('method')
url = request.data.get('url')
param_type = request.data.get('paramType')
params = request.data.get('params')
jmx_path = Jmxs.objects.values('jmx').get(id=jmx_id)['jmx']
if not jmx_path:
return APIRsp(code=400,
msg='无效的jmxID！',
status=status.HTTP_400_BAD_REQUEST)
if child_id:
# 修改已有sampler
try:
child_info = JmxThreadGroup.objects.values('child_info').get(
id=child_id, child_type='sampler')['child_info']
child_xpath = json.loads(child_info)['xpath']
rst = ModifyJMX(jmx_path).add_sampler(sampler_name,
url,
method,
accord=thread_type,
param_type=param_type,
params=params,
xpath=child_xpath)
JmxThreadGroup.objects.filter(id=child_id).update(
child_name=sampler_name,
child_info=json.dumps(rst),
child_thread=thread_type)
return APIRsp(msg='修改sampler成功！')
except:
return APIRsp(code=400,
msg='修改sampler失败，sampler参数有误！',
status=status.HTTP_400_BAD_REQUEST)
else:
try:
# 新增sampler
rst = ModifyJMX(jmx_path).add_sampler(sampler_name,
url,
method,
accord=thread_type,
param_type=param_type,
params=params)
# 保存sampler信息
s = JmxThreadGroup(jmx_id=jmx_id,
child_name=sampler_name,
child_info=json.dumps(rst),
child_thread=thread_type)
s.save()
return APIRsp(msg='新增sampler成功！', data={'samplerId': s.id})
except:
return APIRsp(code=400,
msg='新增sapmler失败，sampler参数有误！',
status=status.HTTP_400_BAD_REQUEST)
def post(self, request):
"""
:param request: {'jmx': 'name': , 'add_user': 1}
:return:
"""
data = {}
jmx = request.FILES.get('jmx')
jmx_alias = request.POST.get('jmxName')
user = request.POST.get('addUser')
if jmx and user:
jmx_name_ext = os.path.splitext(jmx.name)
jmx_name = jmx_name_ext[0]
jmx_ext = jmx_name_ext[1]
if jmx_ext not in settings.JMX_ALLOWED_FILE_TYPE:
return APIRsp(code=205,
msg='无效的格式，请上传.jmx格式的文件',
status=status.HTTP_205_RESET_CONTENT)
jmxfile = jmx_name + "." + Tools.random_str(9) + jmx_ext
jmxpath = settings.JMX_URL + jmxfile
with open(jmxpath, 'wb') as f:
for i in jmx.chunks():
f.write(i)
jmxinfo = ReadJmx(jmxpath).analysis_jmx()
if not jmxinfo:
return APIRsp(code=400,
msg='添加失败，jmx文件错误',
status=status.HTTP_400_BAD_REQUEST)
samplers_info = jmxinfo[0]
csvs_info = jmxinfo[1]
thread_info = jmxinfo[2]
# jmx路径
data['jmx'] = jmxpath
# jmx名称
if jmx_alias:
data['jmx_alias'] = jmx_alias
else:
data['jmx_alias'] = jmx_name
# user的id不存在时，会校验失败
data['add_user'] = user
# 线程组基础信息
data['thread_base_info'] = json.dumps(thread_info)
obj = JmxsSerializer(data=data)
# 校验数据格式
if obj.is_valid():
obj.save()
jmx_id = obj.data['id']
for sampler in samplers_info:
# 保存sampler信息
sampler_children = sampler['children']
del sampler['children']
sp = JmxThreadGroup(jmx_id=jmx_id,
child_name=sampler['name'],
child_info=json.dumps(sampler),
child_thread=sampler['thread_type'])
sp.save()
# 获取保存后得到的id
sampler_id = sp.id
for child in sampler_children:
sc = SamplersChildren(sampler_id=sampler_id,
child_name=child['child_name'],
child_type=child['child_type'],
child_info=json.dumps(child))
sc.save()
if csvs_info:
for csv in csvs_info:
# 保存csv信息
c = JmxThreadGroup(jmx_id=jmx_id,
child_name=csv['name'],
child_type='csv',
child_info=json.dumps(csv),
child_thread=csv['thread_type'])
c.save()
return APIRsp()
os.remove(jmxpath)
return APIRsp(code=400,
msg='添加失败，参数校验未通过',
status=status.HTTP_400_BAD_REQUEST)
else:
return APIRsp(code=400,
msg='添加失败，未传入文件或用户id',
status=status.HTTP_400_BAD_REQUEST)
def get(self, request):
rsp = {
"menuList": [
{
"id":
101,
"authName":
"JMX管理",
"path":
"jmxs",
"children": [{
"id": 121,
"authName": "JMX列表",
"path": "jmxs",
"children": [],
"order": None
}],
"order":
1
},
{
"id":
103,
"authName":
"任务管理",
"path":
"tasks",
"children": [{
"id": 111,
"authName": "任务列表",
"path": "tasks",
"children": [],
"order": None
}, {
"id": 112,
"authName": "任务流水",
"path": "tasks/flows",
"children": [],
"order": None
}],
"order":
2
},
# {
# "id":145,
# "authName":"报告管理",
# "path":"reports",
# "children":[
# {
# "id":146,
# "authName":"报告列表",
# "path":"reports",
# "children":[
#
# ],
# "order":None
# }
# ],
# "order":5
# },
{
"id":
102,
"authName":
"参数管理",
"path":
"params",
"children": [{
"id": 107,
"authName": "参数列表",
"path": "params",
"children": [],
"order": None
}],
"order":
5
}
]
}
return APIRsp(data=rsp)
def post(self, request):
"""
修改csv信息
"""
data = {}
old_jmx_id = request.data.get('jmxId')
user = request.data.get('userId')
if old_jmx_id:
jmx_info = Jmxs.objects.get(id=old_jmx_id)
jmx_path = str(jmx_info.jmx)
jmx_alias = str(jmx_info.jmx_alias)
temp_jmx_path = settings.JMX_URL + Tools.random_str(9) + '.jmx'
shutil.copyfile(jmx_path, temp_jmx_path)
new_jmx_name = jmx_alias + '.' + Tools.random_str(9)
new_jmx_path = settings.JMX_URL + new_jmx_name + '.jmx'
# 解析JMX
jmxinfo = ReadJmx(temp_jmx_path).analysis_jmx()
if not jmxinfo:
return APIRsp(code=400,
msg='添加失败，jmx文件错误',
status=status.HTTP_400_BAD_REQUEST)
samplers_info = jmxinfo[0]
csvs_info = jmxinfo[1]
thread_info = jmxinfo[2]
# jmx路径
data['jmx'] = new_jmx_path
# jmx别名
data['jmx_alias'] = new_jmx_name
# user的id不存在时，会校验失败
data['add_user'] = user
# 线程组基础信息
data['thread_base_info'] = json.dumps(thread_info)
obj = JmxsSerializer(data=data)
# 校验数据格式
if obj.is_valid():
obj.save()
new_jmx_id = obj.data['id']
for sampler in samplers_info:
# 保存sampler信息
sampler_children = sampler['children']
del sampler['children']
sp = JmxThreadGroup(jmx_id=new_jmx_id,
child_name=sampler['name'],
child_info=json.dumps(sampler),
child_thread=sampler['thread_type'])
sp.save()
# 获取保存后得到的id
sampler_id = sp.id
for child in sampler_children:
sc = SamplersChildren(sampler_id=sampler_id,
child_name=child['child_name'],
child_type=child['child_type'],
child_info=json.dumps(child))
sc.save()
if csvs_info:
for csv in csvs_info:
old_csv_name = csv['name']
new_csv_name = Tools.filename(
Tools.filename(
csv['name'])) + '.' + Tools.random_str(9)
new_csv_path = settings.CSV_URL + new_csv_name + '.csv'
with open(new_jmx_path, 'w') as fw:
with open(temp_jmx_path, 'r') as fr:
for line in fr:
if old_csv_name in line:
line = line.replace(
csv['name'], new_csv_name)
fw.write(line)
shutil.copyfile(csv['filename'], new_csv_path)
csv['filename'] = new_csv_path
c = JmxThreadGroup(jmx_id=new_jmx_id,
child_name=csv['name'],
child_type='csv',
child_info=json.dumps(csv),
child_thread=csv['thread_type'])
c.save()
# 保存csv到数据库
co = Csvs(csv=new_csv_path,
jmx_id=new_jmx_id,
add_user_id=user)
co.save()
else:
shutil.copyfile(temp_jmx_path, new_jmx_path)
os.remove(temp_jmx_path)
return APIRsp()
os.remove(new_jmx_path)
os.remove(temp_jmx_path)
return APIRsp(code=400,
msg='添加失败，参数校验未通过',
status=status.HTTP_400_BAD_REQUEST)
else:
return APIRsp(code=400,
msg='修改失败，参数不完整',
status=status.HTTP_400_BAD_REQUEST)
def post(self, request):
data = {}
jmx_name = request.data.get('jmxName')
sampler_name = request.data.get('samplerName')
method = request.data.get('method')
url = request.data.get('url')
param_type = request.data.get('paramType')
params = request.data.get('params')
user = request.data.get('addUser')
if not sampler_name or not method or not url:
return APIRsp(code=400,
msg='创建jmx失败，接口名称、方法、url必传',
status=status.HTTP_400_BAD_REQUEST)
template_path = settings.JMX_URL + 'template.jmx'
new_jmxpath = settings.JMX_URL + jmx_name + "." + Tools.random_str(
9) + '.jmx'
shutil.copyfile(template_path, new_jmxpath)
try:
ModifyJMX(new_jmxpath).add_sampler(sampler_name,
url,
method,
param_type=param_type,
params=params)
except:
os.remove(new_jmxpath)
return APIRsp(code=400,
msg='创建jmx失败，参数错误！',
status=status.HTTP_400_BAD_REQUEST)
jmxinfo = ReadJmx(new_jmxpath).analysis_jmx(upload=False)
if not jmxinfo:
return APIRsp(code=400,
msg='添加失败，jmx文件错误！',
status=status.HTTP_400_BAD_REQUEST)
sampler_info = jmxinfo[0][0]
thread_info = jmxinfo[2]
# jmx路径
data['jmx'] = new_jmxpath
# jmx名称
data['jmx_alias'] = jmx_name
# user的id不存在时，会校验失败
data['add_user'] = user
# 线程组基础信息
data['thread_base_info'] = json.dumps(thread_info)
obj = JmxsSerializer(data=data)
# 校验数据格式
if obj.is_valid():
obj.save()
jmx_id = obj.data['id']
if sampler_info:
# 保存sampler信息
del sampler_info['children']
s = JmxThreadGroup(jmx_id=jmx_id,
child_name=sampler_name,
child_info=json.dumps(sampler_info),
child_thread=sampler_info['thread_type'])
s.save()
return APIRsp(data={'samplerId': s.id}) # sampler的id
os.remove(new_jmxpath)
return APIRsp(code=400,
msg='添加失败，参数校验未通过',
status=status.HTTP_400_BAD_REQUEST)