From cb6a29b37ff7b347ab0ae8c72a3e74cb71d99af9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=A9=E5=B0=91?= Date: Wed, 14 Mar 2012 16:09:12 +0800 Subject: [PATCH] add mapping support --- .../spi/MSSQLServerMappingProvider.java | 25 +++++++++++++------ 1 file changed, 18 insertions(+), 7 deletions(-) diff --git a/src/main/java/com/alibaba/druid/mapping/spi/MSSQLServerMappingProvider.java b/src/main/java/com/alibaba/druid/mapping/spi/MSSQLServerMappingProvider.java index 0ace040dc6..0a5c42875c 100644 --- a/src/main/java/com/alibaba/druid/mapping/spi/MSSQLServerMappingProvider.java +++ b/src/main/java/com/alibaba/druid/mapping/spi/MSSQLServerMappingProvider.java @@ -51,21 +51,32 @@ public class MSSQLServerMappingProvider implements MappingProvider { return query; } - + public SQLDeleteStatement explainToDeleteSQLObject(MappingEngine engine, String sql) { SQLServerStatementParser parser = new SQLServerStatementParser(sql); - return parser.parseDeleteStatement(); + SQLDeleteStatement stmt = parser.parseDeleteStatement(); + + MappingVisitorUtils.setDataSource(engine, stmt); + + return stmt; } - + public SQLUpdateStatement explainToUpdateSQLObject(MappingEngine engine, String sql) { PGSQLStatementParser parser = new PGSQLStatementParser(sql); - return parser.parseUpdateStatement(); + SQLUpdateStatement stmt = parser.parseUpdateStatement(); + + MappingVisitorUtils.setDataSource(engine, stmt); + + return stmt; } - - + public SQLInsertStatement explainToInsertSQLObject(MappingEngine engine, String sql) { PGSQLStatementParser parser = new PGSQLStatementParser(sql); - return (SQLInsertStatement) parser.parseInsert(); + SQLInsertStatement stmt = (SQLInsertStatement) parser.parseInsert(); + + MappingVisitorUtils.setDataSource(engine, stmt); + + return stmt; } @Override