package com.hxzk.gps.service.impl.ThreeModel;
|
|
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.MarsResult;
|
import com.hxzk.gps.controller.System.SystemLogUtil;
|
import com.hxzk.gps.controller.ThreeModel.Results.ThreeModelResult;
|
import com.hxzk.gps.controller.ThreeModel.deo.ThreeModelDeo;
|
import com.hxzk.gps.entity.Company.TbCompany;
|
import com.hxzk.gps.entity.Map.TbMarsHomeset;
|
import com.hxzk.gps.entity.ThreeModel.TbThreemodel;
|
import com.hxzk.gps.mapper.ThreeModel.TbThreemodelMapper;
|
import com.hxzk.gps.service.Company.TbCompanyService;
|
import com.hxzk.gps.service.ThreeModel.TbThreemodelService;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.hxzk.gps.util.GetUser.UserInfoUtil;
|
import com.hxzk.gps.util.MessageUtils.MessageUtils;
|
import com.hxzk.gps.util.Result.ReturnMessage;
|
import com.hxzk.gps.util.Result.TableDto;
|
import com.hxzk.gps.util.Time.TimeUtil;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Value;
|
import org.springframework.stereotype.Service;
|
|
import java.io.File;
|
import java.util.List;
|
|
/**
|
* <p>
|
* 服务实现类
|
* </p>
|
*
|
* @author YuZhiTong
|
* @since 2025-09-03
|
*/
|
@Service
|
public class TbThreemodelServiceImpl extends ServiceImpl<TbThreemodelMapper, TbThreemodel> implements TbThreemodelService {
|
@Autowired
|
TbCompanyService companyService;
|
@Value("${upload.dir-Map-Three}")
|
private String dirMapThree;
|
@Override
|
public ThreeModelResult FindThreemodel(ThreeModelDeo threeModelDeo) {
|
// 开启分页
|
PageHelper.startPage(threeModelDeo.getPageNum(), threeModelDeo.getPageSize());
|
// 创建查询包装器
|
QueryWrapper queryWrapperCompany = new QueryWrapper<>();
|
// 处理部门 ID 为空的情况
|
if ("".equals(threeModelDeo.getDepartmentId())) {
|
List<Long> list = companyService.getChildCompanyIds(UserInfoUtil.getUserCompanyId());
|
queryWrapperCompany.in("companyid", list);
|
} else {
|
List<Long> list = companyService.getChildCompanyIds(threeModelDeo.getDepartmentId());
|
queryWrapperCompany.in("companyid", list);
|
}
|
// 查询三维地图列表
|
List<TbThreemodel> Threemodel = baseMapper.selectList(queryWrapperCompany);
|
PageInfo<TbThreemodel> info = new PageInfo<>(Threemodel);
|
// 封装角色列表
|
TableDto tableDto = new TableDto(info.getList(), info.getPageNum(), info.getPageSize(), info.getTotal());
|
// 获取成功消息
|
String message = MessageUtils.getMessage("Success");
|
// 返回结果
|
return new ThreeModelResult(200, message, tableDto);
|
}
|
|
@Override
|
public ReturnMessage add(TbThreemodel tbThreemodel) {
|
/*
|
声明操作状态
|
* */
|
boolean isSuccess = false;
|
|
try {
|
// 检查是否已存在相同公司ID的记录
|
QueryWrapper<TbThreemodel> checkWrapper = new QueryWrapper<>();
|
checkWrapper.eq("filename", tbThreemodel.getFilename());
|
if (baseMapper.selectCount(checkWrapper) > 0) {
|
return ReturnMessage.ReturnMessageError("ThreeModelExist");
|
}
|
if(tbThreemodel.getCompanyid() == null){
|
tbThreemodel.setCompanyid(UserInfoUtil.getUserCompanyId());
|
}
|
TbCompany company = companyService.getBaseMapper().selectOne(new QueryWrapper<TbCompany>().eq("id",tbThreemodel.getCompanyid()));
|
tbThreemodel.setCompany(company.getSystemname());
|
tbThreemodel.setAddtime(TimeUtil.GetTime());
|
isSuccess = true;
|
baseMapper.insert(tbThreemodel);
|
String fileAdress = dirMapThree+tbThreemodel.getCompany()+"_"+tbThreemodel.getFilename();
|
// 创建文件夹
|
File folder = new File(fileAdress);
|
if (!folder.exists()) {
|
folder.mkdirs();
|
}
|
} catch (Exception e) {
|
log.error("添加三维模型失败");
|
}
|
return SystemLogUtil.handleUpdateResult(isSuccess,"MapModelAdd",tbThreemodel.getCompany());
|
}
|
|
@Override
|
public ReturnMessage update(TbThreemodel tbThreemodel) {
|
/*
|
声明操作状态
|
* */
|
boolean isSuccess = false;
|
|
QueryWrapper queryWrapper = new QueryWrapper<>();
|
queryWrapper.eq("id",tbThreemodel.getId());
|
try {
|
isSuccess= true;
|
baseMapper.update(tbThreemodel,queryWrapper);
|
}catch (Exception e){
|
}
|
return SystemLogUtil.handleUpdateResult(isSuccess,"MapModelUpdate",tbThreemodel.getCompany());
|
}
|
}
|
