開篇
?這篇文章的目的是講解RM Executor模塊當中一些通用的方法，這些方法在各個Executor的父類當中實現(xiàn)的策州，各個子類Executor模塊都會復用，因此抽取出來統(tǒng)一的進行講解宫仗。
?個人是認為抽取通用的內(nèi)容放在一篇文章講解完后可以針對每類Executor講解特有的功能够挂，這樣能夠有更好的理解。這篇文章講解Executor的實現(xiàn)類InsertExecutor藕夫。
類依賴圖
說明：
- 著重講解InsertExecutor實現(xiàn)類孽糖。
InsertExecutor方法介紹
public class InsertExecutor<T, S extends Statement> extends AbstractDMLBaseExecutor<T, S> {
protected TableRecords beforeImage() throws SQLException {
return TableRecords.empty(getTableMeta());
}
}
public class TableRecords {
public static TableRecords empty(TableMeta tableMeta) {
return new TableRecords(tableMeta) {
@Override
public int size() {
return 0;
}
@Override
public List<Field> pkRows() {
return new ArrayList<>();
}
@Override
public void add(Row row) {
throw new UnsupportedOperationException("xxx");
}
@Override
public TableMeta getTableMeta() {
throw new UnsupportedOperationException("xxx");
}
};
}
}
說明：
- Insert操作的執(zhí)行前鏡像顧名思義是空的，所以beforeImage()方法返回空記錄毅贮。
- TableRecords.empty返回的是一個沒有記錄的TableRecords對象办悟。
public class InsertExecutor<T, S extends Statement> extends AbstractDMLBaseExecutor<T, S> {
protected TableRecords afterImage(TableRecords beforeImage) throws SQLException {
SQLInsertRecognizer recogizier = (SQLInsertRecognizer)sqlRecognizer;
List<String> insertColumns = recogizier.getInsertColumns();
TableMeta tmeta = getTableMeta();
TableRecords afterImage = null;
if (tmeta.containsPK(insertColumns)) {
// 處理插入數(shù)據(jù)包含主鍵的情況
List<Object> pkValues = null;
String pk = tmeta.getPkName();
for (int paramIdx = 0; paramIdx < insertColumns.size(); paramIdx++) {
if (insertColumns.get(paramIdx).equalsIgnoreCase(pk)) {
// 處理PreparedStatement類型，從參數(shù)列表中提取
if (statementProxy instanceof PreparedStatementProxy) {
pkValues = ((PreparedStatementProxy) statementProxy).getParamsByIndex(paramIdx);
} else {
// 處理Statement類型,直接從SQL語句中提取
List<List<Object>> insertRows = recogizier.getInsertRows();
pkValues = new ArrayList<>(insertRows.size());
for (List<Object> row : insertRows) {
pkValues.add(row.get(paramIdx));
}
}
//主鍵只有一個
break;
}
}
if (pkValues == null) {
throw new ShouldNeverHappenException();
}
afterImage = getTableRecords(pkValues);
} else {
// 處理主鍵自動生成的場景滩褥，也就是不包含主鍵的情況病蛉。
Map<String, ColumnMeta> pkMetaMap = getTableMeta().getPrimaryKeyMap();
if (pkMetaMap.size() != 1) {
throw new NotSupportYetException();
}
// 獲取primary key的元數(shù)據(jù)信息
ColumnMeta pkMeta = pkMetaMap.values().iterator().next();
if (!pkMeta.isAutoincrement()) {
throw new ShouldNeverHappenException();
}
// 獲取generatedKeys()獲取生成的主鍵
ResultSet genKeys = null;
try {
genKeys = statementProxy.getTargetStatement().getGeneratedKeys();
} catch (SQLException e) {
if ("S1009".equalsIgnoreCase(e.getSQLState())) {
genKeys = statementProxy.getTargetStatement().executeQuery("SELECT LAST_INSERT_ID()");
} else {
throw e;
}
}
List<Object> pkValues = new ArrayList<>();
while (genKeys.next()) {
Object v = genKeys.getObject(1);
pkValues.add(v);
}
// 生成執(zhí)行后鏡像
afterImage = getTableRecords(pkValues);
}
if (afterImage == null) {
throw new SQLException("Failed to build after-image for insert");
}
return afterImage;
}
}
說明：
- afterImage準備執(zhí)行后的鏡像，針對Insert操作就是新增的記錄瑰煎。
- afterImage處理分為兩種場景铺然，分為插入記錄包含主鍵字段和不包含主鍵字段。
- 記錄包含主鍵字段場景下酒甸，區(qū)分preparedStatement和Statement兩種場景魄健，前者去傳入?yún)?shù)中獲取，后者從執(zhí)行的SQL當中解析烘挫。
- 不包含主鍵字段場景下诀艰，通過獲取getGeneratedKeys()返回的值獲取柬甥。
- 獲取主鍵的值以后pkValues饮六，然后通過getTableRecords()生成執(zhí)行后鏡像。
public class InsertExecutor<T, S extends Statement> extends AbstractDMLBaseExecutor<T, S> {
private TableRecords getTableRecords(List<Object> pkValues) throws SQLException {
TableRecords afterImage;
String pk = getTableMeta().getPkName();
StringBuffer selectSQLAppender = new StringBuffer("SELECT * FROM " + getTableMeta().getTableName() + " WHERE ");
for (int i = 1; i <= pkValues.size(); i++) {
selectSQLAppender.append(pk + "=?");
if (i < pkValues.size()) {
selectSQLAppender.append(" OR ");
}
}
PreparedStatement ps = null;
ResultSet rs = null;
try {
ps = statementProxy.getConnection().prepareStatement(selectSQLAppender.toString());
for (int i = 1; i <= pkValues.size(); i++) {
ps.setObject(i, pkValues.get(i - 1));
}
rs = ps.executeQuery();
afterImage = TableRecords.buildRecords(getTableMeta(), rs);
} finally {
if (rs != null) {
rs.close();
}
if (ps != null) {
ps.close();
}
}
return afterImage;
}
}
說明：
- getTableRecords通過針對主鍵信息拼接SELECT的SQL語句拼接成根據(jù)主鍵查詢字段信息的SQL苛蒲。
- 通過buildRecords()方法針對查詢結果ResultSet進行處理卤橄。
- 生成執(zhí)行后鏡像信息在buildRecords()內(nèi)部實現(xiàn)。
public class TableRecords {
public static TableRecords buildRecords(TableMeta tmeta, ResultSet resultSet) throws SQLException {
TableRecords records = new TableRecords(tmeta);
ResultSetMetaData resultSetMetaData = resultSet.getMetaData();
int columnCount = resultSetMetaData.getColumnCount();
while (resultSet.next()) {
List<Field> fields = new ArrayList<>(columnCount);
for (int i = 1; i <= columnCount; i++) {
String colName = resultSetMetaData.getColumnName(i);
ColumnMeta col = tmeta.getColumnMeta(colName);
Field field = new Field();
field.setName(col.getColumnName());
if (tmeta.getPkName().equals(field.getName())) {
field.setKeyType(KeyType.PrimaryKey);
}
field.setType(col.getDataType());
field.setValue(resultSet.getObject(i));
fields.add(field);
}
Row row = new Row();
row.setFields(fields);
records.add(row);
}
return records;
}
}
說明：
- buildRecords內(nèi)部遍歷查詢的結果憑借Row對象臂外。
- 插入的數(shù)據(jù)可能有多條窟扑，外層while循環(huán)獲取所有的行數(shù)，內(nèi)層for循環(huán)遍歷所有的列數(shù)漏健。
期待
?InsertExecutor生成執(zhí)行前后鏡像的過程分析完后嚎货，會接著分析DeleteExecutor。額外的收貨就是分析完對于SQL或者說對阿里開源的druid有了一些深入研究的興趣蔫浆。
