用Java写压测程序（基于Socket）

最近，公司服务器需要做一个压力测试，公司的服务器，是基于Socket的连接，然后去进行数据的传输。

1 . 最开始，我是直接用传统的Socket作为连接，然后，一个程序创建N个Socket，因为普通的Socket是一个阻塞式的Socket，所以呢，每个Socket必须对应着一条线程！这样，从一开始的10线程并发（慢慢压上去）到后面的1000线程并发，逻辑上，测试程序很直观，一个线程发送了请求包，就阻塞等待回包，然后继续，到最后的1000个，2000个，结果发现超出JVM的限制了。所以，这条路，走不通了。

3 . 继续上网扒资料。普通的Socket因为是阻塞当前线程的，所以压测量一上去，其弊端就出来了，所以，不能是线程阻塞的！于是我就找到了不阻塞的Socket，使用异步回调的Socket，嗯，很棒！在这里我粗略介绍一下这个异步的Socket，其位于：java.nio.channels.AsynchronousSocketChannelNIO包：
AsynchronousSocketChannel异步Socket，其连接，发送，接收，都是异步的，其异步回调的接口如下：(代码来了，大家不要怕)

public interface CompletionHandler 
{
        void completed(V result, A attachment);    
        void failed(Throwable exc, A attachment);
}

这不是一个函数式接口，而且也没有提供者接口的闭包，嗯，有点桑心。
从函数的名称看，completed就是完成时的回调，failed就是失败的回调。
成功的回调，有V result，这个东西，在AsynchronousSocketChannel的读取函数，规定了V必须是Integer，这个就是告诉你读到了多少个字符。
而，其他没规定，然后呢，A这个泛型我们就可以自定义啦，一般来说，是传入AsynchronousSocketChannel的对象，到如果，你需要给Socket加上一些自定义的属性，你可以去DIY一个SocketBean类去包装。嗯，好像说的有点多了，就先这样，后期这个方案我也会写篇文章纪念一下。
又过了快一天（脑子笨+网上资料有点少），我又弄出了一个异步Socket的压测，然后，我想到一个问题，它这个回调回是在哪里回调呢？在原线程吗？还是在新开线程？如果是新开线程中回调，就又GG了。。。结果，很悲剧...我竟然忽略这个问题，TMD在新线程中回调...如果，我压测量又上了2000，那，2K个发包，2K个回包，且在新开线程中！！！好了，瓶颈又回到了方案一二的瓶颈....

4 .哎，这次真的心累了。不过我这次慢慢地从JAVA AIP文档查起来，结果发现有个很棒的东西：
Java NIO系列教程（六） Selector
这个NIO中的Selector，引用一下别人的话：

Selector（选择器）是Java NIO中能够检测一到多个NIO通道，并能够知晓通道是否为诸如读写事件做好准备的组件。这样，一个单独的线程可以管理多个channel，从而管理多个网络连接。
我...我...欣喜若狂，这就是我要找的Socket。

这种Socket位于：java.nio.channels.SocketChannel，看来NIO包很重要有木有！这个网上也是很少资料，坑也是一个一个的，弄得我不要不要的！（后面会出文章说明）
通过Selector，去管理N个Socket，注意，这个Selector是跑在一条阻塞的线程上，而其就只需要一条线程！就能管理N个连接。
通过这个东西，进过1半天的努力，我终于弄粗了一个压测4.0版本，然后随便改改一个参数，就上千上万了嘻嘻。

这几天的经历很棒，从一次一次的期望到失望，到一次一次怀疑人生，我都熬过来了，其实也没什么了不起的大事情。加油！写这篇文章，主要是纪念一下这几天，和预告一下我要写一下几篇技术文章（其实很水），因为这方面网上很少资料！