submit

dsk-admin/src/main/java/com/dsk/web/controller/business/BusinessFileController.java

@@ -16,6 +16,7 @@ import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
 import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
 import java.util.List;
 import java.util.stream.Collectors;
@@ -69,17 +70,6 @@ public class BusinessFileController extends BaseController {
         return getDataTable(allFiles);
     }
 
-/*    *//**
-     * 上传文件及文件夹
-     * @param url
-     * @param folderPath
-     * @return
-     *//*
-    @GetMapping("/upload/{url}/{folderPath}")
-    public AjaxResult uploadFolder(@PathVariable("url") String url,@PathVariable("folderPath") String folderPath) throws IOException {
-        return toAjax(FileUtils.uploadFolder(url, LOCALPATH + folderPath));
-    }*/
-
     /**
      * 上传文件及文件夹
      * @param file 文件流
@@ -87,7 +77,7 @@ public class BusinessFileController extends BaseController {
      * @return
      */
     @PostMapping("/upload")
-    public AjaxResult uploadFolder(@RequestPart("file") MultipartFile file,HttpServletRequest request){
+    public AjaxResult uploadFolder(@RequestPart("file") MultipartFile file, HttpServletRequest request){
         try {
             String businessFileName = request.getHeader("FilePath");
         // 上传文件路径
@@ -96,7 +86,7 @@ public class BusinessFileController extends BaseController {
             String fileName = FileUploadUtils.upload(filePath, file);
             String url = serverConfig.getUrl() + fileName;
             AjaxResult ajax = AjaxResult.success();
-            ajax.put("url", url);
+            ajax.put("url", fileName);
             return ajax;
         } catch (IOException e) {
             return AjaxResult.error(e.getMessage());
@@ -105,12 +95,11 @@ public class BusinessFileController extends BaseController {
 
     /**
      * 下载文件
-     * @param url
-     * @param targetFolder
-     * @return
+     * @param filePath 要下载的文件路径
+     * @param response 返回的响应
      */
-    @GetMapping("/download/{url}/{targetFolder}")
-    public AjaxResult downloadFolder(@PathVariable("url") String url,@PathVariable("targetFolder") String targetFolder) throws IOException {
-        return toAjax(FileUtils.downloadFolder(url, targetFolder));
+    @PostMapping("/download")
+    public void downloadFolder(@RequestBody BusinessIdDto filePath, HttpServletResponse response) {
+        FileUtils.downloadByFilePath(filePath.getFilePath(),response);
     }
 }

dsk-common/src/main/java/com/dsk/common/utils/file/FileUtils.java

@@ -4,6 +4,7 @@ import com.dsk.common.config.RuoYiConfig;
 import com.dsk.common.core.domain.entity.BusinessFileVo;
 import com.dsk.common.exception.base.BaseException;
 import com.dsk.common.utils.DateUtils;
+import com.dsk.common.utils.ServletUtils;
 import com.dsk.common.utils.StringUtils;
 import com.dsk.common.utils.uuid.IdUtils;
 import lombok.extern.slf4j.Slf4j;
@@ -29,8 +30,6 @@ import java.nio.charset.StandardCharsets;
 import java.text.SimpleDateFormat;
 import java.util.ArrayList;
 import java.util.List;
-import java.util.zip.ZipEntry;
-import java.util.zip.ZipInputStream;
 
 /**
  * 文件处理工具类
@@ -313,46 +312,95 @@ public class FileUtils
     }
 
     /**
-     * 下载文件
-     * @param url 要下载的文件链接
-     * @param targetFolder 目标文件
+     * 根据文件路径下载文件
+     * @param filePath 要下载的文件路径
+     * @param response 返回的响应
+     */
+    public static void downloadByFilePath(String filePath, HttpServletResponse response) {
+        try {
+            // path是指想要下载的文件的路径
+            File file = new File(filePath);
+//            log.info(file.getPath());
+            if (!file.exists()) throw new BaseException("文件不存在!");
+            // 获取文件名
+            String filename = file.getName();
+//            log.info("文件名: "+filename);
+            // 获取文件后缀名
+//            String ext = filename.substring(filename.lastIndexOf(".") + 1).toLowerCase();
+//            log.info("文件后缀名：" + ext);
+
+            // 将文件写入输入流
+            FileInputStream fileInputStream = new FileInputStream(file);
+            InputStream fis = new BufferedInputStream(fileInputStream);
+            byte[] buffer = new byte[fis.available()];
+            fis.read(buffer);
+
+            // 清空response
+            response.reset();
+            // 设置response的Header
+            response.setCharacterEncoding("UTF-8");
+            //文件类型
+            response.setContentType("application/octet-stream");
+            response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(filename, "UTF-8"));
+            //文件名称
+            response.addHeader("Download-filename", URLEncoder.encode(filename, "UTF-8"));
+            // 告知浏览器文件的大小
+            response.addHeader("Content-Length", "" + file.length());
+
+            OutputStream outputStream = new BufferedOutputStream(response.getOutputStream());
+
+            outputStream.write(buffer);
+            fis.close();
+            outputStream.flush();
+        } catch (IOException ex) {
+            ex.printStackTrace();
+        }
+    }
+
+    /**
+     * 根据文件url下载文件
+     * @param filePath 要下载的文件链接
+     * @param response 响应体
      * @return
      * @throws IOException
      */
-    public static boolean downloadFolder(String url, String targetFolder) throws IOException {
-        URL downloadUrl = new URL(url);
-        HttpURLConnection connection = (HttpURLConnection) downloadUrl.openConnection();
-        connection.setRequestMethod("GET");
-        connection.connect();
+    public static void downloadByUrl(String filePath, HttpServletResponse response){
+        //        String fileUrl = request.getParameter("fileUrl"); // 获取文件链接
 
-        if (connection.getResponseCode() != HttpURLConnection.HTTP_OK) {
-            throw new RuntimeException("下载文件夹失败: " + connection.getResponseMessage());
-        }
+        try {
+            URL url = new URL(filePath);
+            HttpURLConnection connection = (HttpURLConnection) url.openConnection();
+            connection.setRequestMethod("GET");
 
-        try (ZipInputStream zipInputStream = new ZipInputStream(connection.getInputStream())) {
-            ZipEntry entry;
-            while ((entry = zipInputStream.getNextEntry()) != null) {
-                String entryName = entry.getName();
-                if (StringUtils.isBlank(entryName)) {
-                    continue;
-                }
+            // 设置响应头，告诉浏览器下载文件
+            String contentType = connection.getContentType(); // 获取文件类型
+            response.setContentType(contentType);
+            String filename = extractFilenameFromUrl(filePath); // 从链接中提取文件名
+            response.setHeader("Content-Disposition", "attachment; filename=" + filename + "");
 
-                File entryFile = new File(targetFolder, entryName);
-                if (entry.isDirectory()) {
-                    entryFile.mkdirs();
-                } else {
-                    File parent = entryFile.getParentFile();
-                    if (!parent.exists()) {
-                        parent.mkdirs();
+            // 将文件流写入响应输出流
+            InputStream inputStream = connection.getInputStream();
+            byte[] buffer = new byte[4096];
+            int bytesRead = -1;
+            while ((bytesRead = inputStream.read(buffer)) != -1) {
+                response.getOutputStream().write(buffer, 0, bytesRead);
             }
+            inputStream.close();
 
-                    try (FileOutputStream outputStream = new FileOutputStream(entryFile)) {
-                        IOUtils.copy(zipInputStream, outputStream);
-                    }
+            connection.disconnect();
+        } catch (Exception e) {
+            e.printStackTrace();
         }
     }
+
+    // 从文件链接中提取文件名
+    public static String extractFilenameFromUrl(String url) {
+        int slashIndex = url.lastIndexOf('/');
+        int dotIndex = url.lastIndexOf('.');
+        if (dotIndex == -1 || dotIndex < slashIndex) {
+            return "download";
         }
-        return true;
+        return url.substring(slashIndex + 1);
     }
 
     /**
@@ -568,4 +616,17 @@ public class FileUtils
         String baseName = FilenameUtils.getBaseName(fileName);
         return baseName;
     }
+
+    /**
+     * 获取本地服务的域名，端口
+     * @return
+     */
+    public static String getUrl()
+    {
+        HttpServletRequest request = ServletUtils.getRequest();
+        StringBuffer url = request.getRequestURL();
+        String contextPath = request.getServletContext().getContextPath();
+        return url.delete(url.length() - request.getRequestURI().length(), url.length()).append(contextPath).toString();
+
+    }
 }

dsk-system/src/main/java/com/dsk/system/domain/BusinessListDto.java

@@ -11,12 +11,6 @@ import java.util.List;
  **/
 @Data
 public class BusinessListDto {
-
-    /**
-     * 拜访方式
-     */
-    private String visitWay;
-
     /**
      * 项目名称
      */

dsk-system/src/main/java/com/dsk/system/domain/vo/BusinessBrowseVo.java

@@ -121,9 +121,14 @@ public class BusinessBrowseVo {
     private Integer backlogCount;
 
     /**
-     * 资料文档统计
+     * 相关企业统计
      */
     private Integer relateCompanyCount;
 
+    /**
+     * 资料文档统计
+     */
+    private Integer fileCount;
+
 
 }

dsk-system/src/main/java/com/dsk/system/mapper/BusinessFollowRecordMapper.java

@@ -3,6 +3,7 @@ package com.dsk.system.mapper;
 import com.dsk.common.core.domain.entity.BusinessFollowRecord;
 import com.dsk.system.domain.BusinessIdDto;
 import com.dsk.system.domain.BusinessListDto;
+import com.dsk.system.domain.vo.BusinessListVo;
 
 import java.util.List;
 
@@ -20,7 +21,7 @@ public interface BusinessFollowRecordMapper
      * @param userId
      * @return
      */
-    List<String> selectRelateProject (Integer userId);
+    List<BusinessListVo> selectRelateProject (Integer userId);
 
     /**
      * 查询关联业主企业

dsk-system/src/main/java/com/dsk/system/service/IBusinessFollowRecordService.java

@@ -3,6 +3,7 @@ package com.dsk.system.service;
 import com.dsk.common.core.domain.entity.BusinessFollowRecord;
 import com.dsk.system.domain.BusinessIdDto;
 import com.dsk.system.domain.BusinessListDto;
+import com.dsk.system.domain.vo.BusinessListVo;
 
 import java.util.List;
 
@@ -59,7 +60,7 @@ public interface IBusinessFollowRecordService
      * @param userId
      * @return
      */
-    List<String> selectRelateProject (Integer userId);
+    List<BusinessListVo> selectRelateProject (Integer userId);
 
     /**
      * 查询关联业主企业

dsk-system/src/main/java/com/dsk/system/service/impl/BusinessFollowRecordServiceImpl.java

 package com.dsk.system.service.impl;
 
 import com.dsk.common.core.domain.entity.BusinessFollowRecord;
+import com.dsk.common.exception.base.BaseException;
 import com.dsk.common.utils.DateUtils;
+import com.dsk.common.utils.SecurityUtils;
 import com.dsk.system.domain.BusinessIdDto;
 import com.dsk.system.domain.BusinessListDto;
+import com.dsk.system.domain.vo.BusinessListVo;
 import com.dsk.system.mapper.BusinessFollowRecordMapper;
 import com.dsk.system.service.IBusinessFollowRecordService;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -45,11 +48,11 @@ public class BusinessFollowRecordServiceImpl implements IBusinessFollowRecordSer
     @Override
     public List<BusinessFollowRecord> allFollow(BusinessListDto dto) {
         //userId不传值，就查询全部门项目
-//        if (dto.getUserId() == null) {
-//            Long deptId = SecurityUtils.getLoginUser().getDeptId();
-//            if (deptId == null) throw new BaseException("请登录");
-//            dto.setDeptId(deptId.intValue());
-//        }
+        if (dto.getUserId() == null) {
+            Long deptId = SecurityUtils.getLoginUser().getDeptId();
+            if (deptId == null) throw new BaseException("请登录");
+            dto.setDeptId(deptId.intValue());
+        }
         return businessFollowRecordMapper.allFollow(dto);
     }
 
@@ -72,7 +75,7 @@ public class BusinessFollowRecordServiceImpl implements IBusinessFollowRecordSer
     }
 
     @Override
-    public List<String> selectRelateProject(Integer userId) {
+    public List<BusinessListVo> selectRelateProject(Integer userId) {
         return businessFollowRecordMapper.selectRelateProject(userId);
     }
 

dsk-system/src/main/java/com/dsk/system/service/impl/BusinessInfoServiceImpl.java

@@ -80,11 +80,11 @@ public class BusinessInfoServiceImpl implements IBusinessInfoService {
     @Override
     public List<BusinessListVo> selectBusinessInfoList(BusinessListDto dto) {
         //userId不传值，就查询全部门项目
-//        if (dto.getUserId() == null) {
-//            Long deptId = SecurityUtils.getLoginUser().getDeptId();
-//            if (deptId == null) throw new BaseException("请登录");
-//            dto.setDeptId(deptId.intValue());
-//        }
+        if (dto.getUserId() == null) {
+            Long deptId = SecurityUtils.getLoginUser().getDeptId();
+            if (deptId == null) throw new BaseException("请登录");
+            dto.setDeptId(deptId.intValue());
+        }
         return businessInfoMapper.selectBusinessInfoList(dto);
     }
 
@@ -105,7 +105,7 @@ public class BusinessInfoServiceImpl implements IBusinessInfoService {
         businessBrowseVo.setFollowRecordCount(total.getFollowRecordCount());
         businessBrowseVo.setRelateCompanyCount(total.getRelateCompanyCount());
         //资料文档统计
-        businessBrowseVo.setRelateCompanyCount(FileUtils.getAllFileNames(RuoYiConfig.getProfile()+businessId).size());
+        businessBrowseVo.setFileCount(FileUtils.getAllFileNames(RuoYiConfig.getProfile()+businessId).size());
         return businessBrowseVo;
     }
 

dsk-system/src/main/resources/mapper/business/BusinessFollowRecordMapper.xml

@@ -75,29 +75,20 @@
             <if test="deptId != null">
                 and u.dept_id = #{deptId}
             </if>
-            <if test="projectName != null and projectName != ''">
-                and i.project_name = #{projectName}
-            </if>
-            <if test="ownerCompany != null and ownerCompany != ''">
-                and i.construction_unit = #{ownerCompany}
-            </if>
-            <if test="visitWay != null and visitWay != ''">
-                and f.visit_way = #{visitWay}
-            </if>
         </where>
         ORDER BY f.creat_time DESC
     </select>
-    <select id="selectRelateProject" resultType="java.lang.String">
-        select i.project_name
+    <select id="selectRelateProject" resultType="com.dsk.system.domain.vo.BusinessListVo">
+        select i.id,i.project_name as projectName
         from business_info i
-        left join business_follow_record f on f.business_id = i.id
-        where f.user_id = #{userId}
+        left join business_user u on u.business_id = i.id
+        where u.user_id = #{userId}
     </select>
     <select id="selectRelateCompany" resultType="java.lang.String">
         select i.construction_unit
         from business_info i
-        left join business_follow_record f on f.business_id = i.id
-        where f.user_id = #{userId}
+        left join business_user u on u.business_id = i.id
+        where u.user_id = #{userId}
     </select>
 
     <insert id="insertBusinessFollowRecord" parameterType="com.dsk.common.core.domain.entity.BusinessFollowRecord" useGeneratedKeys="true"