package com.doumee.core.servlet;

import lombok.extern.slf4j.Slf4j;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;
import java.io.IOException;

/**
 * 增加响应流副本
 * @author Eva.Caesar Liu
 * @since 2025/03/31 16:44
 */
@Slf4j
public class ServletDuplicateResponseWrapper extends HttpServletResponseWrapper {

/*
    private ServletDuplicateOutputStream servletDuplicateOutputStream;

    public ServletDuplicateResponseWrapper(HttpServletResponse httpServletResponse) {
        super(httpServletResponse);
        try {
            servletDuplicateOutputStream = new ServletDuplicateOutputStream(httpServletResponse.getOutputStream());
        } catch (IOException e) {
            log.error("EVA: build ServletDuplicateResponseWrapper throw an exception", e);
        }
    }

    @Override
    public ServletOutputStream getOutputStream() {
        if (servletDuplicateOutputStream == null) {
            servletDuplicateOutputStream = new ServletDuplicateOutputStream(super.getOutputStream());
        }
        return servletDuplicateOutputStream;
    }
*/

    private ServletDuplicateOutputStream servletDuplicateOutputStream;

    public ServletDuplicateResponseWrapper(HttpServletResponse httpServletResponse) {
        super(httpServletResponse);
    }

    @Override
    public ServletOutputStream getOutputStream() throws IOException {
        if (servletDuplicateOutputStream == null) {
            servletDuplicateOutputStream = new ServletDuplicateOutputStream(super.getOutputStream());
        }
        return servletDuplicateOutputStream;
    }

}