package com.lf.server.service.all;
|
|
import com.lf.server.entity.all.StaticData;
|
import com.lf.server.entity.data.DirEntity;
|
import com.lf.server.entity.data.MetaFileEntity;
|
import com.lf.server.entity.data.VerEntity;
|
import com.lf.server.entity.sys.UserEntity;
|
import com.lf.server.helper.*;
|
import org.apache.commons.logging.Log;
|
import org.apache.commons.logging.LogFactory;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Service;
|
import org.springframework.web.multipart.MultipartFile;
|
import org.springframework.web.multipart.support.StandardMultipartHttpServletRequest;
|
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.File;
|
import java.lang.reflect.Field;
|
import java.sql.Timestamp;
|
import java.util.*;
|
|
/**
|
* 父上传服务
|
* @author WWW
|
*/
|
@Service
|
public class BaseUploadService {
|
@Autowired
|
public PathHelper pathHelper;
|
|
public final Log log = LogFactory.getLog(getClass());
|
|
/**
|
* 查询目录
|
*/
|
public String selectPath() {
|
return pathHelper.getTempPathName();
|
}
|
|
/**
|
* 上传文件
|
*/
|
public <T> List<MetaFileEntity> uploadData(T t, String path, HttpServletRequest req, HttpServletResponse res) throws Exception {
|
StandardMultipartHttpServletRequest request = (StandardMultipartHttpServletRequest) req;
|
req.setCharacterEncoding("utf-8");
|
res.setContentType("application/json;charset=utf-8");
|
|
if (t != null) {
|
setEntity(t, request);
|
}
|
|
return getFiles(path, request);
|
}
|
|
/**
|
* 设置实体类
|
*/
|
private <T> void setEntity(T t, StandardMultipartHttpServletRequest req) {
|
Enumeration<String> enumeration = req.getParameterNames();
|
while (enumeration.hasMoreElements()) {
|
String key = enumeration.nextElement();
|
|
try {
|
Field field = t.getClass().getDeclaredField(key);
|
field.setAccessible(true);
|
String value = req.getParameter(key);
|
|
switch (field.getType().toString()) {
|
case "double":
|
field.set(t, Double.valueOf(value));
|
break;
|
case "long":
|
field.set(t, Long.valueOf(value));
|
break;
|
case "int":
|
field.set(t, Integer.valueOf(value));
|
break;
|
case "class java.sql.Timestamp":
|
field.set(t, Timestamp.valueOf(value));
|
break;
|
//case "class java.lang.String":
|
default:
|
field.set(t, value);
|
break;
|
}
|
} catch (Exception ex) {
|
log.error(ex.getMessage(), ex);
|
}
|
}
|
}
|
|
/**
|
* 获取文件
|
*/
|
public List<MetaFileEntity> getFiles(String subPath, StandardMultipartHttpServletRequest req) throws Exception {
|
List<MetaFileEntity> list = new ArrayList<>();
|
|
String path = pathHelper.getTempPath(subPath);
|
Iterator<String> iterator = req.getFileNames();
|
while (iterator.hasNext()) {
|
MultipartFile file = req.getFile(iterator.next());
|
if (null == file || StringHelper.isEmpty(file.getOriginalFilename())) {
|
continue;
|
}
|
|
MetaFileEntity mf = new MetaFileEntity();
|
mf.setName(file.getOriginalFilename());
|
|
double sizes = FileHelper.sizeToMb(file.getSize());
|
mf.setSizes(sizes);
|
mf.setPath(path + File.separator + mf.getName());
|
|
file.transferTo(new File(mf.getPath()));
|
mf.setGuid(FileHelper.getFileMd5(mf.getPath()));
|
|
list.add(mf);
|
}
|
|
return list;
|
}
|
|
/**
|
* 查询文件
|
*/
|
public List<MetaFileEntity> selectFiles(String subPath, List<String> extList) {
|
String root = pathHelper.getConfig().getTempPath() + File.separator + subPath;
|
|
File file = new File(root);
|
if (!file.exists() && !file.isDirectory()) {
|
return null;
|
}
|
File[] files = file.listFiles();
|
if (null == files || files.length == 0) {
|
return null;
|
}
|
|
List<MetaFileEntity> list = new ArrayList<>();
|
for (File f : files) {
|
String fileName = FileHelper.getFileName(f.getPath());
|
String extName = FileHelper.getExtension(fileName);
|
if (null != extList && !extList.contains(extName)) {
|
continue;
|
}
|
|
MetaFileEntity mf = new MetaFileEntity();
|
mf.setName(fileName);
|
mf.setExtName(extName);
|
mf.setSizes(FileHelper.sizeToMb(f.length()));
|
mf.setPath(subPath + File.separator + fileName);
|
if (!StaticData.ZIP.equals(extName)) {
|
mf.setGuid(FileHelper.getFileMd5(f.getPath()));
|
}
|
|
list.add(mf);
|
}
|
|
return list;
|
}
|
|
/**
|
* 删除文件
|
*/
|
public Integer deleteFiles(List<MetaFileEntity> list) {
|
String root = pathHelper.getConfig().getTempPath();
|
|
int count = 0;
|
for (MetaFileEntity entity : list) {
|
if (!StringHelper.isEmpty(entity.getPath())) {
|
String file = root + File.separator + entity.getPath();
|
|
File f = new File(file);
|
if (f.exists()) {
|
f.delete();
|
count++;
|
}
|
}
|
}
|
|
return count;
|
}
|
|
/**
|
* 查询映射
|
*/
|
public List<MetaFileEntity> selectMappers(UserEntity ue, String path, DirEntity dir, VerEntity ver, String epsgCode) {
|
List<MetaFileEntity> metas = selectFiles(path, StaticData.ALL_EXTENSION);
|
if (null == metas || metas.isEmpty()) {
|
return null;
|
}
|
|
List<MetaFileEntity> list = new ArrayList<>();
|
for (MetaFileEntity meta : metas) {
|
meta.setEventid(StringHelper.getGuid());
|
meta.setDirid(dir.getId());
|
meta.setDepid(ue.getDepid());
|
meta.setVerid(ver.getId());
|
meta.setCreateUser(ue.getId());
|
meta.setEpsgCode(epsgCode);
|
|
if (StaticData.ZIP.equals(meta.getExtName())) {
|
List<MetaFileEntity> subs = getMapperFiles(path, meta);
|
if (null != subs && subs.size() > 0) {
|
list.addAll(subs);
|
continue;
|
}
|
}
|
|
meta.setEventid(StringHelper.getGuid());
|
list.add(meta);
|
}
|
|
return list;
|
}
|
|
/**
|
* 获取映射文件
|
*/
|
private List<MetaFileEntity> getMapperFiles(String path, MetaFileEntity meta) {
|
String zipFile = pathHelper.getConfig().getTempPath() + File.separator + meta.getPath();
|
File file = new File(zipFile);
|
if (!file.exists() || file.isDirectory()) {
|
return null;
|
}
|
|
File zipFolder = new File(pathHelper.getConfig().getTempPath() + File.separator + path + "_zip");
|
if (!zipFolder.exists() || !zipFolder.isDirectory()) {
|
zipFolder.mkdirs();
|
}
|
|
String subPath = zipFolder + File.separator + meta.getName().toLowerCase().replace(".zip", "");
|
ZipHelper.unzip(zipFile, subPath);
|
|
File subFile = new File(subPath);
|
File[] files = subFile.listFiles();
|
if (null == files || files.length == 0) {
|
return null;
|
}
|
|
return getMapperFiles(files, meta, subPath);
|
}
|
|
/**
|
* 获取映射文件
|
*/
|
private List<MetaFileEntity> getMapperFiles(File[] files, MetaFileEntity meta,String subPath) {
|
List<MetaFileEntity> list = new ArrayList<>();
|
for (File f : files) {
|
boolean isGdb = isGdbFile(f);
|
if (!isGdb && f.isDirectory()) {
|
continue;
|
}
|
|
String fileName = FileHelper.getFileName(f.getPath());
|
String extName = FileHelper.getExtension(fileName);
|
if (!StaticData.ALL_EXTENSION.contains(extName)) {
|
continue;
|
}
|
|
boolean isShp = StaticData.SHP.equals(extName);
|
if (isGdb) {
|
List<MetaFileEntity> rs = getGdbMappers(f, meta);
|
if (null != rs && rs.size() > 0) {
|
list.addAll(rs);
|
}
|
continue;
|
}
|
|
MetaFileEntity mf = createMetaFileEntity(meta);
|
mf.setName(fileName);
|
mf.setExtName(extName);
|
mf.setSizes(FileHelper.sizeToMb(f.length()));
|
mf.setPath(f.getPath().substring(subPath.length()));
|
if (isShp) {
|
mf.setTab(fileName);
|
mf.setGuid(getShpMd5(f.getPath()));
|
} else {
|
mf.setGuid(FileHelper.getFileMd5(f.getPath()));
|
}
|
list.add(mf);
|
}
|
|
return list;
|
}
|
|
/**
|
* 创建元数据文件实体类
|
*/
|
private MetaFileEntity createMetaFileEntity(MetaFileEntity meta) {
|
MetaFileEntity mf = new MetaFileEntity();
|
mf.setEventid(StringHelper.getGuid());
|
mf.setDirid(meta.getDirid());
|
mf.setDepid(meta.getDepid());
|
mf.setVerid(meta.getVerid());
|
mf.setCreateUser(meta.getCreateUser());
|
mf.setEpsgCode(meta.getEpsgCode());
|
|
return mf;
|
}
|
|
/**
|
* 获取SHP文件的MD5码
|
*/
|
private String getShpMd5(String shpPath) {
|
List<String> files = new ArrayList<>();
|
files.add(shpPath);
|
|
for (int i = 1, c = StaticData.SHP_EXTENSION.size(); i < c; i++) {
|
String path = shpPath.replace(".shp", StaticData.SHP_EXTENSION.get(i));
|
|
File f = new File(path);
|
if (f.exists() && !f.isDirectory()) {
|
files.add(path);
|
}
|
}
|
|
return getFilesMd5(files);
|
}
|
|
/**
|
* 获取GDB文件的MD5码
|
*/
|
private String getGdbMd5(String path) {
|
File[] files = new File(path).listFiles();
|
if (null == files || files.length == 0) {
|
return StringHelper.getGuid();
|
}
|
|
List<String> list = new ArrayList<>();
|
for (File f : files) {
|
list.add(f.getPath());
|
}
|
|
return getFilesMd5(list);
|
}
|
|
/**
|
* 获取多文件的MD5
|
*/
|
private String getFilesMd5(List<String> strs) {
|
List<String> list = new ArrayList<>();
|
for (String str : strs) {
|
String md5 = FileHelper.getFileMd5(str);
|
if (null != md5) {
|
list.add(md5);
|
}
|
}
|
|
if (list.size() > 0) {
|
String str = StringHelper.join(list, ",");
|
return FileHelper.getStringMd5(str);
|
}
|
|
return StringHelper.getGuid();
|
}
|
|
/**
|
* 获取GDB文件映射
|
*/
|
private List<MetaFileEntity> getGdbMappers(File f,MetaFileEntity meta) {
|
List<String> tabs = GdbHelper.getTabNames(f.getPath());
|
if (null == tabs || tabs.size() == 0) {
|
return null;
|
}
|
|
String md5 = getGdbMd5(f.getPath());
|
for (String tab : tabs) {
|
//
|
}
|
|
return null;
|
}
|
|
/**
|
* 是/否为GDB文件
|
*/
|
private boolean isGdbFile(File f) {
|
if (f.isDirectory() && f.getName().toLowerCase().endsWith(StaticData.GDB)) {
|
File[] files = f.listFiles();
|
if (null == files || files.length == 0) {
|
return false;
|
}
|
|
for (File file : files) {
|
if ("gdb".equals(file.getName())) {
|
return true;
|
}
|
}
|
}
|
|
return false;
|
}
|
|
/**
|
* 获取参数 *
|
* Enumeration<String> headers = req.getHeaderNames();
|
* Enumeration<String> attributes = req.getAttributeNames();
|
*/
|
public Map<String, String> getParams(StandardMultipartHttpServletRequest req) {
|
Map<String, String> map = new HashMap<String, String>(3);
|
|
Enumeration<String> enumeration = req.getParameterNames();
|
while (enumeration.hasMoreElements()) {
|
String key = enumeration.nextElement();
|
String value = req.getParameter(key);
|
map.put(key, value);
|
}
|
|
return map;
|
}
|
}
|
