package com.lf.server.service.data;
|
|
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
import com.lf.server.entity.all.BaseEntity;
|
import com.lf.server.entity.ctrl.TabMapperEntity;
|
import com.lf.server.entity.data.MetaEntity;
|
import com.lf.server.entity.data.MetaFileEntity;
|
import com.lf.server.helper.ClassHelper;
|
import com.lf.server.helper.ExcelHelper;
|
import com.lf.server.helper.FileHelper;
|
import com.lf.server.mapper.all.GeomBaseMapper;
|
import com.lf.server.service.all.BaseQueryService;
|
import com.lf.server.service.all.BaseUploadService;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
|
import java.io.File;
|
import java.lang.reflect.Field;
|
import java.util.List;
|
|
/**
|
* 数据入库
|
* @author WWW
|
*/
|
@Service
|
public class DataLoaderService extends BaseUploadService {
|
@Autowired
|
MetaService metaService;
|
|
@Autowired
|
MetaFileService metaFileService;
|
|
private final static String XLS = ".xls";
|
|
private final static String MDB = ".mdb";
|
|
private final static String SHP = ".shp";
|
|
private final static String GDB = ".gdb";
|
|
/**
|
* 插入文件
|
*/
|
public int insertFiles(MetaEntity entity, List<MetaFileEntity> list, List<TabMapperEntity> tabs) {
|
int count = 0;
|
try {
|
String temp = pathHelper.getConfig().getTempPath();
|
String root = pathHelper.getUploadFullPath();
|
|
for (MetaFileEntity mf : list) {
|
String filePath = getFilePath(temp, root, mf);
|
if (null == filePath) {
|
continue;
|
}
|
|
// 元数据
|
MetaEntity me = createMetaEntity(entity, mf);
|
if (metaService.insert(me) < 1) {
|
continue;
|
}
|
|
// 元数据文件
|
MetaFileEntity mfe = createMetaFileEntity(me, mf, filePath);
|
if (metaFileService.insert(mfe) < 1) {
|
continue;
|
}
|
|
// 入库
|
if (null != me.getType() && insertDb(me, mfe, tabs) > 0) {
|
count++;
|
}
|
}
|
} catch (Exception ex) {
|
log.error(ex.getMessage(), ex);
|
}
|
|
return count;
|
}
|
|
/**
|
* 获取文件路径
|
*/
|
private String getFilePath(String temp, String root, MetaFileEntity mf) {
|
// 移动文件
|
File file = new File(temp + File.separator + mf.getPath());
|
File newFile = new File(root + File.separator + mf.getGuid());
|
|
// 文件类型
|
String metaType = getType(mf.getName().toLowerCase());
|
if (null == metaType) {
|
file.delete();
|
return null;
|
}
|
|
MetaFileEntity old = metaFileService.selectByGuid(mf.getGuid());
|
|
String filePath = null;
|
if (null == old) {
|
filePath = newFile.getPath();
|
file.renameTo(newFile);
|
} else {
|
filePath = old.getPath();
|
file.delete();
|
}
|
|
return filePath;
|
}
|
|
/**
|
* 创建元数据实体
|
*/
|
private MetaEntity createMetaEntity(MetaEntity entity, MetaFileEntity mf) {
|
MetaEntity me = new MetaEntity();
|
me.setDepid(entity.getDepid());
|
me.setDirid(entity.getDirid());
|
me.setVerid(entity.getVerid());
|
me.setType(getType(mf.getName().toLowerCase()));
|
me.setGather(entity.getGather());
|
me.setBatch(entity.getBatch());
|
me.setDescr(entity.getDescr());
|
me.setName(mf.getName());
|
me.setSizes(mf.getSizes());
|
me.setCreateTime(entity.getCreateTime());
|
me.setCreateUser(entity.getCreateUser());
|
|
return me;
|
}
|
|
/**
|
* 创建元数据文件实体
|
*/
|
private MetaFileEntity createMetaFileEntity(MetaEntity me, MetaFileEntity entity, String filePath) {
|
MetaFileEntity mfe = new MetaFileEntity();
|
mfe.setName(entity.getName());
|
mfe.setMetaid(me.getId());
|
mfe.setGuid(entity.getGuid());
|
mfe.setSizes(entity.getSizes());
|
mfe.setPath(FileHelper.getRelativePath(filePath));
|
mfe.setCreateUser(me.getCreateUser());
|
mfe.setCreateTime(me.getCreateTime());
|
|
return mfe;
|
}
|
|
/**
|
* 获取文件类型
|
*/
|
private String getType(String name) {
|
if (name.contains(XLS)) {
|
return "xls";
|
}
|
if (name.contains(MDB)) {
|
return "mdb";
|
}
|
if (name.contains(SHP)) {
|
return "shp";
|
}
|
if (name.contains(GDB)) {
|
return "gdb";
|
}
|
|
return null;
|
}
|
|
/**
|
* 插入数据库
|
*/
|
private Integer insertDb(MetaEntity me, MetaFileEntity mfe, List<TabMapperEntity> tabs) {
|
String root = pathHelper.getConfig().getUploadPath();
|
switch (me.getType()) {
|
case "xls":
|
return insertXls(root, mfe, tabs);
|
case "mdb":
|
return insertMdb(root, mfe, tabs);
|
case "shp":
|
return insertShp(root, mfe, tabs);
|
case "gdb":
|
return insertGdb(root, mfe, tabs);
|
default:
|
return 0;
|
}
|
}
|
|
/**
|
* 插入Excel
|
*/
|
private Integer insertXls(String root, MetaFileEntity mfe, List<TabMapperEntity> tabs) {
|
TabMapperEntity tab = getTabEntity(mfe, tabs);
|
if (null == tab) {
|
return 0;
|
}
|
|
BaseMapper baseMapper = ClassHelper.getBasicMapper(tab.getEntity());
|
if (null == baseMapper) {
|
return 0;
|
}
|
|
String className = ClassHelper.getClassName(baseMapper);
|
Class clazz = ClassHelper.getEntityClass(className);
|
if (null == clazz) {
|
return 0;
|
}
|
|
List<?> list = ExcelHelper.readExcel(clazz, root + File.separator + mfe.getPath());
|
if (null == list || list.isEmpty()) {
|
return 0;
|
}
|
|
setCreateInfo(list, mfe);
|
|
int rows = 0;
|
for (int i = 0, c = list.size(); i < c; i++) {
|
rows += baseMapper.insert(list.get(i));
|
}
|
|
if (baseMapper instanceof GeomBaseMapper) {
|
GeomBaseMapper geomBaseMapper = (GeomBaseMapper) baseMapper;
|
updateGeom(geomBaseMapper, list);
|
}
|
|
return rows;
|
}
|
|
/**
|
* 设置创建信息
|
*/
|
@SuppressWarnings("AlibabaRemoveCommentedCode")
|
private <T> void setCreateInfo(List<T> list, MetaFileEntity mfe) {
|
try {
|
for (T t : list) {
|
Field cuField = t.getClass().getDeclaredField("createuser");
|
cuField.setAccessible(true);
|
cuField.set(t, mfe.getCreateUser());
|
|
Field ctField = t.getClass().getDeclaredField("createtime");
|
ctField.setAccessible(true);
|
ctField.set(t, mfe.getCreateTime());
|
}
|
} catch (Exception ex) {
|
log.error(ex.getMessage(), ex);
|
}
|
}
|
|
/**
|
* 设置空间信息
|
*/
|
private <T> void updateGeom(GeomBaseMapper geomBaseMapper, List<T> list) {
|
try {
|
String tabName = BaseQueryService.getTabName(geomBaseMapper);
|
for (T t : list) {
|
Field xField = t.getClass().getDeclaredField("x");
|
xField.setAccessible(true);
|
double x = (double) xField.get(t);
|
|
Field yField = t.getClass().getDeclaredField("y");
|
yField.setAccessible(true);
|
double y = (double) yField.get(t);
|
|
BaseEntity baseEntity = (BaseEntity) t;
|
Integer gid = baseEntity.getGid();
|
|
String wkt = String.format("POINT(%f %f)", x, y);
|
geomBaseMapper.updateGeom(tabName, gid, wkt);
|
}
|
} catch (Exception ex) {
|
log.error(ex.getMessage(), ex);
|
}
|
}
|
|
/**
|
* 插入MDB
|
*/
|
private Integer insertMdb(String root, MetaFileEntity mfe, List<TabMapperEntity> tabs) {
|
|
return 0;
|
}
|
|
/**
|
* 插入SHP
|
*/
|
private Integer insertShp(String root, MetaFileEntity mfe, List<TabMapperEntity> tabs) {
|
|
return 0;
|
}
|
|
/**
|
* 插入GDB
|
*/
|
private Integer insertGdb(String root, MetaFileEntity mfe, List<TabMapperEntity> tabs) {
|
|
return 0;
|
}
|
|
/**
|
* 获取表映射实体类
|
*/
|
private TabMapperEntity getTabEntity(MetaFileEntity mfe, List<TabMapperEntity> tabs) {
|
for (TabMapperEntity tab : tabs) {
|
if (tab.getFileName().equals(mfe.getName())) {
|
return tab;
|
}
|
}
|
|
return null;
|
}
|
}
|
