一怠晴、自定義攔截器
Spring mvc攔截器和Spring攔截器相比,它里面能夠獲取HttpServletRequest和HttpServletResponse等web對象實(shí)例拜轨。
Spring mvc攔截器的頂層接口是:HandlerInterceptor,包含三個(gè)方法:
- preHandle 目標(biāo)方法執(zhí)行前執(zhí)行
- postHandle 目標(biāo)方法執(zhí)行后執(zhí)行
- afterCompletion 請求完成時(shí)執(zhí)行
為了方便,我們一般情況會用HandlerInterceptor接口的實(shí)現(xiàn)類HandlerInterceptorAdapter類。假如有權(quán)限認(rèn)證朝群、日志、統(tǒng)計(jì)的場景中符,可以使用該攔截器姜胖。
第一步,繼承HandlerInterceptorAdapter類定義攔截器:
public class AuthInterceptor extends HandlerInterceptorAdapter {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
throws Exception {
String requestUrl = request.getRequestURI();
if (checkAuth(requestUrl)) {
return true;
}
return false;
}
private boolean checkAuth(String requestUrl) {
System.out.println("===權(quán)限校驗(yàn)===");
return true;
}
}
第二步淀散,將該攔截器注冊到spring容器:
@Configuration
public class WebAuthConfig extends WebMvcConfigurerAdapter {
@Bean
public AuthInterceptor getAuthInterceptor() {
return new AuthInterceptor();
}
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(new AuthInterceptor());
}
}
第三步右莱,在請求接口時(shí)Spring mvc通過該攔截器,能夠自動攔截該接口档插,并且校驗(yàn)權(quán)限慢蜓。
二、獲取Spring容器對象
2.1 BeanFactoryAware接口
@Service
public class PersonService implements BeanFactoryAware {
private BeanFactory beanFactory;
@Override
public void setBeanFactory(BeanFactory beanFactory) throws BeansException {
this.beanFactory = beanFactory;
}
public void add() {
Person person = (Person) beanFactory.getBean("person");
}
}
實(shí)現(xiàn)BeanFactoryAware接口郭膛,然后重寫setBeanFactory方法晨抡,就能從該方法中獲取到spring容器對象。
2.2 ApplicationContextAware接口
@Service
public class PersonService2 implements ApplicationContextAware {
private ApplicationContext applicationContext;
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
this.applicationContext = applicationContext;
}
public void add() {
Person person = (Person) applicationContext.getBean("person");
}
}
實(shí)現(xiàn)ApplicationContextAware接口则剃,然后重寫setApplicationContext方法耘柱,也能從該方法中獲取到spring容器對象。
2.3 ApplicationListener接口
@Service
public class PersonService3 implements ApplicationListener<ContextRefreshedEvent> {
private ApplicationContext applicationContext;
@Override
public void onApplicationEvent(ContextRefreshedEvent event) {
applicationContext = event.getApplicationContext();
}
public void add() {
Person person = (Person) applicationContext.getBean("person");
}
}
三棍现、全局異常處理
@RestControllerAdvice
public class GlobalExceptionHandler {
@ExceptionHandler(Exception.class)
public String handleException(Exception e) {
if (e instanceof ArithmeticException) {
return "數(shù)據(jù)異常";
}
if (e instanceof Exception) {
return "服務(wù)器內(nèi)部異常";
}
retur nnull;
}
}
只需在handleException方法中處理異常情況调煎,業(yè)務(wù)接口中可以放心使用,不再需要捕獲異常己肮。
四士袄、類型轉(zhuǎn)換器
Spring目前支持3中類型轉(zhuǎn)換器:
Converter<S,T>:將 S 類型對象轉(zhuǎn)為 T 類型對象
ConverterFactory<S, R>:將 S 類型對象轉(zhuǎn)為 R 類型及子類對象
GenericConverter:它支持多個(gè)source和目標(biāo)類型的轉(zhuǎn)化,同時(shí)還提供了source和目標(biāo)類型的上下文朴肺,這個(gè)上下文能讓你實(shí)現(xiàn)基于屬性上的注解或信息來進(jìn)行類型轉(zhuǎn)換
五窖剑、導(dǎo)入配置
有時(shí)我們需要在某個(gè)配置類中引入另外一些類坚洽,被引入的類也加到Spring容器中戈稿。這時(shí)可以使用 @Import
注解完成這個(gè)功能。
5.1 普通類
這種引入方式是最簡單的讶舰,被引入的類會被實(shí)例化bean對象鞍盗。
public class A {
}
@Import(A.class)
@Configuration
public class TestConfiguration {
}
通過@Import
注解引入A類,Spring就能自動實(shí)例化A對象跳昼,然后在需要使用的地方通過@Autowired
注解注入即可:
@Autowired
private A a;
5.2 配置類
這種引入方式是最復(fù)雜的般甲,因?yàn)?code>@Configuration注解還支持多種組合注解,比如:
@Import
@ImportResource
@PropertySource
public class A {
}
public class B {
}
@Import(B.class)
@Configuration
public class AConfiguration {
@Bean
public A a() {
return new A();
}
}
@Import(AConfiguration.class)
@Configuration
public class TestConfiguration {
}
通過@Import
注解引入@Configuration
注解的配置類鹅颊,會把該配置類相關(guān)@Import
敷存、@ImportResource
、@PropertySource
等注解引入的類進(jìn)行遞歸,一次性全部引入锚烦。
5.3 ImportSelector
這種引入方式需要實(shí)現(xiàn)ImportSelector接口:
public class AImportSelector implements ImportSelector {
private static final String CLASS_NAME = "com.sue.cache.service.test13.A";
public String[] selectImports(AnnotationMetadata importingClassMetadata) {
return new String[]{CLASS_NAME};
}
}
@Import(AImportSelector.class)
@Configuration
public class TestConfiguration {
}
這種方式的好處是selectImports方法返回的是數(shù)組觅闽,意味著可以同時(shí)引入多個(gè)類,還是非常方便的涮俄。
5.4 ImportBeanDefinitionRegistrar
這種引入方式需要實(shí)現(xiàn)ImportBeanDefinitionRegistrar接口:
public class AImportBeanDefinitionRegistrar implements ImportBeanDefinitionRegistrar {
@Override
public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {
RootBeanDefinition rootBeanDefinition = new RootBeanDefinition(A.class);
registry.registerBeanDefinition("a", rootBeanDefinition);
}
}
@Import(AImportBeanDefinitionRegistrar.class)
@Configuration
public class TestConfiguration {
}
這種方式是最靈活的蛉拙,能在registerBeanDefinitions方法中獲取到BeanDefinitionRegistry容器注冊對象,可以手動控制BeanDefinition的創(chuàng)建和注冊彻亲。
六孕锄、項(xiàng)目啟動時(shí)
有時(shí)候我們需要在項(xiàng)目啟動時(shí)定制化一些附加功能,比如:加載一些系統(tǒng)參數(shù)苞尝、完成初始化畸肆、預(yù)熱本地緩存等,該怎么辦呢野来?好消息是SpringBoot提供了:CommandLineRunner恼除、ApplicationRunner這兩個(gè)接口幫助我們實(shí)現(xiàn)以上需求。
它們的用法還是挺簡單的曼氛,以ApplicationRunner接口為例:
@Component
public class TestRunner implements ApplicationRunner {
@Autowired
private LoadDataService loadDataService;
public void run(ApplicationArguments args) throws Exception {
loadDataService.load();
}
}
實(shí)現(xiàn)ApplicationRunner接口豁辉,重寫run方法,在該方法中實(shí)現(xiàn)自己定制化需求舀患。如果項(xiàng)目中有多個(gè)類實(shí)現(xiàn)了ApplicationRunner接口徽级,他們的執(zhí)行順序要怎么指定呢?答案是使用@Order(n)注解聊浅,n的值越小越先執(zhí)行餐抢。當(dāng)然也可以通過@Priority注解指定順序。
七低匙、修改BeanDefinition
Spring IOC在實(shí)例化Bean對象之前旷痕,需要先讀取Bean的相關(guān)屬性,保存到BeanDefinition對象中顽冶,然后通過BeanDefinition對象欺抗,實(shí)例化Bean對象。如果想修改BeanDefinition對象中的屬性强重,該怎么辦呢绞呈?我們可以實(shí)現(xiàn)BeanFactoryPostProcessor接口。
@Component
public class MyBeanFactoryPostProcessor implements BeanFactoryPostProcessor {
@Override
public void postProcessBeanFactory(ConfigurableListableBeanFactory configurableListableBeanFactory) throws BeansException {
DefaultListableBeanFactory defaultListableBeanFactory = (DefaultListableBeanFactory) configurableListableBeanFactory;
BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(User.class);
beanDefinitionBuilder.addPropertyValue("id", 123);
beanDefinitionBuilder.addPropertyValue("name", "蘇三說技術(shù)");
defaultListableBeanFactory.registerBeanDefinition("user", beanDefinitionBuilder.getBeanDefinition());
}
}
在postProcessBeanFactory方法中间景,可以獲取BeanDefinition的相關(guān)對象佃声,并且修改該對象的屬性。
八倘要、初始化Bean前后
有時(shí)圾亏,你想在初始化Bean前后,實(shí)現(xiàn)一些自己的邏輯。這時(shí)可以實(shí)現(xiàn):BeanPostProcessor接口志鹃。該接口目前有兩個(gè)方法:
- postProcessBeforeInitialization 該方法在初始化方法之前調(diào)用父晶。
- postProcessAfterInitialization 該方法在初始化方法之后調(diào)用。
例如:
@Component
public class MyBeanPostProcessor implements BeanPostProcessor {
@Override
public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
if (bean instanceof User) {
((User) bean).setUserName("AlanChen");
}
return bean;
}
}
如果Spring中存在User對象弄跌,則將它的userName設(shè)置成:AlanChen甲喝。
其實(shí),我們經(jīng)常使用的注解铛只,比如:@Autowired
埠胖、@Value
、@Resource
淳玩、@PostConstruct
等直撤,是通過AutowiredAnnotationBeanPostProcessor和CommonAnnotationBeanPostProcessor實(shí)現(xiàn)的。
九蜕着、初始化方法
目前Spring中使用比較多的初始化bean的方法有:
- 使用@PostConstruct注解
- 實(shí)現(xiàn)InitializingBean接口
9.1 使用@PostConstruct注解
@Service
public class AService {
@PostConstruct
public void init() {
System.out.println("===初始化===");
}
}
在需要初始化的方法上增加@PostConstruct注解谋竖,這樣就有初始化的能力。
9.2 實(shí)現(xiàn)InitializingBean接口
@Service
public class BService implements InitializingBean {
@Override
public void afterPropertiesSet() throws Exception {
System.out.println("===初始化===");
}
}
實(shí)現(xiàn)InitializingBean接口承匣,重寫afterPropertiesSet方法蓖乘,該方法中可以完成初始化功能。
十韧骗、關(guān)閉容器前
有時(shí)候嘉抒,我們需要在關(guān)閉Spring容器前,做一些額外的工作袍暴,比如:關(guān)閉資源文件等些侍。這時(shí)可以實(shí)現(xiàn)DisposableBean接口,并且重寫它的destroy方法:
@Service
public class DService implements InitializingBean, DisposableBean {
@Override
public void destroy() throws Exception {
System.out.println("DisposableBean destroy");
}
@Override
public void afterPropertiesSet() throws Exception {
System.out.println("InitializingBean afterPropertiesSet");
}
}
這樣Spring容器銷毀前政模,會調(diào)用該destroy方法岗宣,做一些額外的工作。通常情況下淋样,我們會同時(shí)實(shí)現(xiàn)InitializingBean和DisposableBean接口耗式,重寫初始化方法和銷毀方法。
十一习蓬、自定義作用域
我們都知道Spring默認(rèn)支持的Scope只有兩種:
- singleton 單例纽什,每次從Spring容器中獲取到的bean都是同一個(gè)對象措嵌。
- prototype 多例躲叼,每次從Spring容器中獲取到的bean都是不同的對象。
Spring web又對Scope進(jìn)行了擴(kuò)展企巢,增加了:
- RequestScope 同一次請求從Spring容器中獲取到的bean都是同一個(gè)對象枫慷。
- SessionScope 同一個(gè)會話從Spring容器中獲取到的bean都是同一個(gè)對象。
即便如此,有些場景還是無法滿足我們的要求或听。
比如探孝,我們想在同一個(gè)線程中從Spring容器獲取到的bean都是同一個(gè)對象,該怎么辦誉裆?這就需要自定義Scope了顿颅。
第一步實(shí)現(xiàn)Scope接口:
public class ThreadLocalScope implements Scope {
private static final ThreadLocal THREAD_LOCAL_SCOPE = new ThreadLocal();
@Override
public Object get(String name, ObjectFactory<?> objectFactory) {
Object value = THREAD_LOCAL_SCOPE.get();
if (value != null) {
return value;
}
Object object = objectFactory.getObject();
THREAD_LOCAL_SCOPE.set(object);
return object;
}
@Override
public Object remove(String name) {
THREAD_LOCAL_SCOPE.remove();
return null;
}
@Override
public void registerDestructionCallback(String name, Runnable callback) {
}
@Override
public Object resolveContextualObject(String key) {
return null;
}
@Override
public String getConversationId() {
return null;
}
}
第二步將新定義的Scope注入到Spring容器中:
@Component
public class ThreadLocalBeanFactoryPostProcessor implements BeanFactoryPostProcessor {
@Override
public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
beanFactory.registerScope("threadLocalScope", new ThreadLocalScope());
}
}
第三步使用新定義的Scope:
@Scope("threadLocalScope")
@Service
public class CService {
public void add() {
}
}
轉(zhuǎn)載自:蘇三說技術(shù)