89 lines
2.6 KiB
Java
89 lines
2.6 KiB
Java
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();
|
||
}
|
||
} |