test-version22
/
MeterSphere
mirror of https://gitee.com/fit2cloud-feizhiyun/MeterSphere.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix(UI自动化): 修复开源版任务中心报错问题

This commit is contained in:
zhangyong 2022-11-01 14:05:14 +08:00 committed by zhangyongfit2cloud
parent 05d8ec8302
commit 4fbadfa421
5 changed files with 56 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
framework/sdk-parent/sdk/src/main/java/io/metersphere/base/mapper/ext/BaseInformationSchemaTableMapper.java Normal file
View File

@ -0,0 +1,8 @@
package io.metersphere.base.mapper.ext;
import org.apache.ibatis.annotations.Param;
public interface BaseInformationSchemaTableMapper {
String checkExist(@Param("tableName") String tableName);
}

9
framework/sdk-parent/sdk/src/main/java/io/metersphere/base/mapper/ext/BaseInformationSchemaTableMapper.xml Normal file
View File

@ -0,0 +1,9 @@
<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
<mapper namespace="io.metersphere.base.mapper.ext.BaseInformationSchemaTableMapper">
<select id="checkExist" resultType="java.lang.String">
select TABLE_NAME
from information_schema.TABLES
where TABLE_NAME = #{tableName}
</select>
</mapper>

2
framework/sdk-parent/sdk/src/main/java/io/metersphere/base/mapper/ext/BaseTaskMapper.java
View File

@ -8,7 +8,7 @@ import java.util.List;
public interface BaseTaskMapper {
List<TaskCenterDTO> getTasks(@Param("request") TaskCenterRequest request);
List<TaskCenterDTO> getTasks(@Param("request") TaskCenterRequest request, @Param("uiPermission") Boolean uiPermission);
int getRunningTasks(@Param("request") TaskCenterRequest request);

2
framework/sdk-parent/sdk/src/main/java/io/metersphere/base/mapper/ext/BaseTaskMapper.xml
View File

@ -87,6 +87,7 @@
AND t.user_id = #{request.executor}
</if>
)
<if test="uiPermission">
UNION ALL
(SELECT t.id,t.name AS name ,'UI_SCENARIO' AS
executionModule,t.report_type, ifnull(t2.name,'LOCAL') AS actuator, t1.`name` AS executor,t.create_time AS
@ -108,6 +109,7 @@
AND t.user_id = #{request.executor}
</if>
)
</if>
)tt ORDER BY tt.executionTime DESC
</select>

17
framework/sdk-parent/sdk/src/main/java/io/metersphere/task/service/TaskService.java
View File

@ -1,5 +1,6 @@
package io.metersphere.task.service;
import io.metersphere.base.mapper.ext.BaseInformationSchemaTableMapper;
import io.metersphere.base.mapper.ext.BaseScheduleMapper;
import io.metersphere.base.mapper.ext.BaseTaskMapper;
import io.metersphere.commons.constants.MicroServiceName;
@ -30,6 +31,8 @@ public class TaskService {
@Resource
private BaseScheduleMapper baseScheduleMapper;
@Resource
private BaseInformationSchemaTableMapper baseInformationSchemaTableMapper;
@Resource
private BaseCheckPermissionService baseCheckPermissionService;
@Resource
private MicroService microService;
@ -38,6 +41,7 @@ public class TaskService {
private static final String PERF = "PERFORMANCE";
private static final String UI = "UI_SCENARIO";
private static final String UI_SCENARIO_REPORT = "ui_scenario_report";
public List<String> getOwnerProjectIds(String userId) {
Set<String> userRelatedProjectIds = null;
@ -56,7 +60,7 @@ public class TaskService {
if (CollectionUtils.isEmpty(request.getProjects())) {
return new ArrayList<>();
}
return baseTaskMapper.getTasks(request);
return baseTaskMapper.getTasks(request,checkUiPermission());
}
public int getRunningTasks(TaskCenterRequest request) {
@ -110,4 +114,15 @@ public class TaskService {
microService.postForData(MicroServiceName.API_TEST, "/api/automation/stop/batch", reportIds);
}
}
private boolean checkUiPermission(){
try {
String uiScenarioReport = baseInformationSchemaTableMapper.checkExist(UI_SCENARIO_REPORT);
if(StringUtils.isNotEmpty(uiScenarioReport)){
return true;
}
}catch (Exception e){
}
return false;
}
}