JDK6.0学习笔记（十）TCP连接池

1. /**
2.  * TCP开发 连接池
3.  * */
5. import java.net.Socket;
6. import java.util.Hashtable;
8. public class ConnectionPool {
9.     private static final int CONNECTION_POOL_SIZE = 10;
11.     private static final String API_SERVER_HOST = "127.0.0.01";
13.     private static final int API_SERVER_PORT = 80;
15.     private static ConnectionPool self = null;// ConnectionPool的唯一实例
17.     private Hashtable socketPool = null;// 连接池
19.     private boolean[] socketStatusArray = null;// 连接的状态（true-被占用；false-空闲）
21.     public static synchronized void init() {
22.         self = new ConnectionPool();
23.         self.socketPool = new Hashtable();
24.         self.socketStatusArray = new boolean[CONNECTION_POOL_SIZE];
26.         // 初始化连接池
27.         self.buildConnectionPool();
28.     }
30.     public static synchronized void reset() {
31.         self = null;
32.         init();
33.     }
35.     public static Socket getConnection() {
36.         if (self == null)
37.             init();
39.         int i = 0;
40.         for (i = 0; i < CONNECTION_POOL_SIZE; i++) {
41.             if (!self.socketStatusArray[i]) {
42.                 self.socketStatusArray[i] = true;
43.                 break;
44.             }
45.         }
46.         if (i <= CONNECTION_POOL_SIZE)
47.             return (Socket) self.socketPool.get(new Integer(i));
48.         else {
49.             System.out.println("从连接池中获取与邮局的连接失败，已经没有空闲连接！");
50.             throw new RuntimeException("No enough pooled connections.");
51.         }
52.     }
54.     public static void releaseConnection(Socket socket) {
55.         if (self == null)
56.             init();
58.         for (int i = 0; i < CONNECTION_POOL_SIZE; i++) {
59.             if (((Socket) self.socketPool.get(new Integer(i))) == socket) {
60.                 self.socketStatusArray[i] = false;
61.                 break;
62.             }
63.         }
64.     }
66.     public static Socket rebuildConnection(Socket socket) {
67.         if (self == null)
68.             init();
70.         Socket newSocket = null;
71.         for (int i = 0; i < CONNECTION_POOL_SIZE; i++) {
72.             try {
73.                 if (((Socket) self.socketPool.get(new Integer(i))) == socket) {
74.                     System.out.println("重建连接池中的第" + i + "个连接.");
75.                     newSocket = new Socket(API_SERVER_HOST, API_SERVER_PORT);
76.                     self.socketPool.put(new Integer(i), newSocket);
77.                     self.socketStatusArray[i] = true;
78.                     break;
79.                 }
80.             } catch (Exception e) {
81.                 System.out.println("重建连接失败！");
82.                 throw new RuntimeException(e);
83.             }
84.         }
85.         return newSocket;
86.     }
88.     public synchronized static void buildConnectionPool() {
89.         if (self == null)
90.             init();
92.         System.out.println("准备建立连接池.");
93.         Socket socket = null;
94.         try {
95.             for (int i = 0; i < CONNECTION_POOL_SIZE; i++) {
96.                 socket = new Socket(API_SERVER_HOST, API_SERVER_PORT);
97.                 self.socketPool.put(new Integer(i), socket);
98.                 self.socketStatusArray[i] = false;
99.             }
100.         } catch (Exception e) {
101.             System.out.println("与邮局的连接池建立失败！");
102.             throw new RuntimeException(e);
103.         }
104.     }
106.     public synchronized static void releaseAllConnection() {
107.         if (self == null)
108.             init();
110.         // 关闭所有连接
111.         Socket socket = null;
112.         for (int i = 0; i < CONNECTION_POOL_SIZE; i++) {
113.             socket = (Socket) self.socketPool.get(new Integer(i));
114.             try {
115.                 socket.close();
116.             } catch (Exception e) {
117.             }
118.         }
119.     }
120. }