package com.common.util;

import org.springframework.beans.factory.annotation.Autowired;
import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.io.InputStream;
import java.util.function.Function;
import org.springframework.stereotype.Component;
/**
 * MyBatis 工具类
 */


@Component
public class MyBatisUtil {
    private static final Logger logger = LoggerFactory.getLogger(MyBatisUtil.class);
    private static SqlSessionFactory sqlSessionFactory;
    @Autowired
    private SqlSessionFactory autoWiredSqlSessionFactory;
    static {
        init();
    }
    private static void init() {
        try {
            String resource = "mybatis-config.xml";
            InputStream inputStream = Resources.getResourceAsStream(resource);
            sqlSessionFactory = new SqlSessionFactoryBuilder().build(inputStream);
            logger.info("MyBatis 初始化成功");
        } catch (IOException e) {
            logger.error("初始化MyBatis失败", e);
            throw new RuntimeException("初始化MyBatis失败", e);
        }
    }
    public static SqlSession getSqlSession() {
        return sqlSessionFactory.openSession();
    }

    public static SqlSessionFactory getSqlSessionFactory() {
        return sqlSessionFactory;
    }

    /**
     * 获取 SqlSession（自动提交事务）
     */
    public static SqlSession getSqlSessionAutoCommit() {
        return sqlSessionFactory.openSession(true);
    }

    /**
     * 执行查询操作（自动管理资源）
     */
    public static <T> T executeQuery(Function<SqlSession, T> function) {
        SqlSession sqlSession = getSqlSession();
        try {
            return function.apply(sqlSession);
        } finally {
            sqlSession.close();
        }
    }

    /**
     * 执行更新操作（自动提交事务和管理资源）
     */
    public static <T> T executeUpdate(Function<SqlSession, T> function) {
        SqlSession sqlSession = getSqlSession();
        try {
            T result = function.apply(sqlSession);
            sqlSession.commit();
            return result;
        } catch (Exception e) {
            sqlSession.rollback();
            throw new RuntimeException("数据库操作失败", e);
        } finally {
            sqlSession.close();
        }
    }
    /**
     * 重新初始化（用于配置热更新）
     */
    public static void reload() {
        init();
    }
}