package com.example.messagequeue.config;

import com.github.benmanes.caffeine.cache.Caffeine;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
import org.springframework.cache.CacheManager;
import org.springframework.cache.annotation.EnableCaching;
import org.springframework.cache.caffeine.CaffeineCacheManager;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import org.springframework.transaction.annotation.EnableTransactionManagement;

import javax.sql.DataSource;
import java.util.concurrent.Executor;
import java.util.concurrent.ThreadPoolExecutor;//

/**
 * 高性能配置类
 * 配置缓存、线程池和事务管理
 */
@Configuration
@EnableCaching
@EnableAsync
@EnableTransactionManagement
public class HighPerformanceConfig {

 /**
 * 配置Caffeine缓存管理器
 */
 @Bean
 public CacheManager cacheManager() {
 CaffeineCacheManager cacheManager = new CaffeineCacheManager();
 cacheManager.setCaffeine(Caffeine.newBuilder()
 // 初始容量
 .initialCapacity(100)
 // 最大容量
 .maximumSize(1000)
 // 写入后过期时间(5分钟)
 .expireAfterWrite(java.time.Duration.ofMinutes(5))
 // 记录命中率等统计信息
 .recordStats());
 return cacheManager;
 }

 /**
 * 配置异步任务执行器
 */
 @Bean(name = "messageTaskExecutor")
 public Executor messageTaskExecutor() {
 ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
 // 核心线程数
 executor.setCorePoolSize(Runtime.getRuntime().availableProcessors());
 // 最大线程数
 executor.setMaxPoolSize(Runtime.getRuntime().availableProcessors() * 2);
 // 队列容量
 executor.setQueueCapacity(500);
 // 线程名前缀
 executor.setThreadNamePrefix("MessageExecutor-");
 // 拒绝策略:调用者运行
 executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
 // 线程空闲时间
 executor.setKeepAliveSeconds(60);
 // 允许核心线程超时
 executor.setAllowCoreThreadTimeOut(true);
 // 等待所有任务完成后再关闭线程池
 executor.setWaitForTasksToCompleteOnShutdown(true);
 // 等待时间
 executor.setAwaitTerminationSeconds(60);
 executor.initialize();
 return executor;
 }

 /**
 * 配置数据库连接池
 */
 @Bean
 public HikariConfig hikariConfig() {
 HikariConfig config = new HikariConfig();
 // 连接池大小
 config.setMaximumPoolSize(Runtime.getRuntime().availableProcessors() * 2);
 // 最小空闲连接数
 config.setMinimumIdle(2);
 // 连接超时时间
 config.setConnectionTimeout(30000);
 // 空闲连接超时时间
 config.setIdleTimeout(600000);
 // 连接最大生命周期
 config.setMaxLifetime(1800000);
 // 连接测试查询
 config.setConnectionTestQuery("SELECT 1");
 // 数据库连接URL
 config.setJdbcUrl("jdbc:h2:mem:messagequeue;DB_CLOSE_DELAY=-1;DB_CLOSE_ON_EXIT=FALSE");
 // 数据库用户名
 config.setUsername("sa");
 // 数据库密码
 config.setPassword("");
 // 驱动类名
 config.setDriverClassName("org.h2.Driver");
 // 连接池名称
 config.setPoolName("MessageQueuePool");
 return config;
 }

 @Bean
 public DataSource dataSource() {
 return new HikariDataSource(hikariConfig());
 }
}