首页 > 随笔 > SpringBoot代码集

SpringBoot代码集

简书 2026-09-08 04:03 2 阅读 查看原文
SpringBoot BeanFactory 容器对象

一、获取Spring容器对象

1.1 实现BeanFactoryAware接口

实现BeanFactoryAware接口，然后重写setBeanFactory方法，就能从该方法中获取到Spring容器对象。

@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"); } }
1.2 实现ApplicationContextAware接口

实现ApplicationContextAware接口，然后重写setApplicationContext方法，也能从该方法中获取到Spring容器对象。

@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"); } }
1.3 实现ApplicationListener接口

实现ApplicationListener接口，需要注意的是该接口接收的泛型是ContextRefreshedEvent类，然后重写onApplicationEvent方法，也能从该方法中获取到Spring容器对象。

@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"); } }

二、初始化bean

Spring中支持3种初始化bean的方法：

  • xml中指定init-method方法

  • 使用@PostConstruct注解

  • 实现InitializingBean接口

第一种方法太古老了，现在用的人不多，具体用法就不介绍了。

2.1 使用@PostConstruct注解

在需要初始化的方法上增加@PostConstruct注解，这样就有初始化的能力。

@Service public class AService { @PostConstruct public void init() { System.out.println("===初始化==="); } } 
@Component public class AlipayUtils { @Resource private AlipayConfigIOS configIOS; @Resource private AlipayConfigAndroid configAndroid; public AlipayClient alipayClientIOS; public AlipayClient alipayClientAndroid; @PostConstruct public void init(){ System.out.println("===初始化==="); //构建IOS alipayClientIOS = new DefaultAlipayClient( configIOS.getGateWay(), configIOS.getAppId(), configIOS.getAppPrivateKey(), configIOS.getFormat(), configIOS.getCharset(), configIOS.getAliPayPublicKey(), configIOS.getSignType()); //构建Android alipayClientAndroid = new DefaultAlipayClient( configAndroid.getGateWay(), configAndroid.getAppId(), configAndroid.getAppPrivateKey(), configAndroid.getFormat(), configAndroid.getCharset(), configAndroid.getAliPayPublicKey(), configAndroid.getSignType()); } }
2.2 实现InitializingBean接口

实现InitializingBean接口，重写afterPropertiesSet方法，该方法中可以完成初始化功能。

@Service public class BService implements InitializingBean { @Override public void afterPropertiesSet() throws Exception { System.out.println("===初始化==="); } }

三、自定义自己的Scope

我们都知道Spring默认支持的Scope只有两种：

  • singleton 单例，每次从spring容器中获取到的bean都是同一个对象。
  • prototype 多例，每次从spring容器中获取到的bean都是不同的对象。

Spring web又对Scope进行了扩展，增加了：

  • RequestScope 同一次请求从spring容器中获取到的bean都是同一个对象。
  • SessionScope 同一个会话从spring容器中获取到的bean都是同一个对象。

即便如此，有些场景还是无法满足我们的要求。比如，我们想在同一个线程中从spring容器获取到的bean都是同一个对象，该怎么办？这就需要自定义Scope了。

3.1 第一步实现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; } }
3.2 第二步将新定义的Scope注入到Spring容器中
@Component public class ThreadLocalBeanFactoryPostProcessor implements BeanFactoryPostProcessor { @Override public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException { beanFactory.registerScope("threadLocalScope", new ThreadLocalScope()); } }
3.3 第三步使用新定义的Scope
@Scope("threadLocalScope") @Service public class CService { public void add() { } }

四、自定义类型转换

Spring目前支持3种类型转换器：

  • Converter<S,T>：将 S 类型对象转为 T 类型对象

  • ConverterFactory<S, R>：将 S 类型对象转为 R 类型及子类对象

  • GenericConverter：它支持多个source和目标类型的转化，同时还提供了source和目标类型的上下文，这个上下文能让你实现基于属性上的注解或信息来进行类型转换。

这3种类型转换器使用的场景不一样，我们以Converter<S,T>为例。假如：接口中接收参数的实体对象中，有个字段的类型是Date，但是实际传参的是字符串类型：2021-01-03 10:20:15，要如何处理呢？

4.1 第一步，定义一个实体User
@Data public class User { private Long id; private String name; private Date registerDate; }
4.2 第二步，实现Converter接口
public class DateConverter implements Converter<String, Date> { private static final String dateFormat = "yyyy-MM-dd HH:mm:ss"; private static final String shortDateFormat = "yyyy-MM-dd"; @Override public Date convert(String source) { if(StringUtils.isEmpty(value)) { return null; } value = value.trim(); try { if(value.contains("-")) { SimpleDateFormat formatter; if(value.contains(":")) { formatter = new SimpleDateFormat(dateFormat); }else { formatter = new SimpleDateFormat(shortDateFormat); } Date dtDate = formatter.parse(value); return dtDate; }else if(value.matches("^\\d+$")) { Long lDate = new Long(value); return new Date(lDate); } } catch (Exception e) { throw new RuntimeException(String.format("parser %s to Date fail", value)); } throw new RuntimeException(String.format("parser %s to Date fail", value)); } }
4.3 第三步，将新定义的类型转换器注入到Spring容器中
@Configuration public class WebConfig extends WebMvcConfigurerAdapter { @Override public void addFormatters(FormatterRegistry registry) { registry.addConverter(new DateConverter()); } }
4.4 第四步，调用接口
@RequestMapping("/user") @RestController public class UserController { @RequestMapping("/save") public String save(@RequestBody User user) { return "success"; } }

请求接口时User对象中registerDate字段会被自动转换成Date类型。

五、Enable开关

不知道你有没有用过Enable开头的注解，比如：EnableAsync、EnableCaching、EnableAspectJAutoProxy等，这类注解就像开关一样，只要在@Configuration定义的配置类上加上这类注解，就能开启相关的功能，让我们一起实现一个自己的开关。

5.1 第一步，定义一个LogFilter
public class LogFilter implements Filter { @Override public void init(FilterConfig filterConfig) throws ServletException { } @Override public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException { System.out.println("记录请求日志"); chain.doFilter(request, response); System.out.println("记录响应日志"); } @Override public void destroy() { } }
5.2 第二步，注册LogFilter
@ConditionalOnWebApplication public class LogFilterWebConfig { @Bean public LogFilter timeFilter() { return new LogFilter(); } }

注意，这里用了@ConditionalOnWebApplication注解，没有直接使用@Configuration注解。

5.3 第三步，定义开关@EnableLog注解
@Target(ElementType.TYPE) @Retention(RetentionPolicy.RUNTIME) @Documented @Import(LogFilterWebConfig.class) public @interface EnableLog { }
5.4 第四步，启动类加上@EnableLog注解

只需在Springboot启动类加上@EnableLog注解即可开启LogFilter记录请求和响应日志的功能。

转载自：Spring 那些让你爱不释手的代码技巧

相关推荐