diff --git a/src/main/java/com/alibaba/druid/util/Histogram.java b/src/main/java/com/alibaba/druid/util/Histogram.java index d2ae1959ef..a52edd53fa 100644 --- a/src/main/java/com/alibaba/druid/util/Histogram.java +++ b/src/main/java/com/alibaba/druid/util/Histogram.java @@ -77,6 +77,10 @@ public class Histogram { return ranges; } + public long getValue(int index) { + return rangeCounters[index].get(); + } + public String toString() { StringBuilder buf = new StringBuilder(); buf.append('['); diff --git a/src/test/java/com/alibaba/druid/bvt/filter/PoolStatFilterTest.java b/src/test/java/com/alibaba/druid/bvt/filter/PoolStatFilterTest.java new file mode 100644 index 0000000000..6069541a64 --- /dev/null +++ b/src/test/java/com/alibaba/druid/bvt/filter/PoolStatFilterTest.java @@ -0,0 +1,43 @@ +package com.alibaba.druid.bvt.filter; + +import java.sql.Connection; + +import junit.framework.Assert; +import junit.framework.TestCase; + +import com.alibaba.druid.pool.DruidDataSource; +import com.alibaba.druid.util.Histogram; +import com.alibaba.druid.util.JdbcUtils; + +public class PoolStatFilterTest extends TestCase { + + private DruidDataSource dataSource; + + protected void setUp() throws Exception { + dataSource = new DruidDataSource(); + + dataSource.setUrl("jdbc:mock:xxx"); + dataSource.setFilters("stat"); + + dataSource.init(); + } + + protected void tearDown() throws Exception { + JdbcUtils.close(dataSource); + } + + public void test_stat() throws Exception { + + Assert.assertTrue(dataSource.isInited()); + + Histogram histogram = dataSource.getDataSourceStat().getConnectionHoldHistogram(); + + Assert.assertEquals(0, histogram.getValue(0)); + + Connection conn = dataSource.getConnection(); + + conn.close(); + + Assert.assertEquals(1, histogram.getValue(0)); + } +}