如何在Spring中的Rest中上传时验证文件类型,大小?

时间:2015-08-22 19:56:08

标签: java spring rest spring-mvc spring-restcontroller

我正在尝试使用spring和Rest实现文件上传。这就是我迄今为止所做的事情

@RestController
@RequestMapping("/rest/upload")
public class ProfileImageUploadController {

   @Autowired
   ImageValidator imageValidator;

   @RequestMapping(value="/{userId}/image", method=RequestMethod.POST)
   public @ResponseBody String handleFileUpload(
        @PathVariable("userId") Integer userId,
        @ModelAttribute("image") SingleImageFile image,
        BindingResult result){

       MultipartFile file = image.getFile();
       imageValidator.validate(file, result);
       if(!result.hasErrors()){
           String name = file.getOriginalFilename();
           try{
               file.transferTo(new File("/home/maclein/Desktop/"+name));
               return "You have successfully uploaded " + name + "!";
           }catch(Exception e){
               return "You have failed to upload " + name + " => " + e.getMessage();
           }
       } else {
           return result.getFieldErrors().toString();
       }
   }
}

这是我的ImageValidator

@Component
public class ImageValidator implements Validator {
   @Override
   public boolean supports(Class<?> arg0) {
       // TODO Auto-generated method stub
       return false;
   }

   @Override
   public void validate(Object uploadedFile, Errors error) {
       MultipartFile file = (MultipartFile) uploadedFile;

       if(file.isEmpty() || file.getSize()==0)
           error.rejectValue("file", "Please select a file");
       if(!(file.getContentType().toLowerCase().equals("image/jpg") 
            || file.getContentType().toLowerCase().equals("image/jpeg") 
            || file.getContentType().toLowerCase().equals("image/png"))){
           error.rejectValue("file", "jpg/png file types are only supported");
       }
   }
}

但是在通过邮递员进行测试时,如果文件是pdf但是以奇怪的方式显示错误。这是错误的字符串表示

  

&#34; [对象中的字段错误&#39;图像&#39;在字段&#39;文件&#39;:被拒绝的值[org.springframework.web.multipart.commons.CommonsMultipartFile@3fc04a65];代码[jpg / png文件类型仅支持.image.file,jpg / png文件类型仅支持.file,jpg / png文件类型仅支持..org.springframework.web.multipart.MultipartFile,jpg / png文件类型只支持];参数[];默认消息[null]]&#34;

我无法理解为什么错误列表长度为4.我的动机是如果未经验证则在json中显示错误。

如果有任何标准方法可以进行此类验证吗?我是春天和休息的新手。所以有人请告诉我实现目标的方法。

1 个答案:

答案 0 :(得分:1)

 protected List<String> extractErrorMessages(BindingResult result) {
        List<String> errorMessages = new ArrayList<>();
        for (Object object : result.getAllErrors()) {
            if (object instanceof FieldError) {
                FieldError fieldError = (FieldError) object;
                errorMessages.add(fieldError.getCode());
            }
        }
        return errorMessages;
    }

看看fieldError方法,可能在你的情况下你应该使用getField