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

feat: add resource pool inUsed and del org relation

This commit is contained in:
guoyuqi 2023-07-05 19:44:57 +08:00 committed by fit2-zhao
parent db8626ac0e
commit d653f9f483
2 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
backend/framework/sdk/src/main/java/io/metersphere/sdk/dto/TestResourcePoolDTO.java
View File

@ -6,4 +6,7 @@ import lombok.Data;
@Data
public class TestResourcePoolDTO extends TestResourcePool {
private TestResourceDTO testResourceDTO;
private Boolean inUsed;
}

21
backend/framework/sdk/src/main/java/io/metersphere/sdk/service/TestResourcePoolService.java
View File

@ -7,10 +7,7 @@ import io.metersphere.sdk.dto.TestResourceDTO;
import io.metersphere.sdk.dto.TestResourcePoolDTO;
import io.metersphere.sdk.exception.MSException;
import io.metersphere.sdk.util.*;
import io.metersphere.system.domain.TestResourcePool;
import io.metersphere.system.domain.TestResourcePoolBlob;
import io.metersphere.system.domain.TestResourcePoolExample;
import io.metersphere.system.domain.TestResourcePoolOrganization;
import io.metersphere.system.domain.*;
import io.metersphere.system.mapper.TestResourcePoolBlobMapper;
import io.metersphere.system.mapper.TestResourcePoolMapper;
import io.metersphere.system.mapper.TestResourcePoolOrganizationMapper;
@ -38,6 +35,8 @@ public class TestResourcePoolService {
@Resource
private TestResourcePoolBlobMapper testResourcePoolBlobMapper;
@Resource
private TestResourcePoolOrganizationMapper testResourcePoolOrganizationMapper;
@Resource
private SqlSessionFactory sqlSessionFactory;
@ -154,12 +153,20 @@ public class TestResourcePoolService {
if (testResourcePool == null) {
throw new MSException(Translator.get("test_resource_pool_not_exists"));
}
//删除与组织的关系
deleteOrgRelation(testResourcePoolId);
testResourcePool.setUpdateTime(System.currentTimeMillis());
testResourcePool.setEnable(false);
testResourcePool.setDeleted(true);
testResourcePoolMapper.updateByPrimaryKeySelective(testResourcePool);
}
private void deleteOrgRelation(String testResourcePoolId) {
TestResourcePoolOrganizationExample testResourcePoolOrganizationExample = new TestResourcePoolOrganizationExample();
testResourcePoolOrganizationExample.createCriteria().andTestResourcePoolIdEqualTo(testResourcePoolId);
testResourcePoolOrganizationMapper.deleteByExample(testResourcePoolOrganizationExample);
}
public void updateTestResourcePool(TestResourcePoolDTO testResourcePool) {
checkTestResourcePool(testResourcePool);
testResourcePool.setUpdateTime(System.currentTimeMillis());
@ -192,6 +199,12 @@ public class TestResourcePoolService {
TestResourcePoolDTO testResourcePoolDTO = new TestResourcePoolDTO();
BeanUtils.copyBean(testResourcePoolDTO, pool);
testResourcePoolDTO.setTestResourceDTO(testResourceDTO);
TestResourcePoolOrganizationExample testResourcePoolOrganizationExample = new TestResourcePoolOrganizationExample();
testResourcePoolOrganizationExample.createCriteria().andTestResourcePoolIdEqualTo(pool.getId());
List<TestResourcePoolOrganization> testResourcePoolOrganizations = testResourcePoolOrganizationMapper.selectByExample(testResourcePoolOrganizationExample);
if (pool.getAllOrg() || CollectionUtils.isNotEmpty(testResourcePoolOrganizations)) {
testResourcePoolDTO.setInUsed(true);
}
testResourcePoolDTOS.add(testResourcePoolDTO);
});
return testResourcePoolDTOS;