ConcurrentHashMap实现数据库连接的线程安全问题

1. private ConcurrentHashMap<String,FutureTask<Connection>>connectionPool = newConcurrentHashMap<String, FutureTask<Connection>>();
3. public Connection getConnection(String key) throws Exception{
4. FutureTask<Connection>connectionTask=connectionPool.get(key);
5. if(connectionTask!=null){
6. return connectionTask.get();
7. }
8. else{
9. Callable<Connection> callable = new Callable<Connection>(){
10. @Override
11. public Connection call() throws Exception {
12. // TODO Auto-generated method stub
13. return createConnection();
14. }
15. };
16. FutureTask<Connection>newTask = new FutureTask<Connection>(callable);
17. connectionTask = connectionPool.putIfAbsent(key, newTask);
18. if(connectionTask==null){
19. connectionTask = newTask;
20. connectionTask.run();
21. }
22. return connectionTask.get();
23. }
24. }
26. //创建Connection
27. private Connection createConnection(){
28. return null;
29. }
30. 
    

1. 这是在网上看到的一段 ConcurrentHashMap实现数据库连接的代码，发现有个问题：
2. 就是在线程A、B同时执行获取数据库连接的方法时，如果线程A执行完17行，返回null，说明它是首个存入该key的线程，但是不幸的是A线程因为某些原因此时突然挂掉了。线程B执行到17行时发现此时已经有线程存入了key，connectionTask不为null，则直接跳到22行去get数据库连接，但是因为线程A没有执行完创建连接，所以线程B就一直阻塞在get方法上。请各位大神看看是不是存在这个问题，如果存在，如何更好的解决？

问题解决：在可能出现异常的地方添加一个异常捕获，如果发生异常就将之前添加的key-value删除，我理解这样就可避免线程无限阻塞的问题了，各位看看还有没有其他问题，欢迎留言

    try {
        if (connectionFutureTask==null) {
            connectionFutureTask = newTask;
            connectionFutureTask.run();
        }
    } catch (Exception e) {
        connectionPool.remove(key, newTask);
        e.printStackTrace();
        throw new RuntimeException("connectionCreateFail!!!");
    }
    return connectionFutureTask.get();