Merge branch 'master' of https://wenshao@github.com/AlibabaTech/druid.git

2023-03-04 19:17:16 +00:00 · 2012-03-31 09:08:15 +08:00
parent 3b79f5e99c a8bd3b249a
commit d8e5935957
1 changed files with 40 additions and 0 deletions
@@ -0,0 +1,40 @@
+package com.alibaba.druid.filter.stat;
+
+import com.alibaba.druid.logging.Log;
+import com.alibaba.druid.logging.LogFactory;
+import com.alibaba.druid.proxy.jdbc.DataSourceProxy;
+import com.alibaba.druid.sql.visitor.ParameterizedOutputVisitorUtils;
+import com.alibaba.druid.stat.JdbcSqlStat;
+
+public class MergeSqlStatFilter extends StatFilter {
+
+    private final static Log LOG = LogFactory.getLog(MergeSqlStatFilter.class);
+
+    private String           dbType;
+
+    public String getDbType() {
+        return dbType;
+    }
+
+    public void setDbType(String dbType) {
+        this.dbType = dbType;
+    }
+
+    @Override
+    public void init(DataSourceProxy dataSource) {
+        super.init(dataSource);
+
+        if (this.dbType == null || this.dbType.trim().length() == 0) {
+            this.dbType = dataSource.getDbType();
+        }
+    }
+
+    public JdbcSqlStat getSqlStat(String sql) {
+        try {
+            sql = ParameterizedOutputVisitorUtils.parameterize(sql, sql);
+        } catch (Exception e) {
+            LOG.error("merge sql error", e);
+        }
+        return dataSourceStat.getSqlStat(sql);
+    }
+}