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.jfirer.jsql.session.impl.SqlSessionImpl Maven / Gradle / Ivy
package com.jfirer.jsql.session.impl;
import com.jfirer.jsql.curd.CurdInfo;
import com.jfirer.jsql.curd.LockMode;
import com.jfirer.jsql.dialect.Dialect;
import com.jfirer.jsql.executor.SqlInvoker;
import com.jfirer.jsql.mapper.Mapper;
import com.jfirer.jsql.metadata.TableEntityInfo;
import com.jfirer.jsql.model.Model;
import com.jfirer.jsql.session.SqlSession;
import com.jfirer.jsql.transfer.resultset.ResultSetTransfer;
import com.jfirer.jsql.transfer.resultset.impl.IntegerTransfer;
import com.jfirer.baseutil.reflect.ReflectUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.lang.reflect.Field;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.IdentityHashMap;
import java.util.List;
public class SqlSessionImpl implements SqlSession
{
private boolean transactionActive = false;
private boolean closed = false;
private final IdentityHashMap, Class> mappers;
private final Connection connection;
private final SqlInvoker sqlInvoker;
private final IdentityHashMap, CurdInfo> curdInfoMap;
private final Dialect dialect;
private final static Logger logger = LoggerFactory.getLogger(SqlSession.class);
private static final ThreadLocal> cahcedParams = new ThreadLocal>()
{
protected java.util.List initialValue()
{
return new ArrayList();
}
};
private static final ResultSetTransfer countTransfer = new IntegerTransfer();
public SqlSessionImpl(Connection connection, SqlInvoker sqlInvoker, IdentityHashMap, CurdInfo> curdInfoMap, IdentityHashMap, Class> mappers, Dialect dialect)
{
this.connection = connection;
this.sqlInvoker = sqlInvoker;
this.curdInfoMap = curdInfoMap;
this.dialect = dialect;
this.mappers = mappers;
}
@Override
public void beginTransAction()
{
checkIfClosed();
try
{
if ( transactionActive != false )
{
return;
}
connection.setAutoCommit(false);
} catch (SQLException e)
{
ReflectUtil.throwException(e);
}
}
private void checkIfClosed()
{
if ( closed )
{
throw new IllegalStateException("当前Session已经关闭，不能执行其他操作");
}
}
@Override
public void commit()
{
checkIfClosed();
if ( transactionActive == false )
{
throw new IllegalStateException("当前链接未开启事务，无法进行提交");
}
try
{
connection.commit();
connection.setAutoCommit(true);
transactionActive = false;
} catch (SQLException e)
{
ReflectUtil.throwException(e);
}
}
@Override
public void flush()
{
checkIfClosed();
try
{
connection.commit();
} catch (SQLException e)
{
ReflectUtil.throwException(e);
}
}
@Override
public void rollback()
{
checkIfClosed();
if ( transactionActive == false )
{
throw new IllegalStateException("当前链接未开启事务，无需回滚");
}
try
{
connection.rollback();
connection.setAutoCommit(true);
transactionActive = false;
} catch (SQLException e)
{
ReflectUtil.throwException(e);
}
}
@Override
public void close()
{
if ( transactionActive )
{
throw new IllegalStateException("当前链接仍然开启着事务，需要先执行提交");
}
try
{
closed = true;
connection.close();
logger.trace("关闭session");
} catch (SQLException e)
{
throw new RuntimeException("关闭", e);
}
}
@Override
public Connection getConnection()
{
return connection;
}
@SuppressWarnings("unchecked")
@Override
public void save(T entity)
{
TableEntityInfo tableEntityInfo = TableEntityInfo.parse(entity.getClass());
Field pkField = tableEntityInfo.getPkInfo().getField();
try
{
if ( pkField.get(entity) == null )
{
CurdInfo curdInfo = (CurdInfo) curdInfoMap.get(entity.getClass());
List list = cahcedParams.get();
String sql = curdInfo.autoGeneratePkInsert(entity, list);
String pk = insertReturnPk(sql, list);
curdInfo.setPkValue(entity, pk);
list.clear();
}
else
{
update(entity);
}
} catch (Exception e)
{
ReflectUtil.throwException(e);
}
}
@SuppressWarnings("unchecked")
@Override
public void update(T entity)
{
CurdInfo curdInfo = (CurdInfo) curdInfoMap.get(entity.getClass());
List list = cahcedParams.get();
String sql = curdInfo.update(entity, list);
update(sql, list);
list.clear();
}
@SuppressWarnings("unchecked")
@Override
public int delete(Class ckass, Object pk)
{
List list = cahcedParams.get();
CurdInfo curdInfo = (CurdInfo) curdInfoMap.get(ckass);
String sql = curdInfo.delete(pk, list);
int update = update(sql, list);
list.clear();
return update;
}
@SuppressWarnings({"unchecked"})
@Override
public void insert(T entity)
{
List list = cahcedParams.get();
CurdInfo curdInfo = (CurdInfo) curdInfoMap.get(entity.getClass());
String sql = curdInfo.insert(entity, list);
update(sql, list);
list.clear();
}
@SuppressWarnings("unchecked")
@Override
public T get(Class entityClass, Object pk)
{
List list = cahcedParams.get();
CurdInfo curdInfo = (CurdInfo) curdInfoMap.get(entityClass);
String sql = curdInfo.find(pk, list);
T result = query(curdInfo.getBeanTransfer(), sql, list);
list.clear();
return result;
}
@Override
public T get(Class entityClass, Object pk, LockMode mode)
{
List list = cahcedParams.get();
CurdInfo curdInfo = curdInfoMap.get(entityClass);
String sql = curdInfo.find(pk, mode, list);
T result = query(curdInfo.getBeanTransfer(), sql, list);
list.clear();
return result;
}
@Override
public T findOne(Model model)
{
return query(model.getBeanTransfer(), model.getSql(), model.getParams());
}
@Override
public List find(Model model)
{
return queryList(model.getBeanTransfer(), model.getSql(), model.getParams());
}
@Override
public int update(Model model)
{
return update(model.getSql(), model.getParams());
}
@Override
public int delete(Model model)
{
return update(model.getSql(), model.getParams());
}
@Override
public int count(Model model)
{
return query(countTransfer, model.getSql(), model.getParams());
}
@Override
public void insert(Model model)
{
update(model.getSql(), model.getParams());
}
@Override
public int update(String sql, List params)
{
checkIfClosed();
try
{
return sqlInvoker.update(sql, params, connection, dialect);
} catch (SQLException e)
{
ReflectUtil.throwException(e);
return 0;
}
}
@Override
public String insertReturnPk(String sql, List params)
{
checkIfClosed();
try
{
return sqlInvoker.insertWithReturnKey(sql, params, connection, dialect);
} catch (SQLException e)
{
ReflectUtil.throwException(e);
return null;
}
}
@SuppressWarnings("unchecked")
@Override
public T query(ResultSetTransfer transfer, String sql, List params)
{
checkIfClosed();
try
{
return (T) sqlInvoker.queryOne(sql, params, connection, dialect, transfer);
} catch (SQLException e)
{
ReflectUtil.throwException(e);
return null;
}
}
@SuppressWarnings("unchecked")
@Override
public List queryList(ResultSetTransfer transfer, String sql, List params)
{
checkIfClosed();
try
{
return (List) sqlInvoker.queryList(sql, params, connection, dialect, transfer);
} catch (SQLException e)
{
ReflectUtil.throwException(e);
return null;
}
}
@SuppressWarnings("unchecked")
@Override
public T getMapper(Class mapperClass)
{
try
{
Class ckass = mappers.get(mapperClass);
Mapper mapper = ckass.newInstance();
mapper.setSession(this);
return (T) mapper;
} catch (Throwable e)
{
ReflectUtil.throwException(e);
return null;
}
}
}
