bank

女爷i 2022-03-18 05:16 167阅读 0赞

import javax.servlet.ServletOutputStream;
    import javax.servlet.WriteListener;
    import javax.servlet.http.HttpServletResponse;
    import javax.servlet.http.HttpServletResponseWrapper;
    import java.io.ByteArrayOutputStream;
    import java.io.IOException;
    
    /**
     * 返回值输出代理类
     *
     * @author kokJuis
     * @Title: ResponseWrapper
     * @Description:
     * @date 上午9:52:11
     */
    public class ResponseWrapper extends HttpServletResponseWrapper {
    
        private ByteArrayOutputStream buffer;
    
        private ServletOutputStream out;
    
        public ResponseWrapper(HttpServletResponse httpServletResponse) {
            super(httpServletResponse);
            buffer = new ByteArrayOutputStream();
            out = new WrapperOutputStream(buffer);
        }
    
        @Override
        public ServletOutputStream getOutputStream()
                throws IOException {
            return out;
        }
    
        @Override
        public void flushBuffer()
                throws IOException {
            if (out != null) {
                out.flush();
            }
        }
    
        public byte[] getContent()
                throws IOException {
            flushBuffer();
            return buffer.toByteArray();
        }
    
        class WrapperOutputStream extends ServletOutputStream {
            private ByteArrayOutputStream bos;
    
            public WrapperOutputStream(ByteArrayOutputStream bos) {
                this.bos = bos;
            }
    
            @Override
            public void write(int b)
                    throws IOException {
                bos.write(b);
            }
    
            @Override
            public boolean isReady() {
    
                // TODO Auto-generated method stub
                return false;
    
            }
    
            @Override
            public void setWriteListener(WriteListener arg0) {
    
                // TODO Auto-generated method stub
    
            }
        }
    }

import org.springframework.context.annotation.Configuration;
    import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
    import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
    import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
    
    @Configuration
    public class SessionConfiguraction implements WebMvcConfigurer {
        //配置拦截器
        public void addInterceptors(InterceptorRegistry registry){
            //registry.addInterceptor此方法添加拦截器
            registry.addInterceptor(new SessionInterceptor()).addPathPatterns("/**")
                    .excludePathPatterns("/static/**","/user/login/**","/login/**","/welcome/**","/error/**","/lib/layui/**");//需要配置2：----------- 告知拦截器：/static/admin/** 与 /static/user/** 不需要拦截 （配置的是 路径）
        }
    
        /**
         * 添加静态资源文件，外部可以直接访问地址
         * @param registry
         */
        @Override
        public void addResourceHandlers(ResourceHandlerRegistry registry) {
            //需要配置1：----------- 需要告知系统，这是要被当成静态文件的！
            //第一个方法设置访问路径前缀，第二个方法设置资源路径
            registry.addResourceHandler("/static/**").addResourceLocations("classpath:/static/");
        }
    
    }

import org.springframework.web.servlet.HandlerInterceptor;
    
    import javax.servlet.http.HttpServletRequest;
    import javax.servlet.http.HttpServletResponse;
    
    public class SessionInterceptor implements HandlerInterceptor {
        public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object o) throws  Exception{
            String url = request.getRequestURI();
            if (url.endsWith("/xmlRequest")){
                return true;
            }
            Object obj = request.getSession().getAttribute("session_user");
            if (request.getRequestURI().equals("/login")){
                return true;
            }
            if (obj == null){
                response.sendRedirect("/login");
                return  false;
            }
            return true;
        }
    }