在上一篇中我們講述了百萬連接的
測試過程及出現的問題和解決辦法，有同學對其中的測試方法也有興趣，我在這里再補充一下。這里的測試方法主要是指測試客戶端如何發起百萬連接并保持它們。由于受到單機單ip的端口數限制，即如下
pomelo@debian:~/socktest$ cat /proc/sys/net/ipv4/ip_local_port_range 2048 65500 |
單個客戶端ip只能建立6萬多連接，所以我們需要大約50個獨立ip發起300萬的連接。為簡單起見，使用50臺512M內存的虛擬機作為客戶端。問題歸結為兩個方面
1. 單客戶端如何建立并保持6萬連接
2. 如何控制幾十個客戶端
我們以
java編寫客戶端代碼。首先我們嘗試的是一線程一連接的方式。這樣在java中就需要維護6萬個線程及其相關的對象，包括連接對象。很快我們就發現客戶端java進程出現了OOM，512M內存不夠用了。我們分析下測試目標，目的是保持6萬個連接對象，在這種方式下我們相當于是使用了線程這種容器來保存這些對象，但由于線程本身的內存開銷巨大，因此我們應該考慮換個輕量級的瓶子來裝這些連接對象。
最終代碼如下，即在一個線程內順序建立連接，并將連接保存到一個數組中，至此就解決了上文問題中的一個方面
private static int n = 0; private static Socket[] sockList = null; static { n = Integer.valueOf(Config.getProperty("socket.num")); sockList = new Socket[n]; } public void connect(String host, int port) throws UnknownHostException { Socket so = null; long sleepTime = Integer.valueOf(Config.getProperty("socket.sleep")); for (int i=0;i<n;i++) { try { so = new Socket(host, port); sockList[i] = so; Thread.sleep(sleepTime); } catch (IOException e) { logger.error(e.getMessage()); } catch (InterruptedException e) { } } } |
接著，又是如何控制幾十臺客戶端呢？其實比較簡單，就是在
shell腳本中通過ssh調用部署在客戶端機器上的測試程序。主要有兩個問題需要注意，一是ssh需要認證，我們可以在任意機器上產生無密碼的ssh公私鑰對，將公鑰分發到客戶端機器上。二是設置一個ip列表的配置文件。
這樣，兩個問題都解決了，可以開始測試了。測試過程及其問題不完全記錄可以點這里。