Share project "druid" into "http://code.alibabatech.com/svn/druid"

git-svn-id: http://code.alibabatech.com/svn/druid/trunk@2 b9813039-fb51-4c41-a8b9-e21c2acb5095

src/main/java/com/alibaba/druid/stat/JdbcDataSourceStat.java

@@ -0,0 +1,176 @@
+/**
+ * Project: druid
+ * 
+ * File Created at 2010-12-2
+ * $Id$
+ * 
+ * Copyright 1999-2100 Alibaba.com Corporation Limited.
+ * All rights reserved.
+ *
+ * This software is the confidential and proprietary information of
+ * Alibaba Company. ("Confidential Information").  You shall not
+ * disclose such Confidential Information and shall use it only in
+ * accordance with the terms of the license agreement you entered into
+ * with Alibaba.com.
+ */
+package com.alibaba.druid.stat;
+
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.ConcurrentMap;
+
+import javax.management.JMException;
+import javax.management.openmbean.CompositeType;
+import javax.management.openmbean.TabularData;
+import javax.management.openmbean.TabularDataSupport;
+import javax.management.openmbean.TabularType;
+
+import com.alibaba.druid.filter.Filter;
+import com.alibaba.druid.filter.stat.StatFilter;
+import com.alibaba.druid.proxy.jdbc.DataSourceProxy;
+
+public class JdbcDataSourceStat implements JdbcDataSourceStatMBean {
+	private final String name;
+	private final String url;
+
+	private final JdbcConnectionStat connectionStat = new JdbcConnectionStat();
+	private final JdbcResultSetStat resultSetStat = new JdbcResultSetStat();
+	private final JdbcStatementStat statementStat = new JdbcStatementStat();
+
+	private final ConcurrentMap<String, JdbcSqlStat> sqlStatMap = new ConcurrentHashMap<String, JdbcSqlStat>();
+
+	private final ConcurrentMap<Long, JdbcConnectionStat.Entry> connections = new ConcurrentHashMap<Long, JdbcConnectionStat.Entry>();
+
+	public JdbcDataSourceStat(String name, String url) {
+		this.name = name;
+		this.url = url;
+	}
+
+	public void reset() {
+		connectionStat.reset();
+		statementStat.reset();
+		resultSetStat.reset();
+		
+		for (JdbcSqlStat sqlStat : sqlStatMap.values()) {
+			if (sqlStat.getExecuteCount() == 0) {
+				continue;
+			}
+			
+			sqlStat.reset();
+		}
+		sqlStatMap.clear();
+		
+		for (JdbcConnectionStat.Entry connectionStat : connections.values()) {
+			connectionStat.reset();
+		}
+		// connections.clear();
+	}
+
+	public JdbcConnectionStat getConnectionStat() {
+		return connectionStat;
+	}
+
+	public JdbcResultSetStat getResultSetStat() {
+		return resultSetStat;
+	}
+
+	public JdbcStatementStat getStatementStat() {
+		return statementStat;
+	}
+
+	@Override
+	public String getConnectionUrl() {
+		return url;
+	}
+
+	public JdbcSqlStat getSqlCounter(String sql) {
+		return this.sqlStatMap.get(sql);
+	}
+
+	public ConcurrentMap<String, JdbcSqlStat> getSqlStatisticMap() {
+		return this.sqlStatMap;
+	}
+
+	@Override
+	public TabularData getSqlList() throws JMException {
+		CompositeType rowType = JdbcSqlStat.getCompositeType();
+		String[] indexNames = rowType.keySet().toArray(new String[rowType.keySet().size()]);
+
+		TabularType tabularType = new TabularType("SqlListStatistic", "SqlListStatistic", rowType, indexNames);
+		TabularData data = new TabularDataSupport(tabularType);
+
+		for (Map.Entry<String, JdbcSqlStat> entry : sqlStatMap.entrySet()) {
+			data.put(entry.getValue().getCompositeData());
+		}
+
+		return data;
+	}
+
+	public static StatFilter getCounterFilter(DataSourceProxy dataSource) {
+		for (Filter filter : dataSource.getFilters()) {
+			if (filter instanceof StatFilter) {
+				return (StatFilter) filter;
+			}
+		}
+
+		return null;
+	}
+
+	public JdbcSqlStat getSqlCounter(long id) {
+		for (Map.Entry<String, JdbcSqlStat> entry : this.sqlStatMap.entrySet()) {
+			if (entry.getValue().getId() == id) {
+				return entry.getValue();
+			}
+		}
+
+		return null;
+	}
+
+	public final ConcurrentMap<Long, JdbcConnectionStat.Entry> getConnections() {
+		return connections;
+	}
+
+	@Override
+	public TabularData getConnectionList() throws JMException {
+		CompositeType rowType = JdbcConnectionStat.Entry.getCompositeType();
+		String[] indexNames = rowType.keySet().toArray(new String[rowType.keySet().size()]);
+
+		TabularType tabularType = new TabularType("ConnectionListStatistic", "ConnectionListStatistic", rowType,
+				indexNames);
+		TabularData data = new TabularDataSupport(tabularType);
+
+		for (Map.Entry<Long, JdbcConnectionStat.Entry> entry : getConnections().entrySet()) {
+			data.put(entry.getValue().getCompositeData());
+		}
+
+		return data;
+	}
+
+	public String getName() {
+		return name;
+	}
+
+	public String getUrl() {
+		return url;
+	}
+
+	public ConcurrentMap<String, JdbcSqlStat> getSqlStatMap() {
+		return sqlStatMap;
+	}
+
+	@Override
+	public long getConnectionActiveCount() {
+		return this.connections.size();
+	}
+
+	@Override
+	public long getConnectionConnectAliveMillis() {
+		long nowNano = System.nanoTime();
+		long aliveNanoSpan = this.getConnectionStat().getNanoTotal();
+
+		for (JdbcConnectionStat.Entry connection : connections.values()) {
+			aliveNanoSpan += nowNano - connection.getEstablishNano();
+		}
+		return aliveNanoSpan / (1000 * 1000);
+	}
+}