83 lines
2.7 KiB
Java
83 lines
2.7 KiB
Java
package org.csource.fastdfs.pool;
|
|
|
|
import org.csource.common.MyException;
|
|
|
|
import java.io.IOException;
|
|
import java.net.InetSocketAddress;
|
|
import java.util.Map;
|
|
import java.util.concurrent.ConcurrentHashMap;
|
|
|
|
public class ConnectionPool {
|
|
/**
|
|
* key is ip:port, value is ConnectionManager
|
|
*/
|
|
private final static ConcurrentHashMap<String, ConnectionManager> CP = new ConcurrentHashMap<String, ConnectionManager>();
|
|
|
|
public static Connection getConnection(InetSocketAddress socketAddress) throws MyException, IOException {
|
|
if (socketAddress == null) {
|
|
return null;
|
|
}
|
|
String key = getKey(socketAddress);
|
|
ConnectionManager connectionManager;
|
|
synchronized (ConnectionPool.class) {
|
|
connectionManager = CP.get(key);
|
|
if (connectionManager == null) {
|
|
connectionManager = new ConnectionManager(socketAddress);
|
|
CP.put(key, connectionManager);
|
|
}
|
|
}
|
|
return connectionManager.getConnection();
|
|
}
|
|
|
|
public static void releaseConnection(Connection connection) throws IOException {
|
|
if (connection == null) {
|
|
return;
|
|
}
|
|
String key = getKey(connection.getInetSocketAddress());
|
|
ConnectionManager connectionManager = CP.get(key);
|
|
if (connectionManager != null) {
|
|
connectionManager.releaseConnection(connection);
|
|
} else {
|
|
try {
|
|
connection.closeDirectly();
|
|
} catch (IOException e) {
|
|
System.err.println("close socket error, msg:" + e.getMessage());
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
public static void closeConnection(Connection connection) throws IOException {
|
|
if (connection == null) {
|
|
return;
|
|
}
|
|
String key = getKey(connection.getInetSocketAddress());
|
|
ConnectionManager connectionManager = CP.get(key);
|
|
if (connectionManager != null) {
|
|
connectionManager.closeConnection(connection);
|
|
} else {
|
|
connection.closeDirectly();
|
|
}
|
|
}
|
|
|
|
private static String getKey(InetSocketAddress socketAddress) {
|
|
if (socketAddress == null) {
|
|
return null;
|
|
}
|
|
return String.format("%s:%s", socketAddress.getHostName(), socketAddress.getPort());
|
|
}
|
|
|
|
@Override
|
|
public String toString() {
|
|
if (!CP.isEmpty()) {
|
|
StringBuilder builder = new StringBuilder();
|
|
for (Map.Entry<String, ConnectionManager> managerEntry : CP.entrySet()) {
|
|
builder.append("key:[" + managerEntry.getKey() + " ]-------- entry:" + managerEntry.getValue() + "\n");
|
|
}
|
|
return builder.toString();
|
|
}
|
|
return null;
|
|
}
|
|
}
|