package com.hxzk.gps.service.impl.Map;
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.github.pagehelper.PageHelper;
|
import com.github.pagehelper.PageInfo;
|
import com.hxzk.gps.controller.Map.Results.MapResult;
|
import com.hxzk.gps.controller.Map.Results.MarsResult;
|
import com.hxzk.gps.controller.Map.dto.MapTreeTableDto;
|
import com.hxzk.gps.entity.Department.TbDepartMentIcon;
|
import com.hxzk.gps.entity.Department.TbDepartment;
|
import com.hxzk.gps.entity.Map.TbMap;
|
import com.hxzk.gps.mapper.Map.TbMapMapper;
|
import com.hxzk.gps.service.Company.TbCompanyService;
|
import com.hxzk.gps.service.Map.TbMapService;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.hxzk.gps.util.AuToComplete;
|
import com.hxzk.gps.util.GetUser.UserInfoUtil;
|
import com.hxzk.gps.util.MessageUtils.MessageUtils;
|
import com.hxzk.gps.util.Result.TableDto;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Service;
|
|
import java.util.ArrayList;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
import java.util.stream.Collectors;
|
|
/**
|
* <p>
|
* 服务实现类
|
* </p>
|
*
|
* @author YuZhiTong
|
* @since 2025-05-23
|
*/
|
@Service
|
public class TbMapServiceImpl extends ServiceImpl<TbMapMapper, TbMap> implements TbMapService {
|
@Autowired
|
TbCompanyService companyService;
|
@Value("${upload.dir-IP}")
|
private String uploadIp;
|
@Override
|
public MapResult FindMapInfo(MapTreeTableDto mapTreeTableDto) {
|
// 开启分页
|
PageHelper.startPage(mapTreeTableDto.getPageNum(), mapTreeTableDto.getPageSize());
|
// 创建查询包装器
|
QueryWrapper queryWrapperCompany = new QueryWrapper<>();
|
// 处理部门 ID 为空的情况
|
if ("".equals(mapTreeTableDto.getDepartmentId())) {
|
List<Long> list = companyService.getChildCompanyIds(UserInfoUtil.getUserCompanyId());
|
queryWrapperCompany.in("companyid", list);
|
} else {
|
queryWrapperCompany.eq("companyid", mapTreeTableDto.getDepartmentId());
|
}
|
// 查询二维地图列表
|
List<TbMap> maps = baseMapper.selectList(queryWrapperCompany);
|
PageInfo<TbMap> info = new PageInfo<>(maps);
|
// 封装角色列表
|
TableDto tableDto = new TableDto(info.getList(), info.getPageNum(), info.getPageSize(), info.getTotal());
|
// 获取成功消息
|
String message = MessageUtils.getMessage("Success");
|
// 返回结果
|
return new MapResult(200, message, tableDto, mapTreeTableDto.getPageNum(), mapTreeTableDto.getPageSize(), info.getTotal());
|
}
|
|
@Override
|
public List<AuToComplete> FindUserCompanyMap() {
|
QueryWrapper queryWrapper = new QueryWrapper<>();
|
queryWrapper.eq("company", UserInfoUtil.getUserCompany());
|
List<TbMap> mapList = baseMapper.selectList(queryWrapper);
|
List<AuToComplete> addCustomerList = mapList.stream()
|
.map(map -> {
|
String[] mapname = map.getMapname().split("\\.");
|
AuToComplete addCustomer = new AuToComplete();
|
addCustomer.setLabel(mapname[0]);
|
addCustomer.setValue(mapname[0]);
|
return addCustomer;
|
})
|
.collect(Collectors.toList());
|
return addCustomerList;
|
}
|
|
@Override
|
public Map<String, String> FindUserCompanyMapSelect() {
|
QueryWrapper<TbMap> queryWrapper = new QueryWrapper<>();
|
queryWrapper.eq("company", UserInfoUtil.getUserCompany());
|
List<TbMap> mapList = baseMapper.selectList(queryWrapper);
|
|
// 创建Map，key为地图名称，value为图片URL
|
Map<String, String> resultMap = new HashMap<>();
|
for (TbMap map : mapList) {
|
String[] mapname = map.getMapname().split("\\.");
|
resultMap.put(mapname[0], uploadIp+"Map/"+mapname[0]+"."+mapname[1]); // 假设TbMap有getMapUrl方法
|
}
|
|
return resultMap;
|
}
|
}
|
