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

1.概述

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

2.然而，压测程序的书写却让我吃力。（以下是我走过的方案）

---

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

---

• 2 . 然后，我就到处找资料，看到网上有一些专业的压测程序：例如，JMeter，LoadRunner等等。然后我就开始研究了。LoadRunner非常专业，但是专业，就很难短时间入手了，所以我选择了比较容易入门的JMeter。
  这里介绍一下我入门推介的文章（后期我会亲手写一下遇到的坑）：
  JMeter学习（十六）JMeter函数学习
  JMeter学习（十七）JMeter测试Java
  JMeter学习（十八）JMeter测试Java（二）
  这一系列文章能很好的带你入门！
  在这个方案上，我挣扎了一天！到了晚上，终于出来的压测程序！但是，结果却不如人意！（这里我想骂人）不过是我自己没先弄清楚。我以为这个压测软件是通过别的方法，大概是线程复用啥的，结果跑起来，发现还是跟方案一一样！跑到了2K个并发后，就报JVM达到线程数上限了。（欲哭无泪）大概经过了：怀疑自己-怀疑人生-自暴自弃-重新站起来，我继续站了起来，不哭继续撸！

---

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

public interface CompletionHandler<V,A> 
{
        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版本，然后随便改改一个参数，就上千上万了嘻嘻。

---

3 .总结

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

• 希望有人期待。（我自己就很期待）