• 基于Ruoyi和WebUploader的统一附件管理扩展（下）

    目录

    背景

    一、数据库定义

    1、目的

    2、数据库物理表设计

    二、JAVA后台服务定义

    1、实体类定义

    2、数据服务和业务层处理

    3、控制层定义

     三、总结

    背景

           在之前的博文中，简单介绍了如何扩展Ruoyi的大附件上传及统一管理一篇，原文地址：基于Ruoyi和WebUploader的统一附件管理扩展（上）。之前的博文主要集中在前台的讲解，前台主要是围绕WebUploader组件来说明，对应的后台处理没有仔细讲解。本文作为下篇，主要围绕大附件上传的后台设计及实现，通过相关的UML建模，具体阐述后台是如何进行对应的处理。包括断点续传，文件重复判断等特殊功能的实例讲解。希望对你在项目中的使用有所启发。

    一、数据库定义

    1、目的

           定义数据实体的目的是为了可以在各业务中通用，提供统一的组件及封装，同时为了可以在文件处理时满足实时对文件是否重复上传做校验，因此有必要采用数据库的方式进行存储。当然，这里的数据存储未必要使用关系型数据库，使用非关系型数据库，比如MongoDB也是可以的。只需要达到我们的目的即可。

    2、数据库物理表设计

           设计统一附件管理模块，不仅要统一文件上传的操作界面，同时要提供统一的API可以对附件进行检索，因而要将一个业务表的表名传递到附件表中。同时为了，防止一张业务表在不同状态下也可以拥有不同的附件，额外增加一个业务类型的字段。比如一个流程审核的业务，在流程新建阶段关联的附件和审批中关联的附件可以是不一样的。由此字段可以区分开，为了在文件上传过程中加速，也避免减少同一份文件反复上传到磁盘中，造成不必要的资源浪费，因此非常有必要对资源进行判重。关于文件判重，可以参考之前的博文：java文件上传判重姿势浅谈，这里有比较详细的说明。根据上述的需求，可以得到一个数据库物理表的表结构：

            这是一份基于postgresql数据库的表结构，同理在mysql或者其它的数据库下，是一样的，只是数据类型需要稍微修改一下而已。这里将表结构对应的sql文件分享一下：

    1. -- ----------------------------
    2. -- Table structure for biz_file
    3. -- ----------------------------
    4. DROP TABLE IF EXISTS "biz_file";
    5. CREATE TABLE "biz_file" (
    6.   "id" int8 NOT NULL,
    7.   "f_id" varchar(100)  ,
    8.   "b_id" varchar(100)  ,
    9.   "f_type" varchar(30)  NOT NULL,
    10.   "f_name" varchar(512)  NOT NULL,
    11.   "f_desc" varchar(512) ,
    12.   "f_state" int2,
    13.   "f_size" int8 NOT NULL,
    14.   "f_path" varchar(1024)  NOT NULL,
    15.   "table_name" varchar(255) ,
    16.   "md5code" varchar(255) ,
    17.   "directory" varchar(1024) ,
    18.   "create_by" varchar(64) ,
    19.   "create_time" timestamp(6),
    20.   "update_by" varchar(64) ,
    21.   "update_time" timestamp(6),
    22.   "biz_type" varchar(30) 
    23. )
    24. ;
    25. COMMENT ON COLUMN "biz_file"."id" IS '主键';
    26. COMMENT ON COLUMN "biz_file"."f_id" IS '文件id';
    27. COMMENT ON COLUMN "biz_file"."b_id" IS '业务id';
    28. COMMENT ON COLUMN "biz_file"."f_type" IS '文件类型';
    29. COMMENT ON COLUMN "biz_file"."f_name" IS '名称';
    30. COMMENT ON COLUMN "biz_file"."f_desc" IS '文件描述';
    31. COMMENT ON COLUMN "biz_file"."f_state" IS '文件状态';
    32. COMMENT ON COLUMN "biz_file"."f_size" IS '文件大小';
    33. COMMENT ON COLUMN "biz_file"."f_path" IS '文件路径';
    34. COMMENT ON COLUMN "biz_file"."table_name" IS '业务表名';
    35. COMMENT ON COLUMN "biz_file"."md5code" IS 'md5code';
    36. COMMENT ON COLUMN "biz_file"."directory" IS '文件目录';
    37. COMMENT ON COLUMN "biz_file"."create_by" IS '创建人';
    38. COMMENT ON COLUMN "biz_file"."create_time" IS '创建时间';
    39. COMMENT ON COLUMN "biz_file"."update_by" IS '更新人';
    40. COMMENT ON COLUMN "biz_file"."update_time" IS '更新时间';
    41. COMMENT ON COLUMN "biz_file"."biz_type" IS '业务类型';
    42. COMMENT ON TABLE "biz_file" IS '系统附件信息表,用于保存文件上传信息';
    43.  
    44. -- ----------------------------
    45. -- Primary Key structure for table biz_file
    46. -- ----------------------------
    47. ALTER TABLE "biz_file" ADD CONSTRAINT "pk_biz_file" PRIMARY KEY ("id");

    二、JAVA后台服务定义

    1、实体类定义

           实体类主要用于定义跟数据库表相互映射的对象信息，使用sql语句来操作数据库信息。示例中的代码均需要配置lombok来简化相关类的开发工作量。

    1. package com.hngtghy.project.webupload.domain;
    2.  
    3. import com.baomidou.mybatisplus.annotation.TableField;
    4. import com.baomidou.mybatisplus.annotation.TableName;
    5. import com.hngtghy.framework.web.domain.BaseEntity;
    6.  
    7. import lombok.AllArgsConstructor;
    8. import lombok.Getter;
    9. import lombok.NoArgsConstructor;
    10. import lombok.Setter;
    11. import lombok.ToString;
    12.  
    13. @TableName("biz_file")
    14. @NoArgsConstructor
    15. @AllArgsConstructor
    16. @Setter
    17. @Getter
    18. @ToString
    19. public class FileEntity extends BaseEntity {
    20.  
    21. 	private static final long serialVersionUID = 1L;
    22.  
    23. 	private Long id;
    24.  
    25. 	@TableField(value = "f_id")
    26. 	private String fid;
    27. 	
    28. 	@TableField(value = "b_id")
    29. 	private String bid;
    30. 	
    31. 	@TableField(value = "f_type")
    32. 	private String type;
    33. 	
    34. 	@TableField(value = "f_name")
    35. 	private String name;
    36. 	
    37. 	@TableField(value = "f_desc")
    38. 	private String desc;
    39. 	
    40. 	@TableField(value = "f_state")
    41. 	private Integer state;
    42. 	
    43. 	@TableField(value = "f_size")
    44. 	private Long size;
    45. 	
    46. 	@TableField(value = "f_path")
    47. 	private String path;
    48. 	
    49. 	@TableField(value = "table_name")
    50. 	private String tablename = "temp_table";
    51.  
    52. 	private String md5code;
    53. 	
    54. 	private String directory;
    55. 	
    56. 	@TableField(value = "biz_type")
    57. 	private String bizType;
    58.  
    59. }

    2、数据服务和业务层处理

            数据服务，主要基于mybatis-plus来进行增删改查操作，而业务层包含了基础的业务封装，除了调用数据服务外，还有一些额外的业务处理操作。

    1. package com.hngtghy.project.webupload.mapper;
    2.  
    3. import com.baomidou.mybatisplus.core.mapper.BaseMapper;
    4. import com.hngtghy.project.webupload.domain.FileEntity;
    5.  
    6. public interface FileMapper extends BaseMapper {
    7.  
    8. }

             在服务层中定义了对附件对象的删除、查询、修改、保存的功能方法，以接口的形式进行定义。

    1. package com.hngtghy.project.webupload.service;
    2.  
    3. import java.util.List;
    4.  
    5. import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
    6. import com.baomidou.mybatisplus.extension.service.IService;
    7. import com.hngtghy.project.system.user.domain.User;
    8. import com.hngtghy.project.webupload.domain.FileEntity;
    9.  
    10. public interface IFileService extends IService{
    11.  
    12. 	int saveEntity(FileEntity entity) throws Exception;
    13. 	
    14. 	int updateEntity(FileEntity entity) throws Exception;
    15. 	
    16. 	int removeByIds(Long [] ids) throws Exception;
    17. 	
    18. 	int removeByBids(List bids) throws Exception;
    19. 	
    20. 	FileEntity getOneByFid(String fid) throws Exception;
    21. 	
    22. 	int removeByFid(String fid) throws Exception;
    23. 	
    24. 	void deleteErrorFile(User user);
    25. 	
    26.     Long findFileSizeByWrapper(QueryWrapper queryWrapper);
    27.     
    28.     List findListByQueryWrapper(QueryWrapper queryWrapper);
    29.     
    30.     FileEntity findByMd5Code(String md5Code) throws Exception;
    31. }

            其具体的实现类如下，这里主要针对在接口定义的方法进行重写，以满足新的业务需求，比较重要的方法是findByMd5Code，通过这个方法到数据库中查询是否有重复的文件，当有重复文件后，将不再重复上传，关键代码如下：

    1. package com.hngtghy.project.webupload.service;
    2.  
    3. import java.io.File;
    4. import java.io.FileFilter;
    5. import java.sql.PreparedStatement;
    6. import java.util.Arrays;
    7. import java.util.Date;
    8. import java.util.List;
    9.  
    10. import org.springframework.stereotype.Service;
    11. import org.springframework.transaction.annotation.Propagation;
    12. import org.springframework.transaction.annotation.Transactional;
    13.  
    14. import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
    15. import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
    16. import com.hngtghy.common.utils.StringUtils;
    17. import com.hngtghy.common.utils.security.ShiroUtils;
    18. import com.hngtghy.framework.aspectj.lang.annotation.DataSource;
    19. import com.hngtghy.framework.config.HngtghyConfig;
    20. import com.hngtghy.project.system.user.domain.User;
    21. import com.hngtghy.project.webupload.domain.FileEntity;
    22. import com.hngtghy.project.webupload.mapper.FileMapper;
    23.  
    24. @Service
    25. public class FileServiceImpl extends ServiceImpl implements IFileService {
    26. 	
    27. 	
    28. 	@Override
    29. 	@Transactional(propagation = Propagation.REQUIRED, rollbackFor = Exception.class)
    30. 	public int saveEntity(FileEntity entity) throws Exception {
    31. 		entity.setCreateBy(ShiroUtils.getLoginName());
    32. 		entity.setCreateTime(new Date());
    33. 		int result = this.save(entity) ? 1 : 0;
    34. 		return result;
    35. 	}
    36.  
    37. 	@Override
    38. 	@Transactional(propagation = Propagation.REQUIRED, rollbackFor = Exception.class)
    39. 	public int updateEntity(FileEntity entity) throws Exception {
    40. 		int result = this.updateById(entity) ? 1 : 0;
    41. 		return result;
    42. 	}
    43.  
    44. 	@Override
    45. 	@Transactional(propagation = Propagation.REQUIRED, rollbackFor = Exception.class)
    46. 	public int removeByIds(Long[] ids) throws Exception {
    47. 		List removeList = Arrays.asList(ids);
    48. 		for (Long id : ids) {
    49. 			deleteFileOnDiskById(id);
    50. 		}
    51. 		int result = this.removeByIds(removeList) ? 1 : 0;
    52. 		return result;
    53. 	}
    54.  
    55. 	@Override
    56. 	@Transactional(propagation = Propagation.REQUIRED, rollbackFor = Exception.class)
    57. 	public FileEntity getOneByFid(String fid) throws Exception {
    58. 		QueryWrapper queryWrapper = new QueryWrapper();
    59. 		queryWrapper.eq("f_id", fid);
    60. 		FileEntity file = this.getOne(queryWrapper);
    61. 		return file;
    62. 	}
    63.  
    64. 	@Override
    65. 	public int removeByFid(String fid) throws Exception {
    66. 		QueryWrapper queryWrapper = new QueryWrapper();
    67. 		queryWrapper.eq("f_id", fid);
    68. 		deleteFileOnDiskByFid(fid);
    69. 		int result = this.remove(queryWrapper) ? 1 : 0;
    70. 		return result;
    71. 	}
    72.  
    73. 	private void deleteFileOnDiskByFid(String fid) throws Exception {
    74. 		FileEntity file = this.getOneByFid(fid);
    75. 		String file_path = file.getPath();
    76. 		if(!sharedFile(file_path)){
    77. 			this.deleteFileOnDisk(file_path);
    78. 		}
    79. 	}
    80.  
    81. 	private void deleteFileOnDiskById(Long id) throws Exception {
    82. 		FileEntity file = this.getById(id);
    83. 		String file_path = file.getPath();
    84. 		if(!sharedFile(file_path)){
    85. 			this.deleteFileOnDisk(file.getPath());
    86. 		}
    87. 	}
    88. 	
    89. 	private boolean sharedFile(String path){
    90. 		QueryWrapper queryWrapper = new QueryWrapper();
    91. 		queryWrapper.eq("f_path", path);
    92. 		queryWrapper.eq("f_state", 1);
    93. 		List files = this.list(queryWrapper);
    94. 		if (files != null && files.size() > 1) {
    95. 			return  true;
    96. 		}
    97. 		return false;
    98. 	}
    99.  
    100. 	private void deleteFileOnDisk(String path) throws Exception {
    101. 		File file = new File(HngtghyConfig.getProfile() + "/" + path);
    102. 		file.deleteOnExit();
    103. 		if (file.isFile() && file.exists()) {
    104. 			file.delete();
    105. 		}
    106. 	}
    107. 	
    108. 	/**
    109. 	 * 删除失败文件、未绑定文件
    110. 	 * @author 
    111. 	 * @date 
    112. 	 */
    113. 	public void deleteErrorFile(User user){
    114. 		try{
    115. 			String path = HngtghyConfig.getProfile() + "/";
    116. 			QueryWrapper queryWrapper = new QueryWrapper();
    117. 			queryWrapper.eq("f_state", 0);
    118. 			queryWrapper.lt("create_time", "now() - INTERVAL '3 days'");//三天前的失败文件
    119. 			List files = this.list(queryWrapper);
    120. 			if (files != null && files.size() > 0) {
    121. 				this.remove(queryWrapper);
    122. 				for(FileEntity file:files){
    123. 					deleteFileOnDiskById(file.getId()); 
    124. 				}
    125. 			}
    126. 			queryWrapper = new QueryWrapper();
    127. 			queryWrapper.eq("linked", 0);
    128. 			queryWrapper.lt("create_time", "now() - INTERVAL '3 days'");
    129. 			files = this.list(queryWrapper);
    130. 			if (files != null && files.size() > 0) {
    131. 				this.remove(queryWrapper);
    132. 				for(FileEntity file:files){
    133. 					deleteFileOnDiskById(file.getId()); 
    134. 				}
    135. 			}
    136. 			
    137. 			//三天前的分片临时目录
    138. 			String save_path = path + (user != null ? user.getUserId() : "unkown");
    139. 			File directory = new File(save_path);
    140. 			File[] fileArray = directory.listFiles(new FileFilter() {
    141. 				@Override
    142. 				public boolean accept(File pathname) {
    143. 					long last = pathname.lastModified();
    144. 					long diff = System.currentTimeMillis() - last;
    145. 					boolean del = (diff - 24 * 60 * 60 * 1000) > 0;
    146. 					if (pathname.isDirectory() && del) {
    147. 						return true;
    148. 					}
    149. 					return false;
    150. 				}
    151. 			});
    152. 			for(File dir : fileArray){
    153. 				dir.delete();
    154. 			}
    155. 		}catch(Exception e){
    156. 			//无需处理该异常
    157. 		}
    158. 	}
    159.  
    160. 	@Override
    161. 	public Long findFileSizeByWrapper(QueryWrapper queryWrapper) {
    162. 		Long result = this.count(queryWrapper);
    163. 		return result;
    164. 	}
    165.  
    166. 	@Override
    167. 	public List findListByQueryWrapper(QueryWrapper queryWrapper) {
    168. 		List result = this.list(queryWrapper);
    169. 		return result;
    170. 	}
    171.  
    172. 	/**
    173. 	 * @Title: removeByBids
    174. 	 * @Description: 根据bid删除附件
    175. 	 * @param bids
    176. 	 * @return
    177. 	 * @throws Exception 
    178. 	 */
    179. 	@Override
    180. 	public int removeByBids(List bids) throws Exception {
    181. 		QueryWrapper paramWrapper = new QueryWrapper<>();
    182. 		paramWrapper.in("b_id", bids);
    183. 		List files = this.list(paramWrapper);
    184. 		int ret = this.remove(paramWrapper)?1:0;
    185. 		if(files == null || files.size() == 0)
    186. 			return 0;
    187. 		for(FileEntity file:files){
    188. 			String file_path = file.getPath();
    189. 			if(!sharedFile(file_path)){
    190. 				this.deleteFileOnDisk(file_path);
    191. 			}
    192. 		}
    193. 		return ret;
    194. 	}
    195.  
    196. 	@Override
    197. 	public FileEntity findByMd5Code(String md5Code) throws Exception {
    198. 		QueryWrapper queryWrapper = new QueryWrapper();
    199. 		queryWrapper.eq("md5code", md5Code);
    200. 		List list = this.list(queryWrapper);
    201. 		return StringUtils.isEmpty(list) ? null : list.get(0);
    202. 	}
    203. 	
    204. }

    3、控制层定义

            控制层主要用于接收前端WebUploader提交过来的请求，同时调用相应的服务后进行响应。关键如下面所示：
     

    1. package com.hngtghy.project.webupload.controller;
    2.  
    3. import java.io.File;
    4. import java.io.FileFilter;
    5. import java.io.FileInputStream;
    6. import java.io.FileOutputStream;
    7. import java.io.IOException;
    8. import java.io.InputStream;
    9. import java.io.OutputStream;
    10. import java.nio.channels.FileChannel;
    11. import java.util.ArrayList;
    12. import java.util.Arrays;
    13. import java.util.Collections;
    14. import java.util.Comparator;
    15. import java.util.HashSet;
    16. import java.util.List;
    17. import java.util.Set;
    18. import java.util.UUID;
    19.  
    20. import javax.servlet.http.HttpServletResponse;
    21.  
    22. import org.springframework.beans.factory.annotation.Autowired;
    23. import org.springframework.context.ApplicationContext;
    24. import org.springframework.stereotype.Controller;
    25. import org.springframework.ui.ModelMap;
    26. import org.springframework.web.bind.annotation.GetMapping;
    27. import org.springframework.web.bind.annotation.PostMapping;
    28. import org.springframework.web.bind.annotation.RequestMapping;
    29. import org.springframework.web.bind.annotation.RequestParam;
    30. import org.springframework.web.bind.annotation.ResponseBody;
    31. import org.springframework.web.multipart.MultipartFile;
    32.  
    33. import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
    34. import com.hngtghy.common.utils.StringUtils;
    35. import com.hngtghy.common.utils.file.FileTypeUtils;
    36. import com.hngtghy.framework.config.HngtghyConfig;
    37. import com.hngtghy.framework.event.FileUploadEvent;
    38. import com.hngtghy.framework.web.controller.BaseController;
    39. import com.hngtghy.framework.web.domain.AjaxResult;
    40. import com.hngtghy.framework.web.page.LayerTableDataInfo;
    41. import com.hngtghy.project.system.user.domain.User;
    42. import com.hngtghy.project.webupload.domain.FileEntity;
    43. import com.hngtghy.project.webupload.service.IFileService;
    44. import com.github.pagehelper.util.StringUtil;
    45.  
    46. /**
    47.  * 文件上传相关
    48.  * @author wuzuhu
    49.  */
    50. @Controller
    51. @RequestMapping("/uploadfile")
    52. public class UploadFileController extends BaseController {
    53. 	
    54. 	private String prefix = "upload/";
    55.  
    56. 	@Autowired
    57. 	private IFileService fileService;
    58.  
    59. 	private final int block_size = 5*1024*1024;
    60. 	
    61. 	@Autowired
    62. 	private ApplicationContext applicationContext;
    63. 	
    64. 	/**
    65. 	 * 文件列表页面
    66. 	 */
    67. 	@GetMapping("/main")
    68. 	public String main(ModelMap mmap,String bid,String tablename,String bizType,String multipleMode) {
    69. 		mmap.put("bid", bid);//业务表id
    70. 		mmap.put("temp_b_id", UUID.randomUUID().toString());
    71. 		mmap.put("tablename", tablename);//业务表名
    72. 		mmap.put("bizType", bizType);//业务类型
    73. 		mmap.put("multipleMode", multipleMode);//文件多选模式，默认为多选，为空即可。单选需要设置为:single
    74. 		return prefix + "fileTablePage";
    75. 	}
    76. 	
    77. 	/**
    78. 	 * 文件上传进度页面
    79. 	 */
    80. 	@GetMapping("/process")
    81. 	public String upload(ModelMap mmap) {
    82. 		return prefix + "uploadProcessModal";
    83. 	}
    84. 	
    85. 	/**
    86. 	 * 文件上传进度页面
    87. 	 */
    88. 	@GetMapping("/view")
    89. 	public String view(ModelMap mmap) {
    90. 		return prefix + "viewFile";
    91. 	}
    92. 	
    93. 	@ResponseBody
    94. 	@PostMapping("/bigUploader")
    95. 	public AjaxResult bigUploader(String chunk,String chunks,String fid,
    96. 			@RequestParam("file") MultipartFile multipartFile) {
    97. 		String path = HngtghyConfig.getProfile() + "/";
    98. 		try {
    99. 			FileEntity db_file = fileService.getOneByFid(fid);
    100. 			if (db_file == null) {
    101. 				return AjaxResult.error(AjaxResult.Type.WEBUPLOADERROR.value(),"没找到数据库记录");
    102. 			}
    103. 		    if(db_file.getState() == 1){
    104. 		    	return AjaxResult.success();
    105.             }
    106. 		    User user = getSysUser();
    107. 			String save_dir = path + (user != null ? user.getUserId() : "unkown") ;
    108. 			String chunk_dir = save_dir + "/" + fid ;
    109. 			boolean sign = chunks != null && Integer.parseInt(chunks) > 0;
    110. 			String tempmlDir = sign ? chunk_dir : save_dir;
    111. 			String chunkFilePath = sign ? chunk_dir + "/" + chunk : chunk_dir + "." + db_file.getType();
    112. 			if(!sign) {
    113. 				db_file.setState(1);
    114. 				fileService.updateById(db_file);
    115. 			}
    116. 			File tempml = new File(tempmlDir);
    117. 			if (!tempml.exists()) {
    118. 				tempml.mkdirs();
    119. 			}
    120. 			File chunkFile = new File(chunkFilePath);
    121. 			multipartFile.transferTo(chunkFile);
    122. 			return AjaxResult.success();
    123. 		} catch (Exception e) {
    124. 			return AjaxResult.error(AjaxResult.Type.WEBUPLOADERROR.value(),"服务端异常");
    125. 		}
    126. 	}
    127.  
    128. 	@ResponseBody
    129. 	@PostMapping("/merge")
    130. 	public AjaxResult merge(String action,String chunks,String chunk,String chunkSize,String temp_b_id,FileEntity fileObj) {
    131. 		try {
    132. 			User user = getSysUser();
    133. 			String fid = fileObj.getFid();
    134. 			//公共目录
    135. 			String commonDir = user != null ? String.valueOf(user.getUserId()) : "unkown";
    136. 			FileEntity db_file = fileService.getOneByFid(fid);
    137. 			String type = FileTypeUtils.getFileType(fileObj.getName());
    138. 			
    139. 			String path = HngtghyConfig.getProfile() + "/";
    140. 			String chunk_dir = path + commonDir + "/" + fid;
    141. 			
    142. 			//数据库保存目录考虑可迁移 add by wuzuhu on 2022-07-18
    143. 			String dbPath = commonDir + "/" + fid + "." + type;
    144. 			
    145. 			if (action.equals("mergeChunks")) {
    146. 				return mergeChunks(db_file,chunk_dir,chunks,path + dbPath);
    147. 			} 
    148. 			if (action.equals("checkChunk")) {
    149. 				String chunkfile_dir = chunk_dir + "/" + chunk;
    150. 				return checkChunk(chunkfile_dir,chunkSize);
    151. 			}
    152. 			if (action.equals("exSendFile")) {
    153. 				FileEntity md5File = fileService.findByMd5Code(fileObj.getMd5code());
    154. 				if(null != md5File) {
    155. 					dbPath = md5File.getPath();
    156. 					fileObj.setName(md5File.getName());
    157. 				}
    158. 				fileObj.setPath(dbPath);
    159. 				fileObj.setType(type);
    160. 				fileObj.setBid(temp_b_id);
    161. 				fileObj.setState(0);
    162. 				return exSendFile(db_file,path + dbPath,fileObj);
    163. 			}
    164. 		} catch (Exception e) {
    165. 			return AjaxResult.error();
    166. 		}
    167. 		return AjaxResult.success();
    168. 	}
    169. 	
    170. 	private AjaxResult checkChunk(String chunkfile_dir,String chunkSize) {
    171. 		File checkFile = new File(chunkfile_dir);
    172. 		if (checkFile.exists() && checkFile.length() == Integer.parseInt(chunkSize)) {
    173. 			return AjaxResult.error(2,"文件已存在");
    174. 		} else {
    175. 			return AjaxResult.success();
    176. 		}
    177. 	}
    178. 	
    179. 	@SuppressWarnings("resource")
    180. 	private AjaxResult mergeChunks(FileEntity db_file,String chunk_dir,String chunks,String f_path) throws IOException {
    181. 		if (db_file == null) {
    182. 			return AjaxResult.error(AjaxResult.Type.WEBUPLOADERROR.value(), "找不到数据");
    183. 		}
    184. 		if (db_file.getState() == 1) {
    185. 			//未分片文件上传成功合并成功后发布相应事件，各监听器自由监听并执行
    186. 			applicationContext.publishEvent(new FileUploadEvent(this, db_file));
    187. 			return AjaxResult.success();
    188.         }
    189. 		if(db_file.getSize() > block_size){
    190. 			File f = new File(chunk_dir);
    191. 			File[] fileArray = f.listFiles(new FileFilter() {
    192. 				@Override
    193. 				public boolean accept(File pathname) {
    194. 					if (pathname.isDirectory()) {
    195. 						return false;
    196. 					}
    197. 					return true;
    198. 				}
    199. 			});
    200. 			if (fileArray == null || fileArray.length == 0) {
    201. 				return AjaxResult.error(AjaxResult.Type.WEBUPLOADERROR.value(), "找不到分片文件");
    202. 			}
    203. 			if (StringUtil.isNotEmpty(chunks) && fileArray.length != Integer.parseInt(chunks)) {
    204. 				return AjaxResult.error(AjaxResult.Type.WEBUPLOADERROR.value(), "分片文件数量错误");
    205. 			}
    206. 			List fileList = new ArrayList(Arrays.asList(fileArray));
    207. 			Collections.sort(fileList, new Comparator() {
    208. 				@Override
    209. 				public int compare(File o1, File o2) {
    210. 					if (Integer.parseInt(o1.getName()) < Integer.parseInt(o2.getName())) {
    211. 						return -1;
    212. 					}
    213. 					return 1;
    214. 				}
    215. 			});
    216. 			File outputFile = new File(f_path);
    217. 			outputFile.createNewFile();
    218. 			FileChannel outChnnel = new FileOutputStream(outputFile).getChannel();
    219. 			FileChannel inChannel;
    220. 			for (File file : fileList) {
    221. 				inChannel = new FileInputStream(file).getChannel();
    222. 				inChannel.transferTo(0, inChannel.size(), outChnnel);
    223. 				inChannel.close();
    224. 				file.delete();
    225. 			}
    226. 			outChnnel.close();
    227. 			db_file.setState(1);
    228. 			fileService.updateById(db_file);
    229. 			File tempFile = new File(chunk_dir);
    230. 			if (tempFile.isDirectory() && tempFile.exists()) {
    231. 				tempFile.delete();
    232. 			}
    233. 			//分片文件上传成功合并成功后发布相应事件，各监听器自由监听并执行
    234. 			applicationContext.publishEvent(new FileUploadEvent(this, db_file));
    235. 		}
    236. 		return AjaxResult.success();
    237. 	}
    238. 	
    239. 	private AjaxResult exSendFile(FileEntity db_file,String f_path,FileEntity fileObj) throws Exception {
    240. 		if (db_file != null) {
    241. 			return AjaxResult.success();
    242. 		}
    243. 		System.out.println("md5code==>" + fileObj.getMd5code() + "\t f_path=="+ f_path);
    244. 		System.out.println("fileObj id ==>" + fileObj.getId() + "\t fid===>" + fileObj.getFid());
    245. 		
    246. 		fileObj.setState(0);
    247. 		fileService.saveEntity(fileObj);
    248. 		//执行插入
    249. 		File file_indisk = new File(f_path);
    250. 		if (file_indisk.exists() && file_indisk.length() == fileObj.getSize()) {
    251. 			fileObj.setState(1);
    252. 			fileService.updateById(fileObj);
    253. 			//已上传文件上传成功合并成功后发布相应事件，各监听器自由监听并执行
    254. 			applicationContext.publishEvent(new FileUploadEvent(this, fileObj));
    255. 			return AjaxResult.error(2,"文件已存在");
    256. 		} else {
    257. 			return AjaxResult.success();
    258. 		}
    259. 	}
    260.  
    261. 	/**
    262. 	 * 删除文件
    263. 	 * @param fid
    264. 	 */
    265. 	@ResponseBody
    266. 	@RequestMapping("/delete")
    267. 	public AjaxResult delete(@RequestParam("ids[]") Long[] ids) throws Exception {
    268. 		fileService.deleteErrorFile(getSysUser());
    269. 		int result = fileService.removeByIds(ids);
    270. 		return result > 0 ? AjaxResult.success() : AjaxResult.error();
    271. 	}
    272.  
    273. 	/**
    274. 	 * 删除文件
    275. 	 * @param fid
    276. 	 */
    277. 	@ResponseBody
    278. 	@RequestMapping("/deleteByFid")
    279. 	public AjaxResult deleteByFid(String fid) throws Exception{
    280. 		fileService.deleteErrorFile(getSysUser());
    281. 		int result = fileService.removeByFid(fid);
    282. 		return result > 0 ? AjaxResult.success() : AjaxResult.error();
    283. 	}
    284.  
    285.  
    286. 	/**
    287. 	 * 查询文件列表
    288. 	 * @author zlz
    289. 	 * @date 2019年3月19日 下午3:16:32
    290. 	 */
    291. 	@ResponseBody
    292. 	@RequestMapping("/list")
    293. 	public LayerTableDataInfo list(String b_id, String b_ids, String f_id,String name,String tablename,String bizType) throws Exception{
    294. 		startLayerPage();
    295. 		QueryWrapper queryWrapper = new QueryWrapper();
    296. 		queryWrapper.eq("f_state", 1);
    297.         if(StringUtils.isNotBlank(b_id)){
    298.         	queryWrapper.eq("b_id", b_id);
    299.         }
    300.         if(StringUtils.isNotBlank(b_ids)){
    301.         	queryWrapper.in("b_id", splitIds(b_ids));
    302.         }
    303.         if(StringUtils.isNotBlank(f_id)){
    304.         	queryWrapper.eq("f_id", f_id);
    305.         }
    306.         if(StringUtils.isNotBlank(name)){
    307.         	queryWrapper.like("f_name", name);
    308.         }
    309.         if(StringUtils.isNotBlank(tablename)) {
    310.         	queryWrapper.eq("table_name", tablename);
    311.         }
    312.         if(StringUtils.isNotBlank(bizType)){
    313.         	queryWrapper.eq("biz_type", bizType);
    314.         }
    315.         List list = fileService.findListByQueryWrapper(queryWrapper);
    316. 		return getLayerDataTable(list);
    317. 	}
    318. 	
    319. 	/**
    320. 	 * 将 *,*,*,*,*, 样的ID放入Set中
    321. 	 * @return 包含对应各id的数值的列表, 不包含重复id
    322. 	 */
    323. 	private static final Set splitIds(String ids){
    324. 		Set idsSet = new HashSet();
    325. 		if(ids != null && !ids.isEmpty()){
    326. 			String[] data = ids.split(",");
    327. 			if(data != null){
    328. 				for(String d : data){
    329. 					idsSet.add(d);
    330. 				}
    331. 			}
    332. 		}
    333. 		return idsSet;
    334. 	}
    335.  
    336. 	/**
    337. 	 * 下载文件
    338. 	 * @param fid
    339. 	 * @param response
    340. 	 */
    341. 	@RequestMapping("/download")
    342. 	public void download(String fid, HttpServletResponse response) throws Exception{
    343. 		FileEntity file = fileService.getOneByFid(fid);
    344. 		if (file == null) {
    345. 			return;
    346. 		}
    347. 		OutputStream to = null;
    348. 		try {
    349. 			String filename = file.getName();
    350. 			response.setContentType("text/html");
    351. 			response.setHeader("Content-Disposition","attachment;filename=\"" + new String(filename.getBytes(), "ISO-8859-1") + "\"");
    352. 			to = response.getOutputStream();
    353. 			this.getFileInfo(to, file);
    354. 		} catch (Exception e) {
    355. 			e.printStackTrace();
    356. 		} finally {
    357. 			if (to != null) {
    358. 				try {
    359. 					to.flush();
    360. 					to.close();
    361. 				} catch (IOException e) {
    362. 				}
    363. 			}
    364. 		}
    365. 	}
    366. 	
    367. 	private void getFileInfo(OutputStream to,FileEntity file) throws Exception{
    368. 		InputStream in = null;
    369. 		try{
    370. 			File file_download = new File(HngtghyConfig.getProfile() + "/" + file.getPath());
    371. 			in = new FileInputStream(file_download);
    372. 			byte[] buffer = new byte[1024];
    373. 			int got = -1;
    374. 			while ((got = in.read(buffer)) != -1){
    375. 				to.write(buffer,0,got);
    376. 			}
    377. 		}catch(Exception e){
    378. 			throw e;
    379. 		}finally{
    380. 			if(in != null){
    381. 				try {
    382. 					in.close();
    383. 				} catch (IOException e) {
    384. 					
    385. 				}
    386. 			}
    387. 		}
    388. 	}
    389. 	
    390. }

    其类图如下所示：

     

            以上就是完整的后端接收处理逻辑，使用java代码进行开发，这里使用了本地磁盘的方式进行存储，你可以自己扩展一下，比如可以集成分布式存储，都是可以的，这样改造后可以当成企业统一的服务。

     三、总结

            以上就是本文的主要内容，介绍统一附件管理服务的后台开发逻辑。简单介绍了后台附件表的设计，以后围绕附件管理的相关实体定义、服务层、控制层定义。通过给出相关类的类图，便于大家对整体的代码层次有一个比较全面的认识。

  • 相关阅读:
    web前端 html+css+javascript游戏网页设计实例 (网页制作课作业)
    docker 安装 redis
    Echarts社区开源地址
    JavaScript
    xray证书安装使用及Burp联动
    添砖Java之路(其六)——通过集合制作的学生信息管理系统
    C++学习之强制类型转换
    vue 的 render 函数的用法：new Vue({ render: h =＞ h(App), }).$mount(‘#app‘)
    Java筑基32-IO流02-节点流&处理流
    「笔翰如流」上线，TFS-CLUB 社区活动来袭 〖 你更文、我送礼 〗限量绝版谷歌周边等你拿
  • 原文地址：https://blog.csdn.net/yelangkingwuzuhu/article/details/127361371