|
@@ -3,7 +3,6 @@ package com.fd.server.impl;
|
|
|
import com.alibaba.fastjson.JSON;
|
|
|
import com.alibaba.fastjson.JSONArray;
|
|
|
import com.alibaba.fastjson.JSONObject;
|
|
|
-import com.fd.constant.Command;
|
|
|
import com.fd.constant.MsgCode;
|
|
|
import com.fd.constant.TypeCode;
|
|
|
import com.fd.dto.ConfigJsonDto;
|
|
@@ -12,12 +11,15 @@ import com.fd.entity.FileEntity;
|
|
|
//import com.fd.entity.LayerEntity;
|
|
|
import com.fd.entity.OutputFileEntity;
|
|
|
import com.fd.entity.StyleEntity;
|
|
|
+import com.fd.entity.User;
|
|
|
import com.fd.repository.FileRepository;
|
|
|
//import com.fd.repository.LayerRepository;
|
|
|
import com.fd.repository.OutputFileRepository;
|
|
|
import com.fd.repository.StyleRepository;
|
|
|
+import com.fd.repository.UserRepository;
|
|
|
import com.fd.server.CmdServer;
|
|
|
import com.fd.server.RasterServer;
|
|
|
+import com.fd.shiro.JWTUtil;
|
|
|
import com.fd.util.FileUtils;
|
|
|
import com.fd.util.R;
|
|
|
import lombok.extern.log4j.Log4j2;
|
|
@@ -31,7 +33,6 @@ import org.springframework.stereotype.Service;
|
|
|
import org.springframework.web.multipart.MultipartFile;
|
|
|
|
|
|
import java.io.BufferedReader;
|
|
|
-import java.io.File;
|
|
|
import java.io.IOException;
|
|
|
import java.io.InputStreamReader;
|
|
|
import java.util.Date;
|
|
@@ -44,7 +45,7 @@ import java.util.Optional;
|
|
|
*/
|
|
|
@Log4j2
|
|
|
@Service
|
|
|
-public class RasterServerImpl implements RasterServer {
|
|
|
+public class RasterServerImpl extends BaseServerImpl implements RasterServer {
|
|
|
|
|
|
@Value("${input.file.path.raster}")
|
|
|
private String INPUT_FILE_PATH;
|
|
@@ -79,6 +80,9 @@ public class RasterServerImpl implements RasterServer {
|
|
|
@Autowired
|
|
|
private StyleRepository styleRepository;
|
|
|
|
|
|
+ @Autowired
|
|
|
+ private UserRepository userRepository;
|
|
|
+
|
|
|
@Override
|
|
|
public R deleteById(Long fileId) {
|
|
|
// 删除服务器文件
|
|
@@ -208,7 +212,7 @@ public class RasterServerImpl implements RasterServer {
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
- public R uploadBigFile(MultipartFile file, String strCoord) {
|
|
|
+ public R uploadBigFile(MultipartFile file, String strCoord, String token) {
|
|
|
long start = System.currentTimeMillis();
|
|
|
if (file.isEmpty() || file.getSize() <= 0) {
|
|
|
log.info("文件为空");
|
|
@@ -238,6 +242,11 @@ public class RasterServerImpl implements RasterServer {
|
|
|
}
|
|
|
log.info("filePath: {}", filePath);
|
|
|
|
|
|
+ String username = JWTUtil.getUsername(token);
|
|
|
+
|
|
|
+ // 根据用户名查找用户
|
|
|
+ User user = userRepository.findByUsername(username);
|
|
|
+
|
|
|
// 保存信息到db
|
|
|
FileEntity entity = new FileEntity();
|
|
|
entity.setFileName(fileName);
|
|
@@ -261,6 +270,10 @@ public class RasterServerImpl implements RasterServer {
|
|
|
outputFile.setCreateTime(new Date());
|
|
|
outputFile.setUpdateTime(new Date());
|
|
|
|
|
|
+ // 添加分组
|
|
|
+ outputFile.setUserId(user.getId());
|
|
|
+ outputFile.setUserGroup(user.getUserGroup());
|
|
|
+
|
|
|
outputFile = outputFileRepository.save(outputFile);
|
|
|
long end = System.currentTimeMillis();
|
|
|
log.info("end uploadBigFile, total time: {} s", (end - start)/1000);
|