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

improve MSSQLValidConnectionChecker & add testcase

Browse Source
This commit is contained in:
wenshao
2013-06-15 12:05:39 +08:00
parent 3fb946946b
commit 47e60c8fc8
2 changed files with 109 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/pool/vendor/MSSQLValidConnectionChecker.java
Vendored
+69 -66
View File
@@ -1,66 +1,69 @@
/*
* Copyright 1999-2011 Alibaba Group Holding Ltd.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.alibaba.druid.pool.vendor;
import java.io.Serializable;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Statement;
import com.alibaba.druid.pool.ValidConnectionChecker;
import com.alibaba.druid.pool.ValidConnectionCheckerAdapter;
import com.alibaba.druid.support.logging.Log;
import com.alibaba.druid.support.logging.LogFactory;
import com.alibaba.druid.util.JdbcUtils;
/**
* A MSSQLValidConnectionChecker.
*/
public class MSSQLValidConnectionChecker extends ValidConnectionCheckerAdapter implements ValidConnectionChecker, Serializable {
private static final long serialVersionUID = 1L;
private static final String QUERY = "SELECT 1";
private static final Log LOG = LogFactory.getLog(MSSQLValidConnectionChecker.class);
public boolean isValidConnection(final Connection c, String valiateQuery, int validationQueryTimeout) {
try {
if (c.isClosed()) {
return false;
}
} catch (SQLException ex) {
// skip
return false;
}
Statement stmt = null;
try {
stmt = c.createStatement();
stmt.setQueryTimeout(validationQueryTimeout);
stmt.execute(QUERY);
return true;
} catch (SQLException e) {
if (LOG.isWarnEnabled()) {
LOG.warn("warning: connection validation failed for current managed connection.");
}
return false;
} finally {
JdbcUtils.close(stmt);
}
}
}
/*
* Copyright 1999-2011 Alibaba Group Holding Ltd.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.alibaba.druid.pool.vendor;
import java.io.Serializable;
import java.sql.Connection;
import java.sql.SQLException;
import java.sql.Statement;
import com.alibaba.druid.pool.ValidConnectionChecker;
import com.alibaba.druid.pool.ValidConnectionCheckerAdapter;
import com.alibaba.druid.support.logging.Log;
import com.alibaba.druid.support.logging.LogFactory;
import com.alibaba.druid.util.JdbcUtils;
/**
* A MSSQLValidConnectionChecker.
*/
public class MSSQLValidConnectionChecker extends ValidConnectionCheckerAdapter implements ValidConnectionChecker, Serializable {
private static final long serialVersionUID = 1L;
private static final Log LOG = LogFactory.getLog(MSSQLValidConnectionChecker.class);
public MSSQLValidConnectionChecker(){
}
public boolean isValidConnection(final Connection c, String valiateQuery, int validationQueryTimeout) {
try {
if (c.isClosed()) {
return false;
}
} catch (SQLException ex) {
// skip
return false;
}
Statement stmt = null;
try {
stmt = c.createStatement();
stmt.setQueryTimeout(validationQueryTimeout);
stmt.execute(valiateQuery);
return true;
} catch (SQLException e) {
if (LOG.isWarnEnabled()) {
LOG.warn("warning: connection validation failed for current managed connection.");
}
return false;
} finally {
JdbcUtils.close(stmt);
}
}
}
src/test/java/com/alibaba/druid/bvt/pool/vendor/MSSQLValidConnectionCheckerTest.java
Vendored
+40
View File
@@ -0,0 +1,40 @@
package com.alibaba.druid.bvt.pool.vendor;
import java.sql.SQLException;
import junit.framework.TestCase;
import org.junit.Assert;
import com.alibaba.druid.mock.MockConnection;
import com.alibaba.druid.pool.vendor.MSSQLValidConnectionChecker;
public class MSSQLValidConnectionCheckerTest extends TestCase {
public void test_0() throws Exception {
MSSQLValidConnectionChecker checker = new MSSQLValidConnectionChecker();
MockConnection conn = new MockConnection();
Assert.assertTrue(checker.isValidConnection(conn, "select 1", 10));
}
public void test_closed() throws Exception {
MSSQLValidConnectionChecker checker = new MSSQLValidConnectionChecker();
MockConnection conn = new MockConnection();
conn.close();
Assert.assertFalse(checker.isValidConnection(conn, "select 1", 10));
}
public void test_error() throws Exception {
MSSQLValidConnectionChecker checker = new MSSQLValidConnectionChecker();
MockConnection conn = new MockConnection();
conn.setError(new SQLException());
Assert.assertFalse(checker.isValidConnection(conn, "select 1", 10));
}
}