mirror of
https://github.com/alibaba/druid.git
synced 2023-03-04 19:17:16 +00:00
enhance mapping
This commit is contained in:
@@ -64,7 +64,7 @@ public class MSSQLServerMappingProvider implements MappingProvider {
|
||||
SQLServerStatementParser parser = new SQLServerStatementParser(sql);
|
||||
SQLDeleteStatement stmt = parser.parseDeleteStatement();
|
||||
|
||||
MappingVisitorUtils.setDataSource(engine, stmt);
|
||||
MappingVisitorUtils.setTableSource(engine, stmt);
|
||||
|
||||
return stmt;
|
||||
}
|
||||
@@ -73,7 +73,7 @@ public class MSSQLServerMappingProvider implements MappingProvider {
|
||||
SQLServerStatementParser parser = new SQLServerStatementParser(sql);
|
||||
SQLUpdateStatement stmt = parser.parseUpdateStatement();
|
||||
|
||||
MappingVisitorUtils.setDataSource(engine, stmt);
|
||||
MappingVisitorUtils.setTableSource(engine, stmt);
|
||||
|
||||
return stmt;
|
||||
}
|
||||
@@ -82,7 +82,7 @@ public class MSSQLServerMappingProvider implements MappingProvider {
|
||||
SQLServerStatementParser parser = new SQLServerStatementParser(sql);
|
||||
SQLInsertStatement stmt = (SQLInsertStatement) parser.parseInsert();
|
||||
|
||||
MappingVisitorUtils.setDataSource(engine, stmt);
|
||||
MappingVisitorUtils.setTableSource(engine, stmt);
|
||||
|
||||
return stmt;
|
||||
}
|
||||
|
||||
@@ -470,21 +470,21 @@ public class MappingVisitorUtils {
|
||||
return entity;
|
||||
}
|
||||
|
||||
public static void setDataSource(MappingEngine engine, SQLDeleteStatement stmt) {
|
||||
public static void setTableSource(MappingEngine engine, SQLDeleteStatement stmt) {
|
||||
if (stmt.getExprTableSource() == null) {
|
||||
Entity entity = engine.getFirstEntity();
|
||||
stmt.setTableSource(new SQLIdentifierExpr(entity.getName()));
|
||||
}
|
||||
}
|
||||
|
||||
public static void setDataSource(MappingEngine engine, SQLUpdateStatement stmt) {
|
||||
public static void setTableSource(MappingEngine engine, SQLUpdateStatement stmt) {
|
||||
if (stmt.getTableSource() == null) {
|
||||
Entity entity = engine.getFirstEntity();
|
||||
stmt.setTableSource(new SQLIdentifierExpr(entity.getName()));
|
||||
}
|
||||
}
|
||||
|
||||
public static void setDataSource(MappingEngine engine, SQLInsertStatement stmt) {
|
||||
public static void setTableSource(MappingEngine engine, SQLInsertStatement stmt) {
|
||||
if (stmt.getTableSource() == null) {
|
||||
Entity entity = engine.getFirstEntity();
|
||||
stmt.setTableSource(new SQLIdentifierExpr(entity.getName()));
|
||||
|
||||
@@ -64,7 +64,7 @@ public class MySqlMappingProvider implements MappingProvider {
|
||||
MySqlStatementParser parser = new MySqlStatementParser(sql);
|
||||
|
||||
MySqlDeleteStatement stmt = parser.parseDeleteStatement();
|
||||
MappingVisitorUtils.setDataSource(engine, stmt);
|
||||
MappingVisitorUtils.setTableSource(engine, stmt);
|
||||
|
||||
return stmt;
|
||||
}
|
||||
@@ -73,7 +73,7 @@ public class MySqlMappingProvider implements MappingProvider {
|
||||
MySqlStatementParser parser = new MySqlStatementParser(sql);
|
||||
SQLUpdateStatement stmt = parser.parseUpdateStatement();
|
||||
|
||||
MappingVisitorUtils.setDataSource(engine, stmt);
|
||||
MappingVisitorUtils.setTableSource(engine, stmt);
|
||||
|
||||
return stmt;
|
||||
}
|
||||
@@ -82,7 +82,7 @@ public class MySqlMappingProvider implements MappingProvider {
|
||||
MySqlStatementParser parser = new MySqlStatementParser(sql);
|
||||
SQLInsertStatement stmt = (SQLInsertStatement) parser.parseInsert();
|
||||
|
||||
MappingVisitorUtils.setDataSource(engine, stmt);
|
||||
MappingVisitorUtils.setTableSource(engine, stmt);
|
||||
|
||||
return stmt;
|
||||
}
|
||||
|
||||
@@ -45,7 +45,7 @@ public class OracleMappingProvider implements MappingProvider {
|
||||
OracleStatementParser parser = new OracleStatementParser(sql);
|
||||
OracleDeleteStatement stmt = parser.parseDeleteStatement();
|
||||
|
||||
MappingVisitorUtils.setDataSource(engine, stmt);
|
||||
MappingVisitorUtils.setTableSource(engine, stmt);
|
||||
|
||||
return stmt;
|
||||
}
|
||||
@@ -54,7 +54,7 @@ public class OracleMappingProvider implements MappingProvider {
|
||||
OracleStatementParser parser = new OracleStatementParser(sql);
|
||||
SQLUpdateStatement stmt = parser.parseUpdateStatement();
|
||||
|
||||
MappingVisitorUtils.setDataSource(engine, stmt);
|
||||
MappingVisitorUtils.setTableSource(engine, stmt);
|
||||
|
||||
return stmt;
|
||||
}
|
||||
@@ -62,7 +62,7 @@ public class OracleMappingProvider implements MappingProvider {
|
||||
public SQLInsertStatement explainToInsertSQLObject(MappingEngine engine, String sql) {
|
||||
OracleStatementParser parser = new OracleStatementParser(sql);
|
||||
SQLInsertStatement stmt = (SQLInsertStatement) parser.parseInsert();
|
||||
MappingVisitorUtils.setDataSource(engine, stmt);
|
||||
MappingVisitorUtils.setTableSource(engine, stmt);
|
||||
return stmt;
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user