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.AsynchronousSocketChannel
NIO包:
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 .总结
这几天的经历很棒,从一次一次的期望到失望,到一次一次怀疑人生,我都熬过来了,其实也没什么了不起的大事情。加油!写这篇文章,主要是纪念一下这几天,和预告一下我要写一下几篇技术文章(其实很水),因为这方面网上很少资料!
- 希望有人期待。(我自己就很期待)