enhance mapping

This commit is contained in:
温少
2012-05-10 10:57:24 +08:00
parent 574e8b7b99
commit d028f26d3d
4 changed files with 12 additions and 12 deletions
@@ -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;
}