背景
在之前的博文中，简单介绍了如何扩展Ruoyi的大附件上传及统一管理一篇，原文地址：基于Ruoyi和WebUploader的统一附件管理扩展（上）。之前的博文主要集中在前台的讲解，前台主要是围绕WebUploader组件来说明，对应的后台处理没有仔细讲解。本文作为下篇，主要围绕大附件上传的后台设计及实现，通过相关的UML建模，具体阐述后台是如何进行对应的处理。包括断点续传，文件重复判断等特殊功能的实例讲解。希望对你在项目中的使用有所启发。
一、数据库定义
1、目的
定义数据实体的目的是为了可以在各业务中通用，提供统一的组件及封装，同时为了可以在文件处理时满足实时对文件是否重复上传做校验，因此有必要采用数据库的方式进行存储。当然，这里的数据存储未必要使用关系型数据库，使用非关系型数据库，比如MongoDB也是可以的。只需要达到我们的目的即可。
2、数据库物理表设计
设计统一附件管理模块，不仅要统一文件上传的操作界面，同时要提供统一的API可以对附件进行检索，因而要将一个业务表的表名传递到附件表中。同时为了，防止一张业务表在不同状态下也可以拥有不同的附件，额外增加一个业务类型的字段。比如一个流程审核的业务，在流程新建阶段关联的附件和审批中关联的附件可以是不一样的。由此字段可以区分开，为了在文件上传过程中加速，也避免减少同一份文件反复上传到磁盘中，造成不必要的资源浪费，因此非常有必要对资源进行判重。关于文件判重，可以参考之前的博文：java文件上传判重姿势浅谈，这里有比较详细的说明。根据上述的需求，可以得到一个数据库物理表的表结构：
这是一份基于postgresql数据库的表结构，同理在mysql或者其它的数据库下，是一样的，只是数据类型需要稍微修改一下而已。这里将表结构对应的sql文件分享一下：
-- ----------------------------
-- Table structure for biz_file
-- ----------------------------
DROP TABLE IF EXISTS "biz_file";
CREATE TABLE "biz_file" (
"id" int8 NOT NULL,
"f_id" varchar(100) ,
"b_id" varchar(100) ,
"f_type" varchar(30) NOT NULL,
"f_name" varchar(512) NOT NULL,
"f_desc" varchar(512) ,
"f_state" int2,
"f_size" int8 NOT NULL,
"f_path" varchar(1024) NOT NULL,
"table_name" varchar(255) ,
"md5code" varchar(255) ,
"directory" varchar(1024) ,
"create_by" varchar(64) ,
"create_time" timestamp(6),
"update_by" varchar(64) ,
"update_time" timestamp(6),
"biz_type" varchar(30)
)
;
COMMENT ON COLUMN "biz_file"."id" IS '主键';
COMMENT ON COLUMN "biz_file"."f_id" IS '文件id';
COMMENT ON COLUMN "biz_file"."b_id" IS '业务id';
COMMENT ON COLUMN "biz_file"."f_type" IS '文件类型';
COMMENT ON COLUMN "biz_file"."f_name" IS '名称';
COMMENT ON COLUMN "biz_file"."f_desc" IS '文件描述';
COMMENT ON COLUMN "biz_file"."f_state" IS '文件状态';
COMMENT ON COLUMN "biz_file"."f_size" IS '文件大小';
COMMENT ON COLUMN "biz_file"."f_path" IS '文件路径';
COMMENT ON COLUMN "biz_file"."table_name" IS '业务表名';
COMMENT ON COLUMN "biz_file"."md5code" IS 'md5code';
COMMENT ON COLUMN "biz_file"."directory" IS '文件目录';
COMMENT ON COLUMN "biz_file"."create_by" IS '创建人';
COMMENT ON COLUMN "biz_file"."create_time" IS '创建时间';
COMMENT ON COLUMN "biz_file"."update_by" IS '更新人';
COMMENT ON COLUMN "biz_file"."update_time" IS '更新时间';
COMMENT ON COLUMN "biz_file"."biz_type" IS '业务类型';
COMMENT ON TABLE "biz_file" IS '系统附件信息表,用于保存文件上传信息';
-- ----------------------------
-- Primary Key structure for table biz_file
-- ----------------------------
ALTER TABLE "biz_file" ADD CONSTRAINT "pk_biz_file" PRIMARY KEY ("id");
二、JAVA后台服务定义
1、实体类定义
实体类主要用于定义跟数据库表相互映射的对象信息，使用sql语句来操作数据库信息。示例中的代码均需要配置lombok来简化相关类的开发工作量。
package com.hngtghy.project.webupload.domain;
import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableName;
import com.hngtghy.framework.web.domain.BaseEntity;
import lombok.AllArgsConstructor;
import lombok.Getter;
import lombok.NoArgsConstructor;
import lombok.Setter;
import lombok.ToString;
@TableName("biz_file")
@NoArgsConstructor
@AllArgsConstructor
@Setter
@Getter
@ToString
public class FileEntity extends BaseEntity {
private static final long serialVersionUID = 1L;
private Long id;
@TableField(value = "f_id")
private String fid;
@TableField(value = "b_id")
private String bid;
@TableField(value = "f_type")
private String type;
@TableField(value = "f_name")
private String name;
@TableField(value = "f_desc")
private String desc;
@TableField(value = "f_state")
private Integer state;
@TableField(value = "f_size")
private Long size;
@TableField(value = "f_path")
private String path;
@TableField(value = "table_name")
private String tablename = "temp_table";
private String md5code;
private String directory;
@TableField(value = "biz_type")
private String bizType;
}
2、数据服务和业务层处理
数据服务，主要基于mybatis-plus来进行增删改查操作，而业务层包含了基础的业务封装，除了调用数据服务外，还有一些额外的业务处理操作。
package com.hngtghy.project.webupload.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.hngtghy.project.webupload.domain.FileEntity;
public interface FileMapper extends BaseMapper<FileEntity> {
}
在服务层中定义了对附件对象的删除、查询、修改、保存的功能方法，以接口的形式进行定义。
package com.hngtghy.project.webupload.service;
import java.util.List;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.IService;
import com.hngtghy.project.system.user.domain.User;
import com.hngtghy.project.webupload.domain.FileEntity;
public interface IFileService extends IService<FileEntity>{
int saveEntity(FileEntity entity) throws Exception;
int updateEntity(FileEntity entity) throws Exception;
int removeByIds(Long [] ids) throws Exception;
int removeByBids(List<String> bids) throws Exception;
FileEntity getOneByFid(String fid) throws Exception;
int removeByFid(String fid) throws Exception;
void deleteErrorFile(User user);
Long findFileSizeByWrapper(QueryWrapper<FileEntity> queryWrapper);
List<FileEntity> findListByQueryWrapper(QueryWrapper<FileEntity> queryWrapper);
FileEntity findByMd5Code(String md5Code) throws Exception;
}
其具体的实现类如下，这里主要针对在接口定义的方法进行重写，以满足新的业务需求，比较重要的方法是findByMd5Code，通过这个方法到数据库中查询是否有重复的文件，当有重复文件后，将不再重复上传，关键代码如下：
package com.hngtghy.project.webupload.service;
import java.io.File;
import java.io.FileFilter;
import java.sql.PreparedStatement;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Propagation;
import org.springframework.transaction.annotation.Transactional;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.hngtghy.common.utils.StringUtils;
import com.hngtghy.common.utils.security.ShiroUtils;
import com.hngtghy.framework.aspectj.lang.annotation.DataSource;
import com.hngtghy.framework.config.HngtghyConfig;
import com.hngtghy.project.system.user.domain.User;
import com.hngtghy.project.webupload.domain.FileEntity;
import com.hngtghy.project.webupload.mapper.FileMapper;
@Service
public class FileServiceImpl extends ServiceImpl<FileMapper, FileEntity> implements IFileService {
@Override
@Transactional(propagation = Propagation.REQUIRED, rollbackFor = Exception.class)
public int saveEntity(FileEntity entity) throws Exception {
entity.setCreateBy(ShiroUtils.getLoginName());
entity.setCreateTime(new Date());
int result = this.save(entity) ? 1 : 0;
return result;
}
@Override
@Transactional(propagation = Propagation.REQUIRED, rollbackFor = Exception.class)
public int updateEntity(FileEntity entity) throws Exception {
int result = this.updateById(entity) ? 1 : 0;
return result;
}
@Override
@Transactional(propagation = Propagation.REQUIRED, rollbackFor = Exception.class)
public int removeByIds(Long[] ids) throws Exception {
List<Long> removeList = Arrays.asList(ids);
for (Long id : ids) {
deleteFileOnDiskById(id);
}
int result = this.removeByIds(removeList) ? 1 : 0;
return result;
}
@Override
@Transactional(propagation = Propagation.REQUIRED, rollbackFor = Exception.class)
public FileEntity getOneByFid(String fid) throws Exception {
QueryWrapper<FileEntity> queryWrapper = new QueryWrapper<FileEntity>();
queryWrapper.eq("f_id", fid);
FileEntity file = this.getOne(queryWrapper);
return file;
}
@Override
public int removeByFid(String fid) throws Exception {
QueryWrapper<FileEntity> queryWrapper = new QueryWrapper<FileEntity>();
queryWrapper.eq("f_id", fid);
deleteFileOnDiskByFid(fid);
int result = this.remove(queryWrapper) ? 1 : 0;
return result;
}
private void deleteFileOnDiskByFid(String fid) throws Exception {
FileEntity file = this.getOneByFid(fid);
String file_path = file.getPath();
if(!sharedFile(file_path)){
this.deleteFileOnDisk(file_path);
}
}
private void deleteFileOnDiskById(Long id) throws Exception {
FileEntity file = this.getById(id);
String file_path = file.getPath();
if(!sharedFile(file_path)){
this.deleteFileOnDisk(file.getPath());
}
}
private boolean sharedFile(String path){
QueryWrapper<FileEntity> queryWrapper = new QueryWrapper<FileEntity>();
queryWrapper.eq("f_path", path);
queryWrapper.eq("f_state", 1);
List<FileEntity> files = this.list(queryWrapper);
if (files != null && files.size() > 1) {
return true;
}
return false;
}
private void deleteFileOnDisk(String path) throws Exception {
File file = new File(HngtghyConfig.getProfile() + "/" + path);
file.deleteOnExit();
if (file.isFile() && file.exists()) {
file.delete();
}
}
/**
* 删除失败文件、未绑定文件
* @author
* @date
*/
public void deleteErrorFile(User user){
try{
String path = HngtghyConfig.getProfile() + "/";
QueryWrapper<FileEntity> queryWrapper = new QueryWrapper<FileEntity>();
queryWrapper.eq("f_state", 0);
queryWrapper.lt("create_time", "now() - INTERVAL '3 days'");//三天前的失败文件
List<FileEntity> files = this.list(queryWrapper);
if (files != null && files.size() > 0) {
this.remove(queryWrapper);
for(FileEntity file:files){
deleteFileOnDiskById(file.getId());
}
}
queryWrapper = new QueryWrapper<FileEntity>();
queryWrapper.eq("linked", 0);
queryWrapper.lt("create_time", "now() - INTERVAL '3 days'");
files = this.list(queryWrapper);
if (files != null && files.size() > 0) {
this.remove(queryWrapper);
for(FileEntity file:files){
deleteFileOnDiskById(file.getId());
}
}
//三天前的分片临时目录
String save_path = path + (user != null ? user.getUserId() : "unkown");
File directory = new File(save_path);
File[] fileArray = directory.listFiles(new FileFilter() {
@Override
public boolean accept(File pathname) {
long last = pathname.lastModified();
long diff = System.currentTimeMillis() - last;
boolean del = (diff - 24 * 60 * 60 * 1000) > 0;
if (pathname.isDirectory() && del) {
return true;
}
return false;
}
});
for(File dir : fileArray){
dir.delete();
}
}catch(Exception e){
//无需处理该异常
}
}
@Override
public Long findFileSizeByWrapper(QueryWrapper<FileEntity> queryWrapper) {
Long result = this.count(queryWrapper);
return result;
}
@Override
public List<FileEntity> findListByQueryWrapper(QueryWrapper<FileEntity> queryWrapper) {
List<FileEntity> result = this.list(queryWrapper);
return result;
}
/**
* @Title: removeByBids
* @Description: 根据bid删除附件
* @param bids
* @return
* @throws Exception
*/
@Override
public int removeByBids(List<String> bids) throws Exception {
QueryWrapper<FileEntity> paramWrapper = new QueryWrapper<>();
paramWrapper.in("b_id", bids);
List<FileEntity> files = this.list(paramWrapper);
int ret = this.remove(paramWrapper)?1:0;
if(files == null || files.size() == 0)
return 0;
for(FileEntity file:files){
String file_path = file.getPath();
if(!sharedFile(file_path)){
this.deleteFileOnDisk(file_path);
}
}
return ret;
}
@Override
public FileEntity findByMd5Code(String md5Code) throws Exception {
QueryWrapper<FileEntity> queryWrapper = new QueryWrapper<FileEntity>();
queryWrapper.eq("md5code", md5Code);
List<FileEntity> list = this.list(queryWrapper);
return StringUtils.isEmpty(list) ? null : list.get(0);
}
}
3、控制层定义
控制层主要用于接收前端WebUploader提交过来的请求，同时调用相应的服务后进行响应。关键如下面所示：
package com.hngtghy.project.webupload.controller;
import java.io.File;
import java.io.FileFilter;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.channels.FileChannel;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.UUID;
import javax.servlet.http.HttpServletResponse;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationContext;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.hngtghy.common.utils.StringUtils;
import com.hngtghy.common.utils.file.FileTypeUtils;
import com.hngtghy.framework.config.HngtghyConfig;
import com.hngtghy.framework.event.FileUploadEvent;
import com.hngtghy.framework.web.controller.BaseController;
import com.hngtghy.framework.web.domain.AjaxResult;
import com.hngtghy.framework.web.page.LayerTableDataInfo;
import com.hngtghy.project.system.user.domain.User;
import com.hngtghy.project.webupload.domain.FileEntity;
import com.hngtghy.project.webupload.service.IFileService;
import com.github.pagehelper.util.StringUtil;
/**
* 文件上传相关
* @author wuzuhu
*/
@Controller
@RequestMapping("/uploadfile")
public class UploadFileController extends BaseController {
private String prefix = "upload/";
@Autowired
private IFileService fileService;
private final int block_size = 5*1024*1024;
@Autowired
private ApplicationContext applicationContext;
/**
* 文件列表页面
*/
@GetMapping("/main")
public String main(ModelMap mmap,String bid,String tablename,String bizType,String multipleMode) {
mmap.put("bid", bid);//业务表id
mmap.put("temp_b_id", UUID.randomUUID().toString());
mmap.put("tablename", tablename);//业务表名
mmap.put("bizType", bizType);//业务类型
mmap.put("multipleMode", multipleMode);//文件多选模式，默认为多选，为空即可。单选需要设置为:single
return prefix + "fileTablePage";
}
/**
* 文件上传进度页面
*/
@GetMapping("/process")
public String upload(ModelMap mmap) {
return prefix + "uploadProcessModal";
}
/**
* 文件上传进度页面
*/
@GetMapping("/view")
public String view(ModelMap mmap) {
return prefix + "viewFile";
}
@ResponseBody
@PostMapping("/bigUploader")
public AjaxResult bigUploader(String chunk,String chunks,String fid,
@RequestParam("file") MultipartFile multipartFile) {
String path = HngtghyConfig.getProfile() + "/";
try {
FileEntity db_file = fileService.getOneByFid(fid);
if (db_file == null) {
return AjaxResult.error(AjaxResult.Type.WEBUPLOADERROR.value(),"没找到数据库记录");
}
if(db_file.getState() == 1){
return AjaxResult.success();
}
User user = getSysUser();
String save_dir = path + (user != null ? user.getUserId() : "unkown") ;
String chunk_dir = save_dir + "/" + fid ;
boolean sign = chunks != null && Integer.parseInt(chunks) > 0;
String tempmlDir = sign ? chunk_dir : save_dir;
String chunkFilePath = sign ? chunk_dir + "/" + chunk : chunk_dir + "." + db_file.getType();
if(!sign) {
db_file.setState(1);
fileService.updateById(db_file);
}
File tempml = new File(tempmlDir);
if (!tempml.exists()) {
tempml.mkdirs();
}
File chunkFile = new File(chunkFilePath);
multipartFile.transferTo(chunkFile);
return AjaxResult.success();
} catch (Exception e) {
return AjaxResult.error(AjaxResult.Type.WEBUPLOADERROR.value(),"服务端异常");
}
}
@ResponseBody
@PostMapping("/merge")
public AjaxResult merge(String action,String chunks,String chunk,String chunkSize,String temp_b_id,FileEntity fileObj) {
try {
User user = getSysUser();
String fid = fileObj.getFid();
//公共目录
String commonDir = user != null ? String.valueOf(user.getUserId()) : "unkown";
FileEntity db_file = fileService.getOneByFid(fid);
String type = FileTypeUtils.getFileType(fileObj.getName());
String path = HngtghyConfig.getProfile() + "/";
String chunk_dir = path + commonDir + "/" + fid;
//数据库保存目录考虑可迁移 add by wuzuhu on 2022-07-18
String dbPath = commonDir + "/" + fid + "." + type;
if (action.equals("mergeChunks")) {
return mergeChunks(db_file,chunk_dir,chunks,path + dbPath);
}
if (action.equals("checkChunk")) {
String chunkfile_dir = chunk_dir + "/" + chunk;
return checkChunk(chunkfile_dir,chunkSize);
}
if (action.equals("exSendFile")) {
FileEntity md5File = fileService.findByMd5Code(fileObj.getMd5code());
if(null != md5File) {
dbPath = md5File.getPath();
fileObj.setName(md5File.getName());
}
fileObj.setPath(dbPath);
fileObj.setType(type);
fileObj.setBid(temp_b_id);
fileObj.setState(0);
return exSendFile(db_file,path + dbPath,fileObj);
}
} catch (Exception e) {
return AjaxResult.error();
}
return AjaxResult.success();
}
private AjaxResult checkChunk(String chunkfile_dir,String chunkSize) {
File checkFile = new File(chunkfile_dir);
if (checkFile.exists() && checkFile.length() == Integer.parseInt(chunkSize)) {
return AjaxResult.error(2,"文件已存在");
} else {
return AjaxResult.success();
}
}
@SuppressWarnings("resource")
private AjaxResult mergeChunks(FileEntity db_file,String chunk_dir,String chunks,String f_path) throws IOException {
if (db_file == null) {
return AjaxResult.error(AjaxResult.Type.WEBUPLOADERROR.value(), "找不到数据");
}
if (db_file.getState() == 1) {
//未分片文件上传成功合并成功后发布相应事件，各监听器自由监听并执行
applicationContext.publishEvent(new FileUploadEvent(this, db_file));
return AjaxResult.success();
}
if(db_file.getSize() > block_size){
File f = new File(chunk_dir);
File[] fileArray = f.listFiles(new FileFilter() {
@Override
public boolean accept(File pathname) {
if (pathname.isDirectory()) {
return false;
}
return true;
}
});
if (fileArray == null || fileArray.length == 0) {
return AjaxResult.error(AjaxResult.Type.WEBUPLOADERROR.value(), "找不到分片文件");
}
if (StringUtil.isNotEmpty(chunks) && fileArray.length != Integer.parseInt(chunks)) {
return AjaxResult.error(AjaxResult.Type.WEBUPLOADERROR.value(), "分片文件数量错误");
}
List<File> fileList = new ArrayList<File>(Arrays.asList(fileArray));
Collections.sort(fileList, new Comparator<File>() {
@Override
public int compare(File o1, File o2) {
if (Integer.parseInt(o1.getName()) < Integer.parseInt(o2.getName())) {
return -1;
}
return 1;
}
});
File outputFile = new File(f_path);
outputFile.createNewFile();
FileChannel outChnnel = new FileOutputStream(outputFile).getChannel();
FileChannel inChannel;
for (File file : fileList) {
inChannel = new FileInputStream(file).getChannel();
inChannel.transferTo(0, inChannel.size(), outChnnel);
inChannel.close();
file.delete();
}
outChnnel.close();
db_file.setState(1);
fileService.updateById(db_file);
File tempFile = new File(chunk_dir);
if (tempFile.isDirectory() && tempFile.exists()) {
tempFile.delete();
}
//分片文件上传成功合并成功后发布相应事件，各监听器自由监听并执行
applicationContext.publishEvent(new FileUploadEvent(this, db_file));
}
return AjaxResult.success();
}
private AjaxResult exSendFile(FileEntity db_file,String f_path,FileEntity fileObj) throws Exception {
if (db_file != null) {
return AjaxResult.success();
}
System.out.println("md5code==>" + fileObj.getMd5code() + "\t f_path=="+ f_path);
System.out.println("fileObj id ==>" + fileObj.getId() + "\t fid===>" + fileObj.getFid());
fileObj.setState(0);
fileService.saveEntity(fileObj);
//执行插入
File file_indisk = new File(f_path);
if (file_indisk.exists() && file_indisk.length() == fileObj.getSize()) {
fileObj.setState(1);
fileService.updateById(fileObj);
//已上传文件上传成功合并成功后发布相应事件，各监听器自由监听并执行
applicationContext.publishEvent(new FileUploadEvent(this, fileObj));
return AjaxResult.error(2,"文件已存在");
} else {
return AjaxResult.success();
}
}
/**
* 删除文件
* @param fid
*/
@ResponseBody
@RequestMapping("/delete")
public AjaxResult delete(@RequestParam("ids[]") Long[] ids) throws Exception {
fileService.deleteErrorFile(getSysUser());
int result = fileService.removeByIds(ids);
return result > 0 ? AjaxResult.success() : AjaxResult.error();
}
/**
* 删除文件
* @param fid
*/
@ResponseBody
@RequestMapping("/deleteByFid")
public AjaxResult deleteByFid(String fid) throws Exception{
fileService.deleteErrorFile(getSysUser());
int result = fileService.removeByFid(fid);
return result > 0 ? AjaxResult.success() : AjaxResult.error();
}
/**
* 查询文件列表
* @author zlz
* @date 2019年3月19日 下午3:16:32
*/
@ResponseBody
@RequestMapping("/list")
public LayerTableDataInfo list(String b_id, String b_ids, String f_id,String name,String tablename,String bizType) throws Exception{
startLayerPage();
QueryWrapper<FileEntity> queryWrapper = new QueryWrapper<FileEntity>();
queryWrapper.eq("f_state", 1);
if(StringUtils.isNotBlank(b_id)){
queryWrapper.eq("b_id", b_id);
}
if(StringUtils.isNotBlank(b_ids)){
queryWrapper.in("b_id", splitIds(b_ids));
}
if(StringUtils.isNotBlank(f_id)){
queryWrapper.eq("f_id", f_id);
}
if(StringUtils.isNotBlank(name)){
queryWrapper.like("f_name", name);
}
if(StringUtils.isNotBlank(tablename)) {
queryWrapper.eq("table_name", tablename);
}
if(StringUtils.isNotBlank(bizType)){
queryWrapper.eq("biz_type", bizType);
}
List<FileEntity> list = fileService.findListByQueryWrapper(queryWrapper);
return getLayerDataTable(list);
}
/**
* 将 *,*,*,*,*, 样的ID放入Set中
* @return 包含对应各id的数值的列表, 不包含重复id
*/
private static final Set<String> splitIds(String ids){
Set<String> idsSet = new HashSet<String>();
if(ids != null && !ids.isEmpty()){
String[] data = ids.split(",");
if(data != null){
for(String d : data){
idsSet.add(d);
}
}
}
return idsSet;
}
/**
* 下载文件
* @param fid
* @param response
*/
@RequestMapping("/download")
public void download(String fid, HttpServletResponse response) throws Exception{
FileEntity file = fileService.getOneByFid(fid);
if (file == null) {
return;
}
OutputStream to = null;
try {
String filename = file.getName();
response.setContentType("text/html");
response.setHeader("Content-Disposition","attachment;filename=\"" + new String(filename.getBytes(), "ISO-8859-1") + "\"");
to = response.getOutputStream();
this.getFileInfo(to, file);
} catch (Exception e) {
e.printStackTrace();
} finally {
if (to != null) {
try {
to.flush();
to.close();
} catch (IOException e) {
}
}
}
}
private void getFileInfo(OutputStream to,FileEntity file) throws Exception{
InputStream in = null;
try{
File file_download = new File(HngtghyConfig.getProfile() + "/" + file.getPath());
in = new FileInputStream(file_download);
byte[] buffer = new byte[1024];
int got = -1;
while ((got = in.read(buffer)) != -1){
to.write(buffer,0,got);
}
}catch(Exception e){
throw e;
}finally{
if(in != null){
try {
in.close();
} catch (IOException e) {
}
}
}
}
}
其类图如下所示：
以上就是完整的后端接收处理逻辑，使用java代码进行开发，这里使用了本地磁盘的方式进行存储，你可以自己扩展一下，比如可以集成分布式存储，都是可以的，这样改造后可以当成企业统一的服务。
三、总结
以上就是本文的主要内容，介绍统一附件管理服务的后台开发逻辑。简单介绍了后台附件表的设计，以后围绕附件管理的相关实体定义、服务层、控制层定义。通过给出相关类的类图，便于大家对整体的代码层次有一个比较全面的认识。
本文转载自: https://blog.csdn.net/yelangkingwuzuhu/article/details/127361371
版权归原作者 yelangking1 所有， 如有侵权，请联系我们删除。
版权归原作者 yelangking1 所有， 如有侵权，请联系我们删除。