4dkankan-manage-jp/src/main/java/com/fdkankan/manage_jp/controller/BaseController.java

package com.fdkankan.manage_jp.controller;

import com.fdkankan.common.util.DateEditor;
import com.fdkankan.common.util.JwtUtil;
import com.fdkankan.manage_jp.common.ResultCode;
import com.fdkankan.manage_jp.entity.User;
import com.fdkankan.manage_jp.exception.BusinessException;
import com.fdkankan.manage_jp.service.IUserService;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.propertyeditors.StringTrimmerEditor;
import org.springframework.web.bind.WebDataBinder;
import org.springframework.web.bind.annotation.InitBinder;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.Date;

public class BaseController {

    @Autowired
    protected HttpServletRequest request;

    @Autowired
    protected HttpServletResponse response;

    @Autowired
    IUserService userService;

    @InitBinder
    protected void initBinder(WebDataBinder webDataBinder) {
        webDataBinder.registerCustomEditor(String.class, new StringTrimmerEditor(true));
        webDataBinder.registerCustomEditor(Date.class, new DateEditor(true));
    }



    public static void output(HttpServletResponse resp, File file) {
        OutputStream os = null;
        BufferedInputStream bis = null;
        byte[] buff = new byte[1024];
        try {
            os = resp.getOutputStream();
            bis = new BufferedInputStream(new FileInputStream(file));
            int i = 0;
            while ((i = bis.read(buff)) != -1) {
                os.write(buff, 0, i);
                os.flush();
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            try {
                bis.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

    protected String getToken(){
        return request.getHeader("token");
    }
    protected User getUser(){
        String username = JwtUtil.getUsername(getToken());
        if(StringUtils.isBlank(username)){
            throw new BusinessException(ResultCode.USER_NOT_LOGIN);
        }
        User user = userService.getByUserName(username);
        if(user == null){
            throw new BusinessException(ResultCode.USER_NOT_LOGIN);
        }
        return user;
    }

}