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

支持HA

Browse Source 
git-svn-id: http://code.alibabatech.com/svn/druid/trunk@469 b9813039-fb51-4c41-a8b9-e21c2acb5095
This commit is contained in:
wenshao
2011-07-17 10:26:28 +00:00
parent 6e362a137a
commit 09ca0375b2
6 changed files with 133 additions and 73 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/pool/ha/HADataSource.java
+4 -65
View File
@@ -6,22 +6,15 @@ import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.atomic.AtomicInteger;
import javax.sql.DataSource;
import com.alibaba.druid.logging.Log;
import com.alibaba.druid.logging.LogFactory;
import com.alibaba.druid.pool.DruidDataSource;
import com.alibaba.druid.pool.ha.balance.Balancer;
import com.alibaba.druid.pool.ha.balance.RoundRobinBlancer;
public class HADataSource extends MultiDataSource implements DataSource {
private final static Log LOG = LogFactory.getLog(HADataSource.class);
private final AtomicInteger indexErrorCount = new AtomicInteger();
protected ArrayList<DruidDataSource> dataSources = new ArrayList<DruidDataSource>();
protected final List<DruidDataSource> notAvailableDatasources = new CopyOnWriteArrayList<DruidDataSource>();
@@ -52,71 +45,17 @@ public class HADataSource extends MultiDataSource implements DataSource {
this.balancer = balancer;
}
protected int indexFor(MultiDataSourceConnection connection, String sql) throws SQLException {
return balancer.indexFor(connection, sql);
public Connection getConnectionInternal(MultiDataSourceConnection connection, String sql) throws SQLException {
return this.balancer.getConnection(connection, sql);
}
public Connection getConnectionInternal(MultiDataSourceConnection multiDataSourceConnection, String sql) throws SQLException {
int tryCount = 0;
for (;;) {
int size = dataSources.size();
if (size == 0) {
throw new SQLException("can not get connection, no availabe datasources");
}
int index = indexFor(multiDataSourceConnection, sql);
DruidDataSource dataSource = null;
try {
// 处理并发时的错误
dataSource = dataSources.get(index);
} catch (Exception ex) {
indexErrorCount.incrementAndGet();
if (LOG.isDebugEnabled()) {
LOG.debug("getDataSource error, index : " + index, ex);
}
continue;
}
assert dataSource != null;
if (!dataSource.isEnable()) {
handleNotAwailableDatasource(dataSource);
continue;
}
Connection conn = null;
try {
tryCount++;
conn = dataSource.getConnection();
} catch (SQLException ex) {
LOG.error("getConnection error", ex);
if (tryCount >= size) {
throw ex;
}
continue;
}
return conn;
}
}
void handleNotAwailableDatasource(DruidDataSource dataSource) {
public void handleNotAwailableDatasource(DruidDataSource dataSource) {
boolean removed = dataSources.remove(dataSource);
if (removed) {
notAvailableDatasources.add(dataSource);
}
}
public long getIndexErrorCount() {
return indexErrorCount.get();
}
}