Response 重写设置返回值

 
import cn.dolphin.core.exception.IoRuntimeException;
import cn.dolphin.core.result.ApiResult;
import cn.dolphin.core.util.Utf8Util;
import com.alibaba.fastjson.JSON;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import java.io.IOException;
import java.io.PrintWriter;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.lang3.StringUtils;
import org.springframework.http.HttpStatus;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

public class ResponseUtil {
    private static final String DOWN_TYPE = "applicatoin/octet-stream";
    private static final String DOWN_HEADER1 = "Content-Disposition";
    private static final String DOWN_HEADER2 = "attachment; filename=";
    private static final String EXCEL_TYPE = "application/vnd.ms-excel;charset=utf-8";
    public static final String CACHE_CONTROL = "Cache-Control";
    public static final String NO_CACHE = "no-cache";
    public static final String PRAGMA = "Pragma";
    public static final String EXPIRESS = "Expires";
    public static final String FORWARD = "forward:";
    public static final String REDIRECT = "redirect:";

    private ResponseUtil() {
    }

    public static HttpServletResponse getResponse() {
        return ((ServletRequestAttributes)((ServletRequestAttributes)RequestContextHolder.getRequestAttributes())).getResponse();
    }

    public static final void setDownloadHeader(HttpServletResponse response, String fileName) {
        response.reset();
        response.setContentType("applicatoin/octet-stream");
        response.setHeader("Content-Disposition", "attachment; filename=" + Utf8Util.utf2iso(fileName));
    }

    public static final void setNocachHeader(HttpServletResponse response) {
        response.reset();
        response.setHeader("Pragma", "no-cache");
        response.setHeader("Cache-Control", "no-cache");
        response.setDateHeader("Expires", 0L);
    }

    public static final void setExcelHeader(HttpServletResponse response, String fileName) {
        response.reset();
        response.setContentType("application/vnd.ms-excel;charset=utf-8");
        response.setHeader("Content-Disposition", "attachment; filename=" + Utf8Util.utf2iso(fileName));
    }

    public static final void response(String text) {
        response(getResponse(), text);
    }

    public static final void response(HttpServletResponse response, String text) {
        cors(response, (String)null);
        response.setStatus(HttpStatus.OK.value());
        response.setContentType("application/json");
        response.setCharacterEncoding("UTF-8");
        PrintWriter writer = null;

        try {
            writer = response.getWriter();
            writer.write(text);
        } catch (IOException var11) {
            throw new IoRuntimeException("发送JSON响应失败", var11);
        } finally {
            try {
                writer.close();
            } catch (Exception var10) {
                var10.printStackTrace();
            }

        }

    }

    public static final void responseText(String text) {
        responseText(getResponse(), text);
    }

    public static final void responseText(HttpServletResponse response, String text) {
        try {
            response.getWriter().write(text);
            response.getWriter().flush();
        } catch (IOException var10) {
            throw new IoRuntimeException("发送文字响应失败", var10);
        } finally {
            try {
                response.getWriter().close();
            } catch (Exception var9) {
                var9.printStackTrace();
            }

        }

    }

    public static final String forward(String url) {
        return "forward:" + url;
    }

    public static final void sendForward(HttpServletRequest request, HttpServletResponse response, String url) {
        try {
            request.getRequestDispatcher(url).forward(request, response);
        } catch (IOException | ServletException var4) {
            throw new IoRuntimeException("转发失败：url = " + url, var4);
        }
    }

    public static final void sendForward(String url) {
        sendForward(RequestUtil.getRequest(), getResponse(), url);
    }

    public static final String redirect(String url) {
        return "redirect:" + url;
    }

    public static final void sendRedirect(HttpServletResponse response, String url) {
        try {
            response.sendRedirect(url);
        } catch (IOException var3) {
            throw new IoRuntimeException("重定向失败：url = " + url, var3);
        }
    }

    public static final void sendRedirect(String url) {
        sendRedirect(getResponse(), url);
    }

    public static final void responseFile(byte[] bs, String fileName) {
        responseFile(bs, fileName, getResponse());
    }

    public static final void responseFile(byte[] bs, String fileName, HttpServletResponse response) {
        setDownloadHeader(response, fileName);
        ServletOutputStream output = null;

        try {
            output = response.getOutputStream();
            output.write(bs);
            output.flush();
        } catch (IOException var12) {
            throw new IoRuntimeException("发送文件响应失败", var12);
        } finally {
            try {
                output.close();
            } catch (Exception var11) {
                var11.printStackTrace();
            }

        }

    }

    public static final void cors(HttpServletResponse response, String origin) {
        if (StringUtils.isBlank(response.getHeader("Access-Control-Allow-Origin"))) {
            response.setHeader("Access-Control-Allow-Origin", origin == null ? "*" : origin);
        }

    }

    public static final void sse() {
        getResponse().setHeader("X-Accel-Buffering", "no");
    }

    public static void errorToken(HttpServletResponse httpResponse, String msg) throws IOException {
        ApiResult result = ApiResult.error(msg);
        String json = JSON.toJSONString(result);
        httpResponse.setCharacterEncoding("UTF-8");
        httpResponse.setContentType("application/json;charset=UTF-8");
        httpResponse.setHeader("Content-type", "application/json;charset=UTF-8");
        httpResponse.setHeader("Access-Control-Allow-Origin", "*");
        httpResponse.setStatus(401);
        httpResponse.getWriter().write(json);
    }

    public static void errorToken(HttpServletResponse httpResponse, String code, String msg) throws IOException {
        ApiResult result = ApiResult.error(code, msg);
        String json = JSON.toJSONString(result);
        httpResponse.setCharacterEncoding("UTF-8");
        httpResponse.setContentType("application/json;charset=UTF-8");
        httpResponse.setHeader("Content-type", "application/json;charset=UTF-8");
        httpResponse.setHeader("Access-Control-Allow-Origin", "*");
        httpResponse.setStatus(401);
        httpResponse.getWriter().write(json);
    }

    public static void errorAuth(HttpServletResponse httpResponse, String code, String msg) throws IOException {
        ApiResult result = ApiResult.error(code, msg);
        String json = JSON.toJSONString(result);
        httpResponse.setCharacterEncoding("UTF-8");
        httpResponse.setContentType("application/json;charset=UTF-8");
        httpResponse.setHeader("Content-type", "application/json;charset=UTF-8");
        httpResponse.setHeader("Access-Control-Allow-Origin", "*");
        httpResponse.setStatus(401);
        httpResponse.getWriter().write(json);
    }

    public static void printJSON(HttpServletResponse response, Object object) throws Exception {
        response.setCharacterEncoding("UTF-8");
        response.setContentType("application/json");
        PrintWriter printWriter = response.getWriter();
        printWriter.write(JSON.toJSONString(object));
        printWriter.flush();
        printWriter.close();
    }

    public static void responseData(HttpServletResponse response, Object resData) throws JsonProcessingException {
        ObjectMapper objectMapper = new ObjectMapper();
        String jsonString = objectMapper.writeValueAsString(resData);
        response.setCharacterEncoding("UTF-8");
        response.setContentType("application/json; charset=utf-8");
        PrintWriter out = null;

        try {
            out = response.getWriter();
            out.append(jsonString);
        } catch (IOException var9) {
            var9.printStackTrace();
        } finally {
            if (out != null) {
                out.close();
            }

        }

    }
}