com.alibaba.druid.pool.DruidDataSource#setDbType

Source File: DruidStatTest.java From metrics with Apache License 2.0

6 votes

@Before
public void setUp() throws Exception {
    TabularData sqlList = JdbcStatManager.getInstance().getSqlList();
    if (sqlList.size() > 0) {
        for (Object item : JdbcStatManager.getInstance().getSqlList().values()) {
            String text = JSONUtils.toJSONString(item);
            System.out.println(text);
        }
    }

    Assert.assertEquals(0, JdbcStatManager.getInstance().getSqlList().size());

    dataSource = new DruidDataSource();
    dataSource.setUrl("jdbc:mock:xx");
    dataSource.setFilters("mergeStat");
    dataSource.setDbType("mysql");
}

Source File: DruidMetricsGaugeSetTest.java From metrics with Apache License 2.0

5 votes

@Test
public void testDruidMetricsSet() throws Exception {
    DruidDataSource dataSource = new DruidDataSource();
    dataSource.setUrl("jdbc:mock:xx");
    dataSource.setFilters("mergeStat");
    dataSource.setDbType("mysql");

    ManualClock clock = new ManualClock();
    MetricName name = new MetricName("druid2.sql");
    MetricManager.register("druid2", name,
            new DruidMetricsGaugeSet(5, TimeUnit.SECONDS, clock,20, name));

    doSelect(dataSource, 1000, 2000);
    doUpdate(dataSource, 2000, 2500);
    clock.addSeconds(6);

    SortedMap<MetricName, Gauge> druidMetrics = MetricManager.getIMetricManager().getGauges("druid2", MetricFilter.ALL);
    Assert.assertEquals(26, druidMetrics.size());
    Gauge g = druidMetrics.get(new MetricName("druid2.sql.ExecuteCount")
            .tagged("sql", "SELECT *\nFROM t\nWHERE t.id = ?"));
    Assert.assertEquals(1000L, g.getValue());

    doSelect(dataSource, 2000, 2500);
    clock.addSeconds(6);

    druidMetrics = MetricManager.getIMetricManager().getGauges("druid2", MetricFilter.ALL);
    Assert.assertEquals(26, druidMetrics.size());
    Gauge g2 = druidMetrics.get(new MetricName("druid2.sql.ExecuteCount")
            .tagged("sql", "SELECT *\nFROM t\nWHERE t.id = ?"));
    Assert.assertEquals(500L, g2.getValue());
}

Source File: DruidMetricsGaugeSetTest.java From metrics with Apache License 2.0

5 votes

@Test
public void testMaxSqlSize() throws Exception {
    DruidDataSource dataSource = new DruidDataSource();
    dataSource.setUrl("jdbc:mock:xx");
    dataSource.setFilters("mergeStat");
    dataSource.setDbType("mysql");

    ManualClock clock = new ManualClock();
    MetricName name = new MetricName("druid.sql");
    MetricManager.register("druid", name,
            new DruidMetricsGaugeSet(5, TimeUnit.SECONDS, clock,1,name));

    doSelect(dataSource, 1000, 2000);
    doUpdate(dataSource, 2000, 2500);
    clock.addSeconds(6);

    SortedMap<MetricName, Gauge> druidMetrics = MetricManager.getIMetricManager().getGauges("druid", MetricFilter.ALL);
    Assert.assertEquals(13, druidMetrics.size());
    // on Java 6 TabularDataSupport returns unordered list (HashMap)
    // So the return value it uncertain, it may be select or update
    // on Java 8+  TabularDataSupport returns ordered list (LinkedHashMap)
    Gauge g1 = druidMetrics.get(new MetricName("druid.sql.ExecuteCount")
            .tagged("sql", "SELECT *\nFROM t\nWHERE t.id = ?"));
    Gauge g2 = druidMetrics.get(new MetricName("druid.sql.ExecuteCount")
            .tagged("sql", "UPDATE t\nSET name = ?\nWHERE t.id = ?"));
    if (null != g1) {
        Assert.assertEquals(1000L, g1.getValue());
    } else {
        Assert.assertEquals(500L, g2.getValue());
    }
}