forked from github/dataease
128 lines
4.7 KiB
Java
128 lines
4.7 KiB
Java
package io.dataease.service.chart;
|
|
|
|
import io.dataease.base.domain.*;
|
|
import io.dataease.base.mapper.ChartGroupMapper;
|
|
import io.dataease.base.mapper.ext.ExtChartGroupMapper;
|
|
import io.dataease.commons.utils.AuthUtils;
|
|
import io.dataease.commons.utils.BeanUtils;
|
|
import io.dataease.controller.request.chart.ChartGroupRequest;
|
|
import io.dataease.controller.request.dataset.DataSetTableRequest;
|
|
import io.dataease.dto.chart.ChartGroupDTO;
|
|
import org.apache.commons.collections.CollectionUtils;
|
|
import org.apache.commons.lang3.StringUtils;
|
|
import org.springframework.stereotype.Service;
|
|
|
|
import javax.annotation.Resource;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.UUID;
|
|
import java.util.stream.Collectors;
|
|
|
|
|
|
@Service
|
|
public class ChartGroupService {
|
|
@Resource
|
|
private ChartGroupMapper chartGroupMapper;
|
|
@Resource
|
|
private ChartViewService chartViewService;
|
|
|
|
@Resource
|
|
private ExtChartGroupMapper extChartGroupMapper;
|
|
|
|
public ChartGroupDTO save(ChartGroup chartGroup) {
|
|
checkName(chartGroup);
|
|
if (StringUtils.isEmpty(chartGroup.getId())) {
|
|
chartGroup.setId(UUID.randomUUID().toString());
|
|
chartGroup.setCreateBy(AuthUtils.getUser().getUsername());
|
|
chartGroup.setCreateTime(System.currentTimeMillis());
|
|
chartGroupMapper.insert(chartGroup);
|
|
} else {
|
|
chartGroupMapper.updateByPrimaryKeySelective(chartGroup);
|
|
}
|
|
ChartGroupDTO ChartGroupDTO = new ChartGroupDTO();
|
|
BeanUtils.copyBean(ChartGroupDTO, chartGroup);
|
|
ChartGroupDTO.setLabel(ChartGroupDTO.getName());
|
|
return ChartGroupDTO;
|
|
}
|
|
|
|
public void delete(String id) {
|
|
ChartGroupRequest ChartGroup = new ChartGroupRequest();
|
|
ChartGroup.setId(id);
|
|
List<ChartGroupDTO> tree = tree(ChartGroup);
|
|
List<String> ids = new ArrayList<>();
|
|
getAllId(tree, ids);
|
|
ChartGroupExample ChartGroupExample = new ChartGroupExample();
|
|
ChartGroupExample.createCriteria().andIdIn(ids);
|
|
chartGroupMapper.deleteByExample(ChartGroupExample);
|
|
// 删除所有chart
|
|
deleteChart(ids);
|
|
}
|
|
|
|
public void deleteChart(List<String> sceneIds) {
|
|
for (String sceneId : sceneIds) {
|
|
chartViewService.deleteBySceneId(sceneId);
|
|
}
|
|
}
|
|
|
|
public ChartGroup getScene(String id) {
|
|
return chartGroupMapper.selectByPrimaryKey(id);
|
|
}
|
|
|
|
public List<ChartGroupDTO> tree(ChartGroupRequest chartGroup) {
|
|
chartGroup.setUserId(String.valueOf(AuthUtils.getUser().getUserId()));
|
|
if(chartGroup.getLevel() == null){
|
|
chartGroup.setLevel(0);
|
|
}
|
|
List<ChartGroupDTO> treeInfo = extChartGroupMapper.search(chartGroup);
|
|
getAll(treeInfo, chartGroup);
|
|
return treeInfo;
|
|
}
|
|
|
|
public void getAll(List<ChartGroupDTO> list, ChartGroupRequest chartGroup) {
|
|
for (ChartGroupDTO obj : list) {
|
|
ChartGroupRequest newChartGroup = new ChartGroupRequest();
|
|
newChartGroup.setUserId(String.valueOf(AuthUtils.getUser().getUserId()));
|
|
newChartGroup.setName(chartGroup.getName());
|
|
newChartGroup.setType(chartGroup.getType());
|
|
newChartGroup.setPid(obj.getId());
|
|
newChartGroup.setSort(chartGroup.getSort());
|
|
List<ChartGroupDTO> treeInfo = extChartGroupMapper.search(newChartGroup);
|
|
obj.setChildren(treeInfo);
|
|
if (CollectionUtils.isNotEmpty(treeInfo)) {
|
|
getAll(treeInfo, chartGroup);
|
|
}
|
|
}
|
|
}
|
|
|
|
public List<String> getAllId(List<ChartGroupDTO> list, List<String> ids) {
|
|
for (ChartGroupDTO dto : list) {
|
|
ids.add(dto.getId());
|
|
if (CollectionUtils.isNotEmpty(dto.getChildren())) {
|
|
getAllId(dto.getChildren(), ids);
|
|
}
|
|
}
|
|
return ids;
|
|
}
|
|
|
|
private void checkName(ChartGroup chartGroup) {
|
|
ChartGroupExample chartGroupExample = new ChartGroupExample();
|
|
ChartGroupExample.Criteria criteria = chartGroupExample.createCriteria();
|
|
if (StringUtils.isNotEmpty(chartGroup.getPid())) {
|
|
criteria.andPidEqualTo(chartGroup.getPid());
|
|
}
|
|
if (StringUtils.isNotEmpty(chartGroup.getType())) {
|
|
criteria.andTypeEqualTo(chartGroup.getType());
|
|
}
|
|
if (StringUtils.isNotEmpty(chartGroup.getName())) {
|
|
criteria.andNameEqualTo(chartGroup.getName());
|
|
}
|
|
if (StringUtils.isNotEmpty(chartGroup.getId())) {
|
|
criteria.andIdNotEqualTo(chartGroup.getId());
|
|
}
|
|
List<ChartGroup> list = chartGroupMapper.selectByExample(chartGroupExample);
|
|
if (list.size() > 0) {
|
|
throw new RuntimeException("Name can't repeat in same group.");
|
|
}
|
|
}
|
|
}
|