SpringMVC文件上传（二）指定文件

来源：互联网发布：淘宝网达芙妮女凉鞋编辑：程序博客网时间：2024/06/06 01:18

在上面的一节中，我们查看运行的结果，即然在一个zip格式的文件，这显然不是我们所想要的，所以这一节，我们主要处理就是，要用户必须上传图片，而且上传的文件不能太大，如里不满足的话，那么页面将会报错，报错信息要以友好地形式提示。

1.上传是图片文件

我们修改pictureUploadController的类，修改的目的是文件必须是图片的。下面是笔者修改后的代码。

package masterSpringMvc.profile;import org.apache.tomcat.util.http.fileupload.IOUtils;import org.springframework.core.io.FileSystemResource;import org.springframework.core.io.Resource;import org.springframework.stereotype.Controller;import org.springframework.web.bind.annotation.RequestMapping;import org.springframework.web.bind.annotation.RequestMethod;import org.springframework.web.multipart.MultipartFile;import org.springframework.web.servlet.mvc.support.RedirectAttributes;import java.io.*;@Controllerpublic class PictureUploadController {public static final Resource PICTURES_DIR = newFileSystemResource("./pictures");@RequestMapping("upload")public String uploadPage() {return "profile/uploadPage";}@RequestMapping(value = "/upload", method = RequestMethod.POST)public String onUpload(MultipartFile file, RedirectAttributesredirectAttrs) throws IOException {if (file.isEmpty() || !isImage(file)) {redirectAttrs.addFlashAttribute("error", "Incorrect file.Please upload a picture.");return "redirect:/upload";}copyFileToPictures(file);return "profile/uploadPage";}private Resource copyFileToPictures(MultipartFile file) throwsIOException {String fileExtension = getFileExtension(file.getOriginalFilename());File tempFile = File.createTempFile("pic", fileExtension,PICTURES_DIR.getFile());try (InputStream in = file.getInputStream();OutputStream out = new FileOutputStream(tempFile)) {IOUtils.copy(in, out);}return new FileSystemResource(tempFile);}private boolean isImage(MultipartFile file) {return file.getContentType().startsWith("image");}

2.视图层修改

我们首先还是在两份国际化文件中添加下面的字段，对于法语的文件，读者自己翻译添加。

upload.io.exception=An error occurred while uploading the file. Please try again.upload.file.too.big=Your file is too big.

之后，我们在uploadPage中添加下面的代码，这个作用就是当文件上传不是图片时风们要友好地提示。

<div class="col s12 center red-text" th:text="${error}"th:if="${error}">Error during upload</div>

3.总结

这里笔者已经对上传文件做了处理，要求就是上传的文件必须是图片，如日不是图片，那么将会提示下图的信息。可是还有问题，笔者有之前说过了要对文件的大小作判断，如果大小超出了范围，我们也要友好提示，可是这一章节里，我们并没有实现。不要着急，我们下一节将会讲解。

源码下载：git@github.com:owenwilliam/masterSpringMVC.git

0 0