package com.example.demo.config;

import com.zaxxer.hikari.HikariDataSource;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.session.SqlSessionFactory;
import org.mybatis.spring.SqlSessionFactoryBean;
import org.mybatis.spring.SqlSessionTemplate;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.boot.jdbc.DataSourceBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.support.PathMatchingResourcePatternResolver;
import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.transaction.PlatformTransactionManager;

import javax.sql.DataSource;

@Slf4j
@Configuration
public class Mysql1DataSourceConfig {

@Bean(name = "mysql1DataSource")
@ConfigurationProperties(prefix = "spring.datasource.mysql1")
public DataSource mysql1DataSource() {

    // 1. 先用 builder 创建出来
    HikariDataSource ds = DataSourceBuilder
            .create()
            .type(HikariDataSource.class)
            .build();          // 这里已经把 yml 里的 url/username/password 装好了

    // 2. 再补连接初始化 SQL（每次拿到连接自动执行）
    ds.setConnectionInitSql(
            "SET SESSION profiling=0,long_query_time=10,sql_mode='TRADITIONAL';");

    // 3. 如有需要还可继续补 ds.setMaximumPoolSize() 等
    return ds;
    // return DataSourceBuilder.create().type(HikariDataSource.class).build();
}
    @Bean(name = "mysqlTransactionManager")
    public PlatformTransactionManager mysqlTransactionManager(@Qualifier("mysql1DataSource") DataSource dataSource) {
        return new DataSourceTransactionManager(dataSource);
    }
    @Bean(name = "mysql1SqlSessionFactory")
//    @Primary
    public SqlSessionFactory mysql1SqlSessionFactory(@Qualifier("mysql1DataSource") DataSource dataSource,
                                                     @Qualifier("globalConfiguration1") org.apache.ibatis.session.Configuration globalConfiguration1) throws Exception {
        SqlSessionFactoryBean sessionFactory = new SqlSessionFactoryBean();
        sessionFactory.setDataSource(dataSource);
        sessionFactory.setMapperLocations(new PathMatchingResourcePatternResolver().getResources("classpath:{mapper,cashMapper}/*.xml"));
        sessionFactory.setConfiguration(globalConfiguration1);
        return sessionFactory.getObject();
    }

    @Bean(name = "mysql1SqlSessionTemplate")
//    @Primary
    public SqlSessionTemplate mysql1SqlSessionTemplate(@Qualifier("mysql1SqlSessionFactory") SqlSessionFactory sqlSessionFactory) {
        return new SqlSessionTemplate(sqlSessionFactory);
    }
    @Bean
    @ConfigurationProperties(prefix = "mybatis.configuration.mysql1")
    public org.apache.ibatis.session.Configuration globalConfiguration1() {
        return new org.apache.ibatis.session.Configuration();
    }
}