Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.zhuang.data.mybatis.MyBatisDbAccessor Maven / Gradle / Ivy
package com.zhuang.data.mybatis;
import com.zhuang.data.DbAccessor;
import com.zhuang.data.enums.DbDialect;
import com.zhuang.data.exception.ExecuteSqlException;
import com.zhuang.data.exception.GetConnectionException;
import com.zhuang.data.model.PageInfo;
import com.zhuang.data.mybatis.model.PageQueryParameter;
import com.zhuang.data.mybatis.util.MappedStatementUtils;
import com.zhuang.data.mybatis.util.SqlSessionFactoryUtils;
import com.zhuang.data.orm.enums.SqlType;
import com.zhuang.data.util.DbDialectUtils;
import com.zhuang.data.util.EntityUtils;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
public class MyBatisDbAccessor extends DbAccessor {
private SqlSessionFactory sqlSessionFactory;
private SqlSession globalSqlSession;
public static final String PAGE_QUERY_PARAMETER_KEY = "PAGE_QUERY_PARAMETER_KEY";
public MyBatisDbAccessor(boolean autoCommit) {
this(SqlSessionFactoryUtils.getSqlSessionFactory(), autoCommit);
}
public MyBatisDbAccessor(SqlSessionFactory sqlSessionFactory, boolean autoCommit) {
this(sqlSessionFactory, (autoCommit ? null : sqlSessionFactory.openSession()), autoCommit);
}
public MyBatisDbAccessor(SqlSessionFactory sqlSessionFactory, SqlSession sqlSession, boolean autoCommit) {
this.sqlSessionFactory = sqlSessionFactory;
this.globalSqlSession = sqlSession;
super.autoCommit = autoCommit;
super.dbDialect = DbDialectUtils.getDbDialectByDataSource(sqlSessionFactory.getConfiguration().getEnvironment().getDataSource());
}
public SqlSessionFactory getSqlSessionFactory() {
return sqlSessionFactory;
}
public DbDialect getDbDialect() {
return dbDialect;
}
@Override
public Connection getConnection() {
try {
if (autoCommit) {
return sqlSessionFactory.getConfiguration().getEnvironment().getDataSource().getConnection();
} else {
return globalSqlSession.getConnection();
}
} catch (SQLException e) {
throw new GetConnectionException("MyBatisDbAccessor获取Connection失败！", e);
}
}
@Override
public String getConfigFile() {
return configFile;
}
@Override
public boolean getAutoCommit() {
return autoCommit;
}
@Override
public T queryEntity(String sql, Object parameter, Class entityType) {
T result;
if (autoCommit) {
SqlSession sqlSession = sqlSessionFactory.openSession();
try {
result = queryEntity(sql, parameter, sqlSession);
} finally {
sqlSession.close();
}
} else {
result = queryEntity(sql, parameter, globalSqlSession);
}
return result;
}
@Override
public List queryEntities(String sql, Object parameter, Class entityType) {
List result;
if (autoCommit) {
SqlSession sqlSession = sqlSessionFactory.openSession();
try {
result = queryEntities(sql, parameter, sqlSession);
} finally {
sqlSession.close();
}
} else {
result = queryEntities(sql, parameter, globalSqlSession);
}
return result;
}
@Override
public int executeNonQuery(String sql, Object parameter) {
int result;
if (autoCommit) {
SqlSession sqlSession = sqlSessionFactory.openSession();
try {
result = executeNonQuery(sql, parameter, sqlSession);
sqlSession.commit();
} catch (Exception e) {
sqlSession.rollback();
throw new ExecuteSqlException(e.getMessage(), e);
} finally {
sqlSession.close();
}
} else {
result = executeNonQuery(sql, parameter, globalSqlSession);
}
return result;
}
@Override
public List pageQueryEntities(String sql, PageInfo pageInfo, Object parameter, Class entityType) {
List result;
if (autoCommit) {
SqlSession sqlSession = sqlSessionFactory.openSession();
try {
result = pageQueryEntities(sql, pageInfo, parameter, sqlSession);
} finally {
sqlSession.close();
}
} else {
result = pageQueryEntities(sql, pageInfo, parameter, globalSqlSession);
}
return result;
}
@Override
public void commit() {
if (!autoCommit && globalSqlSession != null) {
globalSqlSession.commit(true);
}
}
@Override
public void rollback() {
if (!autoCommit && globalSqlSession != null) {
globalSqlSession.rollback(true);
}
}
@Override
public void close() {
if (!autoCommit && globalSqlSession != null) {
globalSqlSession.close();
}
}
@Override
public T select(Object objKey, Class entityType) {
String mappedStatementId = MappedStatementUtils.getMappedStatementId(sqlSessionFactory.getConfiguration(), dbDialect, SqlType.SELECT, entityType, objKey.getClass());
return queryEntity(mappedStatementId, objKey, entityType);
}
@Override
public List selectList(Object objParams, Class entityType) {
Map mapParams = EntityUtils.entityToMap(objParams, true);
String[] propertyNames = new String[mapParams.keySet().size()];
propertyNames = mapParams.keySet().toArray(propertyNames);
String mappedStatementId = MappedStatementUtils.getMappedStatementId(sqlSessionFactory.getConfiguration(), dbDialect, SqlType.SELECT, entityType, mapParams.getClass(), propertyNames);
return queryEntities(mappedStatementId, mapParams, entityType);
}
@Override
public T selectOne(Object objParams, Class entityType) {
List list = selectList(objParams, entityType);
return list.size() > 0 ? list.get(0) : null;
}
@Override
public int selectCount(Object objParams, Class entityType) {
Map mapParams = EntityUtils.entityToMap(objParams, true);
String[] propertyNames = new String[mapParams.keySet().size()];
propertyNames = mapParams.keySet().toArray(propertyNames);
String mappedStatementId = MappedStatementUtils.getMappedStatementId(sqlSessionFactory.getConfiguration(), dbDialect, SqlType.SELECT_COUNT, entityType, mapParams.getClass(), propertyNames);
return queryEntity(mappedStatementId, mapParams, Integer.class);
}
@Override
public int insert(Object entity) {
String mappedStatementId = MappedStatementUtils.getMappedStatementId(sqlSessionFactory.getConfiguration(), dbDialect, SqlType.INSERT, entity.getClass(), entity.getClass());
return executeNonQuery(mappedStatementId, entity);
}
@Override
public int update(Object entity) {
return update(entity, false);
}
@Override
public int update(Object entity, boolean excludeNullFields) {
String mappedStatementId;
if (excludeNullFields) {
Map map = EntityUtils.entityToMap(entity, true);
String[] propertyNames = map.keySet().toArray(new String[]{});
mappedStatementId = MappedStatementUtils.getMappedStatementId(sqlSessionFactory.getConfiguration(), dbDialect, SqlType.UPDATE, entity.getClass(), entity.getClass(), propertyNames);
} else {
mappedStatementId = MappedStatementUtils.getMappedStatementId(sqlSessionFactory.getConfiguration(), dbDialect, SqlType.UPDATE, entity.getClass(), entity.getClass());
}
return executeNonQuery(mappedStatementId, entity);
}
@Override
public int insertOrUpdate(Object entity) {
int result = update(entity);
if (result < 1) {
result = insert(entity);
}
return result;
}
@Override
public int delete(Object objKey, Class entityType) {
String mappedStatementId = MappedStatementUtils.getMappedStatementId(sqlSessionFactory.getConfiguration(), dbDialect, SqlType.DELETE, entityType, objKey.getClass());
return executeNonQuery(mappedStatementId, objKey);
}
private T queryEntity(String sql, Object parameter, SqlSession sqlSession) {
T result;
result = sqlSession.selectOne(sql, parameter);
return result;
}
private List pageQueryEntities(String sql, PageInfo pageInfo, Object parameter, SqlSession sqlSession) {
List result;
Map mapParameter = EntityUtils.entityToMap(parameter);
if (mapParameter == null) {
mapParameter = new HashMap<>();
}
PageQueryParameter pageQueryParameter = new PageQueryParameter();
pageQueryParameter.setTarget(this);
// pageQueryParameter.setParameter(parameter);
pageQueryParameter.setOrderClause(pageInfo.getOrderClause());
pageQueryParameter.setPageIndex(pageInfo.getPageIndex());
pageQueryParameter.setRowCount(pageInfo.getRowCount());
mapParameter.put(PAGE_QUERY_PARAMETER_KEY, pageQueryParameter);
result = sqlSession.selectList(sql, mapParameter);
pageInfo.setTotalRowCount(pageQueryParameter.getTotalRowCount());
return result;
}
private int executeNonQuery(String sql, Object parameter, SqlSession sqlSession) {
int result;
MappedStatement mappedStatement = sqlSession.getConfiguration().getMappedStatement(sql);
SqlCommandType sqlCommandType = mappedStatement.getSqlCommandType();
if (sqlCommandType == SqlCommandType.INSERT) {
result = sqlSession.insert(sql, parameter);
} else if (sqlCommandType == SqlCommandType.UPDATE) {
result = sqlSession.update(sql, parameter);
} else if (sqlCommandType == SqlCommandType.DELETE) {
result = sqlSession.delete(sql, parameter);
} else {
throw new RuntimeException("sqlCommandType:" + sqlCommandType.toString() + "没有对应的执行方法！");
}
return result;
}
private List queryEntities(String sql, Object parameter, SqlSession sqlSession) {
List result;
result = sqlSession.selectList(sql, parameter);
return result;
}
}
