package com.mzl.flower.web.upload;
|
|
import com.mzl.flower.base.BaseController;
|
import com.mzl.flower.base.R;
|
import com.mzl.flower.config.UploadProperties;
|
import com.mzl.flower.config.exception.ValidationException;
|
import com.mzl.flower.dto.response.upload.UploadResultDTO;
|
import com.mzl.flower.utils.FileUtil;
|
import com.mzl.flower.utils.UUIDGenerator;
|
import io.swagger.annotations.Api;
|
import io.swagger.annotations.ApiOperation;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.http.ResponseEntity;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.ResponseBody;
|
import org.springframework.web.bind.annotation.RestController;
|
import org.springframework.web.multipart.MultipartFile;
|
import org.springframework.web.multipart.MultipartHttpServletRequest;
|
|
import javax.servlet.http.HttpServletRequest;
|
import java.util.ArrayList;
|
import java.util.Arrays;
|
import java.util.Iterator;
|
import java.util.List;
|
|
@RestController
|
@RequestMapping("/api/upload")
|
@Api(value = "文件上传", tags = "文件上传")
|
public class FileUploadResource extends BaseController {
|
|
@Autowired
|
private UploadProperties uploadProperties;
|
private static List<String> fileType = Arrays.asList("zip", "jpg", "png", "gif", "jpeg", "pdf", "doc", "docx", "xls", "xlsx", "ppt", "pptx", "mp4", "wmv", "rmvb", "mpg", "mpeg", "3gp", "mov", "m4v", "avi", "flv");
|
|
|
|
@PostMapping("/file")
|
@ResponseBody
|
@ApiOperation(value = "上传文件", httpMethod = "POST", notes = "上传文件")
|
public ResponseEntity<?> uploadFile(final String _fileName, HttpServletRequest request) throws Exception {
|
List<UploadResultDTO> result = new ArrayList<UploadResultDTO>();
|
|
MultipartHttpServletRequest mRequest = (MultipartHttpServletRequest) request;
|
Iterator<String> iter = mRequest.getFileNames();
|
while (iter.hasNext()) {
|
MultipartFile file = mRequest.getFile(iter.next());
|
String fileName = null;
|
if (StringUtils.isNotEmpty(_fileName)) {
|
fileName = _fileName;
|
} else {
|
fileName = file.getOriginalFilename();
|
}
|
//获取文件类型
|
String extName;
|
try {
|
extName = fileName.substring(fileName.lastIndexOf("."))
|
.toLowerCase();
|
extName = extName.replace(".", "");
|
} catch (Exception e) {
|
extName = "";
|
}
|
if (!fileType.contains(extName.toLowerCase())) {
|
throw new ValidationException("文件类型格式不正确");
|
}
|
|
|
byte[] bytes = file.getBytes();
|
String path = uploadProperties.getLocalServerPath();
|
UploadResultDTO uploadResult = new UploadResultDTO();
|
uploadResult.setName(fileName);
|
fileName = UUIDGenerator.getUUID() + FileUtil.getFileEndfix(fileName);
|
String url = FileUtil.saveFile(path, bytes, fileName, uploadProperties.getFileGroup());
|
|
uploadResult.setUrl(uploadProperties.getFileServerPath() + url);
|
uploadResult.setType(extName);
|
uploadResult.setSize(file.getSize());
|
|
result.add(uploadResult);
|
}
|
if (result.isEmpty()) {
|
throw new ValidationException("文件为空");
|
}
|
return returnData(R.SUCCESS.getCode(), result);
|
}
|
}
|
