监控Spring Bean初始化时间的实现方式

package com.ruoyi.framework;

import lombok.extern.slf4j.Slf4j;
import org.jetbrains.annotations.NotNull;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.Map;

@Component
@Slf4j
public class BeanInitializationTimeLogger implements BeanPostProcessor {
    private final Map<String, Long> startTimes = new HashMap<>();

    @Override
    public Object postProcessBeforeInitialization(@NotNull Object bean, @NotNull String beanName) throws BeansException {
        // 记录 Bean 初始化开始时间
        startTimes.put(beanName, System.currentTimeMillis());
        return bean;
    }

    @Override
    public Object postProcessAfterInitialization(@NotNull Object bean, @NotNull String beanName) throws BeansException {
        // 计算 Bean 初始化耗时
        Long startTime = startTimes.get(beanName);
        if (startTime != null) {
            long duration = System.currentTimeMillis() - startTime;
            if (duration > 1000) { // 假设耗时超过1秒的 Bean 需要记录
                log.info("Bean :{} took :{} ms to initialize.", beanName, duration);
            }
        }
        return bean;
    }
}