用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 .总结

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

  • 希望有人期待。(我自己就很期待)
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 211,194评论 6 490
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,058评论 2 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 156,780评论 0 346
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,388评论 1 283
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,430评论 5 384
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,764评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,907评论 3 406
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,679评论 0 266
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,122评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,459评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,605评论 1 340
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,270评论 4 329
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,867评论 3 312
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,734评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,961评论 1 265
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,297评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,472评论 2 348

推荐阅读更多精彩内容