项目实训(4) - 用户头像

---

前言

用户头像的上传与下载。

配置信息

file:
  # 文件保存地址
  path: "/www/meeting"

picture:
  # 文件格式，目前只允许jpg文件
  pattern: .jpg

1、PictureUtil.java

@Component
public class PictureUtil {

    @Value("${file.path}")
    public String path;

    @Value("${picture.pattern}")
    private String filePattern;

    /**
     * 文件后缀名，.jpg
     */
    @Autowired
    private UUIDUtil uuidUtil;

    /**
     * 文件地址 /file/pic/{图片类型}/{uid}.jpg
     * 处理图片
     * @param type 类型
     * @param file 图片
     */
    public void handlePicture(String type, MultipartFile file, long uid)
            throws IOException{
        String originalName = file.getOriginalFilename();
        if (originalName == null) {
            throw new IllegalStateException("文件名不能为空");
        }
        if (!originalName.endsWith(filePattern)) {
            throw new FileFormatException("不支持的文件格式");
        }
        // 新的文件名
        String fileName = "" + uid + filePattern;
        // 文件目的地址
        File dest = new File(path + '/' + type + '/' + fileName);
        if(!dest.getParentFile().exists()){
            dest.getParentFile().mkdir();
        }
        file.transferTo(dest);
    }

    public byte[] openPicture(String filename, String type)
            throws FileNotFoundException {
        File source = new File(path + '/' + type + '/' + filename);
        if (source.exists() && source.isFile()) {
            InputStream inputStream = null;
            ByteArrayOutputStream outputStream = null;
            try {
                inputStream = new FileInputStream(source);
                outputStream = new ByteArrayOutputStream();
                byte[] buff = new byte[1024];
                int read = -1;
                while ((read = inputStream.read(buff)) != -1) {
                    outputStream.write(buff, 0, read);
                }
                if (outputStream.size() == 0) {
                    throw new FileNotFoundException("空文件");
                }
                return outputStream.toByteArray();
            } catch (IOException exception) {
                return null;
            } finally {
                handleClose(inputStream, outputStream);
            }
        } else {
            throw new FileNotFoundException("文件不存在");
        }
    }

    private void handleClose(InputStream inputStream, OutputStream outputStream) {
        try {
            if (inputStream != null) {
                inputStream.close();
            }
            if (outputStream != null) {
                outputStream.close();
            }
        } catch (IOException exception) {
            exception.printStackTrace();
        }
    }

}

2、其它

前几天重新写了User类，其中profile字段变为boolean类型，表示是否采用默认头像。
在文件的上传与下载中，除了用到了PictureUtil 类，其它相较于之前的功能没有什么区别，这里就不做展示了。
然后就是在main函数所在的类里面， 添加了MultipartConfigElement的bean，用于限制文件大小。根据网上的说法，这个bean要配置在main函数所在的类里面，具体原因我不太清楚。

3、全局异常

增加了全局异常处理，留一下记录，将来使用。

@ControllerAdvice
public class GlobalExceptionHandler {

    /**
     * 捕获MissingServletRequestParameterException异常，400
     * @param exception MissingServletRequestParameterException
     * @return 400响应
     */
    @ResponseBody
    @ExceptionHandler(value = {MissingServletRequestParameterException.class})
    @ResponseStatus(HttpStatus.BAD_REQUEST)
    public Map<String, Object> handleException(MissingServletRequestParameterException exception) {
        Map<String, Object> map = new HashMap<>();
        map.put("code", 400);
        map.put("message", exception.getMessage());
        return map;
    }

    /**
     * 捕获MissingServletRequestParameterException异常，400
     * @param exception MissingServletRequestParameterException
     * @return 400响应
     */
    @ResponseBody
    @ExceptionHandler(value = {MissingRequestHeaderException.class})
    @ResponseStatus(HttpStatus.BAD_REQUEST)
    public Map<String, Object> handleException(MissingRequestHeaderException exception) {
        Map<String, Object> map = new HashMap<>();
        map.put("code", 400);
        map.put("message", exception.getMessage());
        return map;
    }

    /**
     * 捕获UnAuthorizedException异常，401
     * @param exception UnAuthorizedException
     * @return 401响应
     */
    @ResponseBody
    @ExceptionHandler(value = {UnAuthorizedException.class})
    @ResponseStatus(HttpStatus.UNAUTHORIZED)
    public Map<String, Object> handleException(UnAuthorizedException exception) {
        Map<String, Object> map = new HashMap<>();
        map.put("code", 401);
        map.put("message", exception.getMsg());
        return map;
    }

    /**
     * 捕获NoHandlerFoundException异常，405
     * @param exception NoHandlerFoundException
     * @return 404响应
     */
    @ResponseBody
    @ExceptionHandler(value = {NoHandlerFoundException.class})
    @ResponseStatus(HttpStatus.NOT_FOUND)
    public Map<String, Object> handleException(NoHandlerFoundException exception) {
        Map<String, Object> map = new HashMap<>();
        map.put("code", 404);
        map.put("message", exception.getMessage());
        return map;
    }

    /**
     * 捕获FileNotFoundException异常，404
     * @param exception FileNotFoundException
     * @return 404响应
     */
    @ResponseBody
    @ExceptionHandler(value = {FileNotFoundException.class})
    @ResponseStatus(HttpStatus.NOT_FOUND)
    public Map<String, Object> handleException(FileNotFoundException exception) {
        Map<String, Object> map = new HashMap<>();
        map.put("code", 404);
        map.put("message", exception.getMessage());
        return map;
    }

    /**
     * 捕获HttpRequestMethodNotSupportedException异常，405
     * @param exception HttpRequestMethodNotSupportedException
     * @return 405响应
     */
    @ResponseBody
    @ExceptionHandler(value = {HttpRequestMethodNotSupportedException.class})
    @ResponseStatus(HttpStatus.METHOD_NOT_ALLOWED)
    public Map<String, Object> handleException(HttpRequestMethodNotSupportedException exception) {
        Map<String, Object> map = new HashMap<>();
        map.put("code", 405);
        map.put("message", exception.getMessage());
        return map;
    }

    /**
     * 捕获MaxUploadSizeExceededException异常，文件大小超出上限
     * @param exception MaxUploadSizeExceededException
     * @return 500响应
     */
    @ResponseBody
    @ExceptionHandler(value = {MaxUploadSizeExceededException.class})
    public Map<String, Object> handleException(MaxUploadSizeExceededException exception) {
        Map<String, Object> map = new HashMap<>();
        map.put("code", 500);
        map.put("message", exception.getMessage());
        return map;
    }

    /**
     * 捕获其它异常
     * @param exception Exception
     * @return 500响应
     */
    @ResponseBody
    @ExceptionHandler(value = {Exception.class})
    public Map<String, Object> exceptionHandler(Exception exception) {
        System.out.println(exception.getMessage());
        System.out.println(exception.getClass());
        Map<String, Object> map = new HashMap<>();
        map.put("code", 500);
        map.put("message", "服务器出现异常");
        return map;
    }

}