package org.springblade.modules.yw.service.impl;
|
|
import com.alibaba.fastjson.JSONArray;
|
import com.alibaba.fastjson.JSONObject;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import org.apache.logging.log4j.util.Strings;
|
import org.geotools.feature.FeatureCollection;
|
import org.geotools.feature.FeatureIterator;
|
import org.geotools.geojson.feature.FeatureJSON;
|
import org.geotools.geojson.geom.GeometryJSON;
|
import org.locationtech.jts.geom.Geometry;
|
import org.opengis.feature.simple.SimpleFeature;
|
import org.springblade.common.utils.FileUtil;
|
import org.springblade.common.utils.ShapeFileUtil;
|
import org.springblade.core.tool.api.R;
|
import org.springblade.core.tool.utils.BeanUtil;
|
import org.springblade.modules.yw.entity.FirmInfo;
|
import org.springblade.modules.yw.entity.GeomInfoEntity;
|
import org.springblade.modules.yw.mapper.GeomInfoMapper;
|
import org.springblade.modules.yw.service.IFirmInfoService;
|
import org.springblade.modules.yw.service.IGeomInfoService;
|
import org.springblade.modules.yw.vo.GeomInfoVO;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.multipart.MultipartFile;
|
|
import java.io.*;
|
import java.util.*;
|
|
/**
|
* 空间信息表 服务实现类
|
*
|
* @author BladeX
|
* @since 2024-11-05
|
*/
|
@Service
|
public class GeomInfoServiceImpl extends ServiceImpl<GeomInfoMapper, GeomInfoEntity> implements IGeomInfoService {
|
|
@Autowired
|
private IFirmInfoService firmInfoService;
|
|
/**
|
* 自定义分页
|
* @param page
|
* @param geomInfo
|
* @return
|
*/
|
@Override
|
public IPage<GeomInfoVO> selectGeomInfoPage(IPage<GeomInfoVO> page, GeomInfoVO geomInfo) {
|
return page.setRecords(baseMapper.selectGeomInfoPage(page, geomInfo));
|
}
|
|
/**
|
* 导入 shp zip 包文件解析空间信息
|
* @param multipartFile
|
* @return
|
*/
|
@Override
|
public R importShpZip(MultipartFile multipartFile) {
|
// 判断文件是否为 zip 文件
|
if (!FileUtil.getFileExtension(multipartFile).equals("zip")){
|
return R.data(400,"文件格式不对，必须是 zip 文件","文件格式不对，必须是 zip 文件");
|
}
|
// 转 file
|
File file = FileUtil.toFile(multipartFile);
|
// 获取文件信息
|
try {
|
List<Map<String, Object>> list = ShapeFileUtil.shpToGeoJson(file);
|
// 写入数据库,暂时考虑只有一组数据的情况
|
if (list.size()>0){
|
Map<String, Object> map = list.get(0);
|
GeomInfoEntity geomInfoEntity = Objects.requireNonNull(BeanUtil.copy(map, GeomInfoEntity.class));
|
String geometry = map.get("geometry").toString();
|
geomInfoEntity.setGeom("'" + geometry +"'");
|
// 保存
|
baseMapper.saveGeoInfo(geomInfoEntity);
|
// 返回
|
return R.data(200,geometry,"操作成功！");
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
// 返回
|
return R.data(200,"操作失败","操作失败！读取数据错误！");
|
}
|
|
/**
|
* 导出 空间信息shp zip 包
|
* @param ids
|
* @return
|
*/
|
@Override
|
public List<GeomInfoVO> getGeomInfoList(String ids) {
|
// 查询空间数据
|
return baseMapper.getGeomInfoList(ids);
|
}
|
|
|
/**
|
* 导入 shp zip 包文件解析空间信息
|
* @param multipartFile
|
* @return
|
*/
|
@Override
|
public R importSpaceShpZip(MultipartFile multipartFile) {
|
// 判断文件是否为 zip 文件
|
if (!FileUtil.getFileExtension(multipartFile).equals("zip")){
|
return R.data(400,"文件格式不对，必须是 zip 文件","文件格式不对，必须是 zip 文件");
|
}
|
// 转 file
|
File file = FileUtil.toFile(multipartFile);
|
// 获取文件信息
|
try {
|
List<Map<String, Object>> list = ShapeFileUtil.shpToGeoJson(file);
|
// 写入数据库,暂时考虑只有一组数据的情况
|
if (list.size()>0){
|
for (Map<String, Object> map : list) {
|
GeomInfoEntity geomInfoEntity = Objects.requireNonNull(BeanUtil.copy(map, GeomInfoEntity.class));
|
String geometry = map.get("geometry").toString();
|
geomInfoEntity.setGeom("'" + geometry +"'");
|
String firmName = null;
|
if (null!=map.get("firmName")){
|
firmName = map.get("firmName").toString();
|
}
|
// 设置所属单位
|
setFirmId(geomInfoEntity,firmName);
|
// 判断是否已经存在
|
Long id = isSave(geomInfoEntity);
|
if (id!=null){
|
// 覆盖更新
|
// updateById();
|
continue;
|
}
|
// 保存
|
baseMapper.saveGeoInfo(geomInfoEntity);
|
}
|
// 返回
|
return R.data(200,list,"操作成功！");
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
}
|
// 返回
|
return R.data(200,"操作失败","操作失败！读取数据错误！");
|
}
|
|
/**
|
* 设置所属企业id
|
* @param geomInfoEntity
|
* @param firmName
|
*/
|
private void setFirmId(GeomInfoEntity geomInfoEntity,String firmName) {
|
if (!Strings.isBlank(firmName)) {
|
QueryWrapper<FirmInfo> wrapper = new QueryWrapper<>();
|
wrapper.like("name", '%' +firmName + '%');
|
FirmInfo one = firmInfoService.getOne(wrapper);
|
if (null!=one)
|
geomInfoEntity.setFirmId(one.getId());
|
}
|
}
|
|
/**
|
* 判断是否已经保存
|
* @param geomInfoEntity
|
* @return
|
*/
|
private Long isSave(GeomInfoEntity geomInfoEntity) {
|
QueryWrapper<GeomInfoEntity> wrapper = new QueryWrapper<>();
|
wrapper.eq("name",geomInfoEntity);
|
GeomInfoEntity one = getOne(wrapper);
|
if (null!=one){
|
return one.getId();
|
}
|
return null;
|
}
|
|
/**
|
* 导入 geojson 文件解析空间信息
|
* @param multipartFile
|
* @return
|
*/
|
@Override
|
public R importGeoJson(MultipartFile multipartFile) {
|
// 转 file
|
File file = FileUtil.toFile(multipartFile);
|
|
FeatureJSON featureJSON = new FeatureJSON();
|
try {
|
FeatureCollection featureCollection = featureJSON.readFeatureCollection(file);
|
FeatureIterator iterator = featureCollection.features();
|
// 遍历feature转为json对象
|
while (iterator.hasNext()) {
|
SimpleFeature feature = (SimpleFeature) iterator.next();
|
StringWriter writer = new StringWriter();
|
featureJSON.writeFeature(feature, writer);
|
String temp = writer.toString();
|
JSONObject json = JSONObject.parseObject(temp);
|
Object value = feature.getDefaultGeometryProperty().getValue();
|
System.out.println("value = " + value);
|
try {
|
GeomInfoEntity geomInfoEntity = new GeomInfoEntity();
|
// 空间坐标
|
geomInfoEntity.setGeom("'" + value + "'");
|
// 空间类型
|
if (json.getJSONObject("geometry").get("type") != null) {
|
geomInfoEntity.setType(json.getJSONObject("geometry").getString("type"));
|
}
|
// 保存
|
baseMapper.saveGeoInfo(geomInfoEntity);
|
} catch (NumberFormatException e) {
|
log.error("NumberFormatException: {}", e);
|
} catch (Exception e) {
|
log.error("Exception: {}", e);
|
}
|
}
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
// 返回
|
return R.data(200,"","操作成功！");
|
}
|
}
|
