spring boot拦截器

package me.ocheng.common.config;

import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.databind.DeserializationFeature;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.dataformat.xml.XmlMapper;
import me.ocheng.common.converter.AbstractHttpMessageConverter;
import me.ocheng.common.core.ApplicationContext;
import me.ocheng.common.interceptor.AbstractInterceptor;
import me.ocheng.common.resolvers.AbstractMethodArgumentResolver;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.http.HttpMessageConverters;
import org.springframework.context.annotation.Configuration;
import org.springframework.http.MediaType;
import org.springframework.http.converter.*;
import org.springframework.http.converter.json.MappingJackson2HttpMessageConverter;
import org.springframework.http.converter.support.AllEncompassingFormHttpMessageConverter;
import org.springframework.http.converter.xml.MappingJackson2XmlHttpMessageConverter;
import org.springframework.http.converter.xml.SourceHttpMessageConverter;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.servlet.config.annotation.*;

import java.util.Iterator;
import java.util.List;
import java.util.Map;

/**
 
 */
@Configuration
@EnableWebMvc
public class WebMvcConfig implements WebMvcConfigurer {

    @Autowired
    private HttpMessageConverters messageConverters;

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        Map<String, AbstractInterceptor> interceptors = ApplicationContext.getInstance().getBeansOfType(AbstractInterceptor.class);

        Iterator<AbstractInterceptor> iterable = interceptors.values().iterator();

        while (iterable.hasNext()) {
            registry.addInterceptor(iterable.next());
        }
    }

    @Override
    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {

        Map<String, AbstractMethodArgumentResolver> interceptors = ApplicationContext.getInstance().getBeansOfType(AbstractMethodArgumentResolver.class);

        Iterator<AbstractMethodArgumentResolver> iterable = interceptors.values().iterator();

        while (iterable.hasNext()) {
            argumentResolvers.add(iterable.next());
        }

    }

    @Override
    public void configureMessageConverters(List<HttpMessageConverter<?>> converters) {
//        converters.addAll(messageConverters.getConverters());

        converters.add(new ByteArrayHttpMessageConverter());
        converters.add(new StringHttpMessageConverter());
        converters.add(new ResourceHttpMessageConverter());
        converters.add(new ResourceRegionHttpMessageConverter());
        converters.add(new SourceHttpMessageConverter<>());
        converters.add(new AllEncompassingFormHttpMessageConverter());

        ObjectMapper objectMapper = new ObjectMapper();
        objectMapper.disable(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES);
        objectMapper.configure(JsonParser.Feature.ALLOW_UNQUOTED_CONTROL_CHARS, true);
        converters.add(new MappingJackson2HttpMessageConverter(objectMapper));

        XmlMapper xmlMapper = new XmlMapper();
        xmlMapper.disable(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES);
        xmlMapper.configure(JsonParser.Feature.ALLOW_UNQUOTED_CONTROL_CHARS, true);
        converters.add(new MappingJackson2XmlHttpMessageConverter(xmlMapper));

        Map<String, AbstractHttpMessageConverter> interceptors = ApplicationContext.getInstance().getBeansOfType(AbstractHttpMessageConverter.class);

        Iterator<AbstractHttpMessageConverter> iterable = interceptors.values().iterator();

        while (iterable.hasNext()) {
            converters.add((HttpMessageConverter<?>) iterable.next());
        }
    }

    @Override
    public void configureContentNegotiation(ContentNegotiationConfigurer configurer) {
        configurer.defaultContentType(MediaType.APPLICATION_JSON);
    }

    @Override
    public void addCorsMappings(CorsRegistry registry) {
        registry.addMapping("/**")
                .allowedOrigins("*")
                .allowedMethods("GET", "POST", "PUT", "DELETE", "OPTIONS")
                .allowedHeaders("*")
                .exposedHeaders("X-Security-Token", "X-Security-Token-Renewal", "Proxy-Connection", "Accept", "Cache-Control", "Content-Type", "Content-Range", "DNT", "If-Modified-Since", "Keep-Alive", "Origin", "User-Agent", "X-Requested-With")
                .maxAge(3600);
    }
}

package me.ocheng.common.converter;

import org.springframework.http.HttpInputMessage;
import org.springframework.http.HttpOutputMessage;
import org.springframework.http.MediaType;
import org.springframework.http.converter.HttpMessageConverter;
import org.springframework.http.converter.HttpMessageNotReadableException;
import org.springframework.http.converter.HttpMessageNotWritableException;

import java.io.IOException;
import java.util.List;

/**
  
 * @date 2018-06-22
 */
public abstract class AbstractHttpMessageConverter<T> implements HttpMessageConverter<T> {

    @Override
    public abstract boolean canRead(Class<?> clazz, MediaType mediaType);

    @Override
    public abstract boolean canWrite(Class<?> clazz, MediaType mediaType);

    @Override
    public abstract List<MediaType> getSupportedMediaTypes();

    @Override
    public abstract T read(Class<? extends T> clazz, HttpInputMessage inputMessage) throws IOException, HttpMessageNotReadableException;

    @Override
    public abstract void write(T t, MediaType contentType, HttpOutputMessage outputMessage) throws IOException, HttpMessageNotWritableException;
}

package me.ocheng.common.core;

import org.springframework.core.env.Environment;

import java.util.Map;

/**
 * @author Eric
 * @date 2017/2/22
 */
public class ApplicationContext {

    private final static ApplicationContext INSTANCE = new ApplicationContext();

    private org.springframework.context.ApplicationContext context;
    private Environment environment;

    private ApplicationContext() {
    }

    public static ApplicationContext getInstance() {
        return INSTANCE;
    }

    public org.springframework.context.ApplicationContext getContext() {
        return context;
    }

    public void setContext(org.springframework.context.ApplicationContext context) {
        this.context = context;
    }

    public Environment getEnvironment() {
        return environment;
    }

    public void setEnvironment(Environment environment) {
        this.environment = environment;
    }

    public <T> T getBean(String name) {
        if (context == null) {
            return null;
        }

        Object bean = context.getBean(name);

        if (bean == null) {
            return null;
        }

        return (T) bean;
    }

    public <T> T getBean(Class type) {
        if (context == null) {
            return null;
        }

        Object bean = context.getBean(type);

        if (bean == null) {
            return null;
        }

        return (T) bean;
    }

    public <T> Map<String, T> getBeansOfType(Class<T> clazz) {
        return context.getBeansOfType(clazz);
    }
}

package me.ocheng.common.interceptor;

import org.springframework.lang.Nullable;
import org.springframework.web.servlet.ModelAndView;
import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

public abstract class AbstractInterceptor extends HandlerInterceptorAdapter {

    public abstract boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
            throws Exception;

    public abstract void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler,
                                    @Nullable ModelAndView modelAndView) throws Exception;

    public abstract void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler,
                                         @Nullable Exception ex) throws Exception;

    public abstract void afterConcurrentHandlingStarted(HttpServletRequest request, HttpServletResponse response,
                                                        Object handler) throws Exception;

}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值