使用gopacket来实现mysql的流量录制和回放

背景

丰巢作为一家快速发展的科技公司,在我们平时的工作中,有很多的数据库迁移、改造、测试等工作。在我们之前做的一次异构数据库迁移过程中,我们投入了相对较多的人力。我本人一直在思考如何投入最少的人力,并且能使这个过程更加安全稳定,其中一个技术难点便是对于生产环境的流量在测试环境进行持续的回放,并在这个过程中发现问题所在。

技术选型

一开始本着不重复造轮子的原则,想在开源领域找到一款合适我们的产品。主要调研了tcpcopy框架,但是它不能满足我司的实际情况:

  • 我们后台服务使用的都是长连接技术,tcpcopy只有等到下一次连接登录时才能完成实际的链路创建,这个在我们的环境中完全行不通,我们需要所有录制的信息都可以100%被回放;
  • 我们有很多的场景,库名和表名的映射都发生了变化,tcpcopy等现有技术无法解决我们的这个问题;
  • 我们希望在录制和回放的过程中,不但能够校验是否发生错误、响应延迟等,还能对response做校验,甚至是对两个数据库的数据一致性做校验;

为了满足上面3个要求,我们开始了自研之路。在开始之前,我们需要一款能从数据链路层抓包的框架,我们选择了google开源的gopacket,因为我们之前使用gopacket实现过redis的实时命令分析工具,因此对它非常有信心。

功能分解

我对这款工具的定位是可以持续的输出业务价值,因此我希望它能最短的时间里便能产生作用,在制定第一期的功能列表时,本着不求大而全,只求有用的原则:

  • mysql协议理解:目前版本中只需理解登录、命令执行和连接关闭即可,非常简单;
  • 命令解析及输出:直接输出实际的sql文本命令,命令会标识出具体是哪个连接的命令及发送的时间;
  • 回放功能:支持原速回放和倍速回放;
  • 延迟和错误检测;

技术实现

gopacket使用

gopacket的使用非常简单,只需要在go.mod中引用gopacket的最新版本即可:

require github.com/google/gopacket v1.1.17

代码示例:

    devices, err := pcap.FindAllDevs()
    if err != nil {
        log.Fatal(err)
    }

    for _, deviceGet := range devices {
        for _, address := range deviceGet.Addresses {
            if address.IP.String() == *ip {
                device = deviceGet.Name
                break
            }
        }
    }
    if device == "" {
        fmt.Println("the ip don't match the network device. maybe you don't have the sudo authority.")
        return
    }
    
    handle, err = pcap.OpenLive(device, snapshot_len, true, timeout)
    if err != nil {
        log.Fatal(err)
    }
    defer handle.Close()

    var filter string = "tcp and dst host "
    filter = filter + *ip + " and dst port " + strconv.Itoa(*port)

    err = handle.SetBPFFilter(filter)
    if err != nil {
        log.Fatal(err)
    }
    
    packetSource := gopacket.NewPacketSource(handle, handle.LinkType())
    var origin []byte
    for packet := range packetSource.Packets() {
        //do something
    }

倍速实现

关于回放逻辑中的倍速实现,我们需要知道几个参数:流量录制时的开始时间recordB、结束时间recordE、命令的执行时间recordC、流量回放时的开始时间replayB,通过这些参数我们就可以推断出命令在命令在回放时的实际执行时间replayC,公式如下,speed为放大的倍数,默认为1:

replayC = (recordC+(replayB-recordB)-replayB) /speed + replayB

效果

大家也可以看出,我们在第一期 实现中,逻辑都比较简单,开发时间也就只花了5天左右的时间,我们使用此工具对我司支付平台的生产环境mysql流量进行了录制,并在测试环境中的TiDB上进行了回放,共发现了两个问题:

  • TiKV的一个已经解决的bug,我司目前tidb的版本是2.1.9,当transfer leader时遇到了conf change时,会出现短暂的慢查询现象(连接会被TiDB占住一段时间)PR详见
  • nginx在做tcp proxy时,会偶尔出现连接卡死的现象,只有到达超时时间,连接被kill后,此连接的业务才会恢复。我司之前使用的是第三方的nginx_tcp_proxy_module的模块。后来把nginx的版本升级为1.17.1并使用nginx官方的stream模块,此问题也解决了;

未来

我对这个工具还是很期待的,我希望它能够在资源可控的条件下,实时跑在生产环境mysql的实例服务器上,总结一下后续的计划:

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

推荐阅读更多精彩内容