Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

multiStatmentAllow config

Browse Source
This commit is contained in:
温少
2012-03-20 01:30:13 +08:00
parent 0756bff598
commit 636a815c84
18 changed files with 17 additions and 4362 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/test/java/com/alibaba/druid/bvt/filter/TraceFilterTest.java
-157
View File
@@ -1,157 +0,0 @@
package com.alibaba.druid.bvt.filter;
import java.io.InputStream;
import java.io.Reader;
import java.sql.Blob;
import java.sql.Clob;
import java.sql.Connection;
import java.sql.NClob;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.Statement;
import java.sql.Types;
import junit.framework.Assert;
import junit.framework.TestCase;
import com.alibaba.druid.filter.trace.TraceFilter;
import com.alibaba.druid.mock.MockDriver;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.stat.DruidDataSourceStatManager;
import com.alibaba.druid.stat.JdbcStatContext;
import com.alibaba.druid.stat.JdbcStatManager;
public class TraceFilterTest extends TestCase {
private MockDriver driver;
private DruidDataSource dataSource;
private TraceFilter filter;
protected void setUp() throws Exception {
Assert.assertEquals(0, DruidDataSourceStatManager.getInstance().getDataSourceList().size());
driver = new MockDriver();
dataSource = new DruidDataSource();
dataSource.setUrl("jdbc:mock:xxx");
dataSource.setDriver(driver);
dataSource.setInitialSize(1);
dataSource.setMaxActive(2);
dataSource.setMaxIdle(2);
dataSource.setMinIdle(1);
dataSource.setMinEvictableIdleTimeMillis(300 * 1000); // 300 / 10
dataSource.setTimeBetweenEvictionRunsMillis(180 * 1000); // 180 / 10
dataSource.setTestWhileIdle(true);
dataSource.setTestOnBorrow(false);
dataSource.setValidationQuery("SELECT 1");
dataSource.setFilters("trace");
filter = (TraceFilter) dataSource.getProxyFilters().get(0);
JdbcStatContext statContext = new JdbcStatContext();
statContext.setTraceEnable(true);
JdbcStatManager.getInstance().setStatContext(statContext);
}
protected void tearDown() throws Exception {
dataSource.close();
Assert.assertEquals(0, DruidDataSourceStatManager.getInstance().getDataSourceList().size());
JdbcStatManager.getInstance().setStatContext(null);
}
public void test_exuecute() throws Exception {
Connection conn = dataSource.getConnection();
Statement stmt = conn.createStatement();
stmt.execute("SELECT 1");
stmt.close();
conn.close();
}
public void test_exuecuteQuery() throws Exception {
Connection conn = dataSource.getConnection();
Statement stmt = conn.createStatement();
ResultSet rs = stmt.executeQuery("SELECT 1");
rs.next();
rs.close();
stmt.close();
conn.close();
}
public void test_preExuecuteQuery() throws Exception {
Connection conn = dataSource.getConnection();
PreparedStatement stmt = conn.prepareStatement("SELECT ?");
stmt.setInt(1, 123);
ResultSet rs = stmt.executeQuery();
rs.next();
rs.close();
stmt.close();
conn.close();
}
public void test_set() throws Exception {
Connection conn = dataSource.getConnection();
PreparedStatement stmt = conn.prepareStatement("SELECT ?");
stmt.setInt(1, 123);
stmt.setArray(1, null);
stmt.setAsciiStream(1, null);
stmt.setAsciiStream(1, null, 0);
stmt.setAsciiStream(1, null, 0L);
stmt.setBigDecimal(1, null);
stmt.setBinaryStream(1, null);
stmt.setBinaryStream(1, null, 0);
stmt.setBinaryStream(1, null, 0L);
stmt.setBlob(1, (Blob) null);
stmt.setBlob(1, (InputStream) null);
stmt.setBlob(1, (InputStream) null, 0);
stmt.setBoolean(1, true);
stmt.setByte(1, (byte) 12);
stmt.setBytes(1, null);
stmt.setCharacterStream(1, null);
stmt.setCharacterStream(1, null, 0);
stmt.setCharacterStream(1, null, 0L);
stmt.setClob(1, (Clob) null);
stmt.setClob(1, (Reader) null);
stmt.setClob(1, (Reader) null, 1);
stmt.setDate(1, null);
stmt.setDate(1, null, null);
stmt.setDouble(1, 1D);
stmt.setFloat(1, 1F);
stmt.setLong(1, 1L);
stmt.setNCharacterStream(1, null);
stmt.setNCharacterStream(1, null, 0);
stmt.setNClob(1, (NClob) null);
stmt.setNClob(1, (Reader) null);
stmt.setNClob(1, (Reader) null, 1);
stmt.setNull(1, Types.INTEGER);
stmt.setNull(1, Types.INTEGER, "int");
stmt.setObject(1, null);
stmt.setObject(1, null, Types.INTEGER);
stmt.setObject(1, null, Types.INTEGER, 2);
stmt.setRef(1, null);
stmt.setRowId(1, null);
stmt.setShort(1, (short) 1);
stmt.setSQLXML(1, null);
stmt.setString(1, null);
stmt.setTime(1, null);
stmt.setTime(1, null, null);
stmt.setTimestamp(1, null);
stmt.setTimestamp(1, null, null);
stmt.setUnicodeStream(1, null, 0);
stmt.setURL(1, null);
ResultSet rs = stmt.executeQuery();
rs.next();
rs.previous();
rs.next();
rs.close();
stmt.close();
conn.close();
}
}
src/test/java/com/alibaba/druid/bvt/filter/TraceFilterTest_close_error.java
-101
View File
@@ -1,101 +0,0 @@
package com.alibaba.druid.bvt.filter;
import java.sql.SQLException;
import java.util.Properties;
import junit.framework.Assert;
import junit.framework.TestCase;
import com.alibaba.druid.filter.FilterChainImpl;
import com.alibaba.druid.filter.trace.TraceFilter;
import com.alibaba.druid.mock.MockConnection;
import com.alibaba.druid.mock.MockDriver;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.proxy.jdbc.ConnectionProxyImpl;
import com.alibaba.druid.stat.DruidDataSourceStatManager;
import com.alibaba.druid.stat.JdbcStatContext;
import com.alibaba.druid.stat.JdbcStatManager;
public class TraceFilterTest_close_error extends TestCase {
private MockDriver driver;
private DruidDataSource dataSource;
private TraceFilter filter;
protected void setUp() throws Exception {
Assert.assertEquals(0, DruidDataSourceStatManager.getInstance().getDataSourceList().size());
driver = new MockDriver() {
};
dataSource = new DruidDataSource();
dataSource.setUrl("jdbc:mock:TraceFilterTest_close_error");
dataSource.setDriver(driver);
dataSource.setInitialSize(1);
dataSource.setMaxActive(2);
dataSource.setMaxIdle(2);
dataSource.setMinIdle(1);
dataSource.setMinEvictableIdleTimeMillis(300 * 1000); // 300 / 10
dataSource.setTimeBetweenEvictionRunsMillis(180 * 1000); // 180 / 10
dataSource.setTestWhileIdle(true);
dataSource.setTestOnBorrow(false);
dataSource.setValidationQuery("SELECT 1");
dataSource.setFilters("trace");
filter = (TraceFilter) dataSource.getProxyFilters().get(0);
JdbcStatContext statContext = new JdbcStatContext();
statContext.setTraceEnable(true);
JdbcStatManager.getInstance().setStatContext(statContext);
}
protected void tearDown() throws Exception {
dataSource.close();
Assert.assertEquals(0, DruidDataSourceStatManager.getInstance().getDataSourceList().size());
JdbcStatManager.getInstance().setStatContext(null);
}
public void test_close_error() throws Exception {
FilterChainImpl chain = new FilterChainImpl(dataSource);
MockConnection conn = new MockConnection() {
public void close() throws SQLException {
throw new SQLException();
}
};
{
SQLException error = null;
try {
filter.connection_close(chain, new ConnectionProxyImpl(dataSource, conn, new Properties(), 0));
} catch (SQLException ex) {
error = ex;
}
Assert.assertNotNull(error);
}
}
public void test_close_error_2() throws Exception {
FilterChainImpl chain = new FilterChainImpl(dataSource);
MockConnection conn = new MockConnection() {
public void close() throws SQLException {
throw new RuntimeException();
}
};
{
Exception error = null;
try {
filter.connection_close(chain, new ConnectionProxyImpl(dataSource, conn, new Properties(), 0));
} catch (Exception ex) {
error = ex;
}
Assert.assertNotNull(error);
}
}
}
src/test/java/com/alibaba/druid/bvt/filter/TraceFilterTest_commit_error.java
-192
View File
@@ -1,192 +0,0 @@
package com.alibaba.druid.bvt.filter;
import java.sql.SQLException;
import java.sql.Savepoint;
import java.util.Properties;
import junit.framework.Assert;
import junit.framework.TestCase;
import com.alibaba.druid.filter.FilterChainImpl;
import com.alibaba.druid.filter.trace.TraceFilter;
import com.alibaba.druid.mock.MockConnection;
import com.alibaba.druid.mock.MockDriver;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.proxy.jdbc.ConnectionProxyImpl;
import com.alibaba.druid.stat.DruidDataSourceStatManager;
import com.alibaba.druid.stat.JdbcStatContext;
import com.alibaba.druid.stat.JdbcStatManager;
public class TraceFilterTest_commit_error extends TestCase {
private MockDriver driver;
private DruidDataSource dataSource;
private TraceFilter filter;
protected void setUp() throws Exception {
Assert.assertEquals(0, DruidDataSourceStatManager.getInstance().getDataSourceList().size());
driver = new MockDriver() {
};
dataSource = new DruidDataSource();
dataSource.setUrl("jdbc:mock:TraceFilterTest_commit_error");
dataSource.setDriver(driver);
dataSource.setInitialSize(1);
dataSource.setMaxActive(2);
dataSource.setMaxIdle(2);
dataSource.setMinIdle(1);
dataSource.setMinEvictableIdleTimeMillis(300 * 1000); // 300 / 10
dataSource.setTimeBetweenEvictionRunsMillis(180 * 1000); // 180 / 10
dataSource.setTestWhileIdle(true);
dataSource.setTestOnBorrow(false);
dataSource.setValidationQuery("SELECT 1");
dataSource.setFilters("trace");
filter = (TraceFilter) dataSource.getProxyFilters().get(0);
JdbcStatContext statContext = new JdbcStatContext();
statContext.setTraceEnable(true);
JdbcStatManager.getInstance().setStatContext(statContext);
}
protected void tearDown() throws Exception {
dataSource.close();
Assert.assertEquals(0, DruidDataSourceStatManager.getInstance().getDataSourceList().size());
JdbcStatManager.getInstance().setStatContext(null);
}
public void test_commit_error() throws Exception {
FilterChainImpl chain = new FilterChainImpl(dataSource);
MockConnection conn = new MockConnection() {
private int i = 0;
public void commit() throws SQLException {
if (i++ % 2 == 1) {
throw new RuntimeException();
} else {
throw new SQLException();
}
}
};
{
SQLException error = null;
try {
filter.connection_commit(chain, new ConnectionProxyImpl(dataSource, conn, new Properties(), 0));
} catch (SQLException ex) {
error = ex;
}
Assert.assertNotNull(error);
}
{
RuntimeException error = null;
try {
filter.connection_commit(chain, new ConnectionProxyImpl(dataSource, conn, new Properties(), 0));
} catch (RuntimeException ex) {
error = ex;
}
Assert.assertNotNull(error);
}
}
public void test_rollback_error() throws Exception {
FilterChainImpl chain = new FilterChainImpl(dataSource);
MockConnection conn = new MockConnection() {
private int i = 0;
public void rollback() throws SQLException {
int count = i++ % 3;
if (count % 3 == 1) {
throw new SQLException();
} else if (count % 3 == 2) {
throw new RuntimeException();
}
}
};
filter.connection_rollback(chain, new ConnectionProxyImpl(dataSource, conn, new Properties(), 0));
{
SQLException error = null;
try {
filter.connection_rollback(chain, new ConnectionProxyImpl(dataSource, conn, new Properties(), 0));
} catch (SQLException ex) {
error = ex;
}
Assert.assertNotNull(error);
}
{
RuntimeException error = null;
try {
filter.connection_rollback(chain, new ConnectionProxyImpl(dataSource, conn, new Properties(), 0));
} catch (RuntimeException ex) {
error = ex;
}
Assert.assertNotNull(error);
}
}
public void test_rollback_2() throws Exception {
FilterChainImpl chain = new FilterChainImpl(dataSource);
MockConnection conn = new MockConnection() {
private int i = 0;
public void rollback(Savepoint savepoint) throws SQLException {
int count = i++ % 3;
if (count % 3 == 1) {
throw new SQLException();
} else if (count % 3 == 2) {
throw new RuntimeException();
}
}
};
Savepoint savepoint = new Savepoint() {
@Override
public int getSavepointId() throws SQLException {
// TODO Auto-generated method stub
return 0;
}
@Override
public String getSavepointName() throws SQLException {
// TODO Auto-generated method stub
return null;
}
};
filter.connection_rollback(chain, new ConnectionProxyImpl(dataSource, conn, new Properties(), 0), savepoint);
{
SQLException error = null;
try {
filter.connection_rollback(chain, new ConnectionProxyImpl(dataSource, conn, new Properties(), 0),
savepoint);
} catch (SQLException ex) {
error = ex;
}
Assert.assertNotNull(error);
}
{
RuntimeException error = null;
try {
filter.connection_rollback(chain, new ConnectionProxyImpl(dataSource, conn, new Properties(), 0),
savepoint);
} catch (RuntimeException ex) {
error = ex;
}
Assert.assertNotNull(error);
}
}
}
src/test/java/com/alibaba/druid/bvt/filter/TraceFilterTest_connect_error.java
-72
View File
@@ -1,72 +0,0 @@
package com.alibaba.druid.bvt.filter;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Properties;
import junit.framework.Assert;
import junit.framework.TestCase;
import com.alibaba.druid.filter.trace.TraceFilter;
import com.alibaba.druid.mock.MockDriver;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.stat.DruidDataSourceStatManager;
import com.alibaba.druid.stat.JdbcStatContext;
import com.alibaba.druid.stat.JdbcStatManager;
public class TraceFilterTest_connect_error extends TestCase {
private MockDriver driver;
private DruidDataSource dataSource;
private TraceFilter filter;
protected void setUp() throws Exception {
Assert.assertEquals(0, DruidDataSourceStatManager.getInstance().getDataSourceList().size());
driver = new MockDriver() {
public Connection connect(String url, Properties info) throws SQLException {
throw new SQLException();
}
};
dataSource = new DruidDataSource();
dataSource.setUrl("jdbc:mock:xxx");
dataSource.setDriver(driver);
dataSource.setInitialSize(1);
dataSource.setMaxActive(2);
dataSource.setMaxIdle(2);
dataSource.setMinIdle(1);
dataSource.setMinEvictableIdleTimeMillis(300 * 1000); // 300 / 10
dataSource.setTimeBetweenEvictionRunsMillis(180 * 1000); // 180 / 10
dataSource.setTestWhileIdle(true);
dataSource.setTestOnBorrow(false);
dataSource.setValidationQuery("SELECT 1");
dataSource.setFilters("trace");
filter = (TraceFilter) dataSource.getProxyFilters().get(0);
JdbcStatContext statContext = new JdbcStatContext();
statContext.setTraceEnable(true);
JdbcStatManager.getInstance().setStatContext(statContext);
}
protected void tearDown() throws Exception {
dataSource.close();
Assert.assertEquals(0, DruidDataSourceStatManager.getInstance().getDataSourceList().size());
JdbcStatManager.getInstance().setStatContext(null);
}
public void test_exuecute() throws Exception {
{
SQLException error = null;
try {
dataSource.getConnection();
} catch (SQLException ex) {
error = ex;
}
Assert.assertNotNull(error);
}
}
}
src/test/java/com/alibaba/druid/bvt/filter/TraceFilterTest_connect_error_2.java
-72
View File
@@ -1,72 +0,0 @@
package com.alibaba.druid.bvt.filter;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.Properties;
import junit.framework.Assert;
import junit.framework.TestCase;
import com.alibaba.druid.filter.trace.TraceFilter;
import com.alibaba.druid.mock.MockDriver;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.stat.DruidDataSourceStatManager;
import com.alibaba.druid.stat.JdbcStatContext;
import com.alibaba.druid.stat.JdbcStatManager;
public class TraceFilterTest_connect_error_2 extends TestCase {
private MockDriver driver;
private DruidDataSource dataSource;
private TraceFilter filter;
protected void setUp() throws Exception {
Assert.assertEquals(0, DruidDataSourceStatManager.getInstance().getDataSourceList().size());
driver = new MockDriver() {
public Connection connect(String url, Properties info) throws SQLException {
throw new RuntimeException();
}
};
dataSource = new DruidDataSource();
dataSource.setUrl("jdbc:mock:xxx");
dataSource.setDriver(driver);
dataSource.setInitialSize(1);
dataSource.setMaxActive(2);
dataSource.setMaxIdle(2);
dataSource.setMinIdle(1);
dataSource.setMinEvictableIdleTimeMillis(300 * 1000); // 300 / 10
dataSource.setTimeBetweenEvictionRunsMillis(180 * 1000); // 180 / 10
dataSource.setTestWhileIdle(true);
dataSource.setTestOnBorrow(false);
dataSource.setValidationQuery("SELECT 1");
dataSource.setFilters("trace");
filter = (TraceFilter) dataSource.getProxyFilters().get(0);
JdbcStatContext statContext = new JdbcStatContext();
statContext.setTraceEnable(true);
JdbcStatManager.getInstance().setStatContext(statContext);
}
protected void tearDown() throws Exception {
dataSource.close();
Assert.assertEquals(0, DruidDataSourceStatManager.getInstance().getDataSourceList().size());
JdbcStatManager.getInstance().setStatContext(null);
}
public void test_exuecute() throws Exception {
{
Exception error = null;
try {
dataSource.getConnection();
} catch (Exception ex) {
error = ex;
}
Assert.assertNotNull(error);
}
}
}
src/test/java/com/alibaba/druid/bvt/pool/DruidDataSourceFilterTest.java
-12
View File
@@ -32,18 +32,6 @@ public class DruidDataSourceFilterTest extends TestCase {
dataSource.close();
}
public void test_filter_2() throws Exception {
DruidDataSource dataSource = new DruidDataSource();
Assert.assertEquals(0, dataSource.getProxyFilters().size());
dataSource.setFilters("stat,trace");
Assert.assertEquals(2, dataSource.getProxyFilters().size());
dataSource.close();
}
public void test_filter_3() throws Exception {
DruidDataSource dataSource = new DruidDataSource();
dataSource.setUrl("jdbc:mock:");