为TiDB DM添加阿里云RDS/DMS Online DDL支持

Foreword

停更了好几个月,在百忙之中抽点时间写篇小的流水账吧。

Online DDL即无锁表结构变更,能够避免对表(尤其是大表)进行更改时,长时间阻塞DML操作。我们当前采用TiDB的DM组件实现上游许多业务库的合库合表与数据汇聚,DM原生支持的Online DDL工具有pt-osc(Percona开源)与gh-ost(GitHub开源)两种。但是,我们的业务库绝大多数都是阿里云RDS MySQL,采用其DMS工具做无锁变更。这导致我们每次都需要手动执行DDL语句,再指定位点恢复DM任务,操作起来比较繁琐,所以还是自己动手丰衣足食比较好。

gh-ost/DMS Online DDL Procedure

通过观察DMS Online DDL产生的binlog,可以发现它的风格与gh-ost相同。简单复习一下gh-ost的处理流程:

  1. 在目标库上创建影子表(ghost table,后缀为_gho),它的结构与被变更的原表完全相同;
  2. 创建日志表(log table,后缀为_ghc)用于记录整个DDL执行过程中的状态;
  3. 在影子表上执行DDL语句;
  4. gh-ost将自身伪装成MySQL slave,创建binlog streamer,将原表的全量数据和增量binlog迁移到影子表(两个操作同时进行);
  5. 迁移完毕后,将原表锁表,并重命名加上_del后缀,再将影子表重命名去掉_gho后缀,完成cut-over。两个重命名操作是原子性执行的;
  6. 删除_ghc_del表,并关闭binlog streamer。

原子性cut-over的思想十分巧妙,基于MySQL内部锁表之后,执行RENAME操作的优先级比任何DML都要高这一简单的原理。具体的分析可参见国外大佬的这篇文章

阿里云DMS的执行流程与上述一样,只是表的命名有所变化而已:

  • 影子表:tp_[id]_ogt_[table]
  • 日志表:tp_[id]_ogl_[table]
  • 删除表:tp_[id]_del_[table]

下面直接上代码。

Hacking to the Code

DM将Online DDL过程中的表分为3类,代码文件online_ddl.go中的定义是:

type TableType string

const (
    realTable  TableType = "real table"
    ghostTable TableType = "ghost table"
    trashTable TableType = "trash table" // means we should ignore these tables
)
  • real table:原始表;
  • ghost table:影子表;
  • trash table:日志表与删除表。

另外还会通过OnlineDDLStorage结构体来维护执行过程中的必要信息,如数据库连接、schema/table名称、DDL语句等。

type OnlineDDLStorage struct {
    sync.RWMutex

    cfg *config.SubTaskConfig

    db        *conn.BaseDB
    dbConn    *DBConn
    schema    string // schema name, set through task config
    tableName string // table name with schema, now it's task name
    id        string // the source ID of the upstream MySQL/MariaDB replica.

    // map ghost schema => [ghost table => ghost ddl info, ...]
    ddls map[string]map[string]*GhostDDLInfo

    logCtx *tcontext.Context
}

DM内部的Online DDL模块是插件化的,只需实现OnlinePlugin接口即可。该接口的定义如下,注释写得比较清楚,笔者就不多废话了。

// OnlinePlugin handles online ddl solutions like pt, gh-ost.
type OnlinePlugin interface {
    // Apply does:
    // * detect online ddl
    // * record changes
    // * apply online ddl on real table
    // returns sqls, replaced/self schema, replaced/self table, error
    Apply(tctx *tcontext.Context, tables []*filter.Table, statement string, stmt ast.StmtNode) ([]string, string, string, error)
    // Finish would delete online ddl from memory and storage
    Finish(tctx *tcontext.Context, schema, table string) error
    // TableType returns ghhost/real table
    TableType(table string) TableType
    // RealName returns real table name that removed ghost suffix and handled by table router
    RealName(schema, table string) (string, string)
    // ResetConn reset db connection
    ResetConn(tctx *tcontext.Context) error
    // Clear clears all online information
    // TODO: not used now, check if we could remove it later
    Clear(tctx *tcontext.Context) error
    // Close closes online ddl plugin
    Close()
}

我们需要重点实现的方法是Apply()、TableType()和RealName()。后两者的逻辑比较简单,代码如下。

func (r *AliRDS) TableType(table string) TableType {
    if len(table) > 8 && strings.HasPrefix(table, "tp_") {
        if strings.Contains(table, "_ogt_") {
            return ghostTable
        }

        if strings.Contains(table, "_ogl_") || strings.Contains(table, "_del_") {
            return trashTable
        }
    }

    return realTable
}

func (r *AliRDS) RealName(schema, table string) (string, string) {
    tp := r.TableType(table)
    idx := -1

    if tp == ghostTable {
        idx = strings.Index(table, "_ogt_")
    } else if tp == trashTable {
        idx = strings.Index(table, "_ogl_")
        if idx == -1 {
            idx = strings.Index(table, "_del_")
        }
    }

    if idx > 0 {
        table = table[idx+5:]
    }
    return schema, table
}

接下来思考如何将DMS的步骤转化成TiDB的处理方式:

  • 只执行原表上的DML操作;
  • 不创建日志表;
  • 不创建影子表,但是将要执行的DDL记录到DM元数据库(默认为dm_meta)以及DM-Worker的内存里;
  • 在cut-over阶段,将上述记录的DDL的影子表名替换成原表名,直接执行替换后的DDL。

然后就可以顺理成章地写出Apply()方法了。

func (r *AliRDS) Apply(tctx *tcontext.Context, tables []*filter.Table, statement string, stmt ast.StmtNode) ([]string, string, string, error) {
    if len(tables) < 1 {
        return nil, "", "", terror.ErrSyncerUnitAliRDSApplyEmptyTable.Generate()
    }

    schema, table := tables[0].Schema, tables[0].Name
    targetSchema, targetTable := r.RealName(schema, table)
    tp := r.TableType(table)

    switch tp {
    case realTable:
        switch stmt.(type) {
        case *ast.RenameTableStmt:
            if len(tables) != parserpkg.SingleRenameTableNameNum {
                return nil, "", "", terror.ErrSyncerUnitAliRDSRenameTableNotValid.Generate()
            }

            tp1 := r.TableType(tables[1].Name)
            if tp1 == trashTable {
                return nil, "", "", nil
            } else if tp1 == ghostTable {
                return nil, "", "", terror.ErrSyncerUnitAliRDSRenameToGhostTable.Generate(statement)
            }
        }
        return []string{statement}, schema, table, nil

    case trashTable:
        switch stmt.(type) {
        case *ast.RenameTableStmt:
            if len(tables) != parserpkg.SingleRenameTableNameNum {
                return nil, "", "", terror.ErrSyncerUnitAliRDSRenameTableNotValid.Generate()
            }

            tp1 := r.TableType(tables[1].Name)
            if tp1 == ghostTable {
                return nil, "", "", terror.ErrSyncerUnitAliRDSRenameGhostTblToOther.Generate(statement)
            }
        }

    case ghostTable:
        switch stmt.(type) {
        case *ast.CreateTableStmt:
            err := r.storage.Delete(tctx, schema, table)
            if err != nil {
                return nil, "", "", err
            }
        case *ast.DropTableStmt:
            err := r.storage.Delete(tctx, schema, table)
            if err != nil {
                return nil, "", "", err
            }
        case *ast.RenameTableStmt:
            if len(tables) != parserpkg.SingleRenameTableNameNum {
                return nil, "", "", terror.ErrSyncerUnitAliRDSRenameTableNotValid.Generate()
            }

            tp1 := r.TableType(tables[1].Name)
            if tp1 == realTable {
                rdsInfo := r.storage.Get(schema, table)
                if rdsInfo != nil {
                    return rdsInfo.DDLs, tables[1].Schema, tables[1].Name, nil
                }
                return nil, "", "", terror.ErrSyncerUnitAliRDSOnlineDDLOnGhostTbl.Generate(schema, table)
            } else if tp1 == ghostTable {
                return nil, "", "", terror.ErrSyncerUnitAliRDSRenameGhostTblToOther.Generate(statement)
            }

            err := r.storage.Delete(tctx, schema, table)
            if err != nil {
                return nil, "", "", err
            }
        }

    default:
        err := r.storage.Save(tctx, schema, table, targetSchema, targetTable, statement)
        if err != nil {
            return nil, "", "", err
        }
    }

    return nil, schema, table, nil
}

最后不要忘了添加Online DDL配置项的定义,以及相关的错误信息等边角代码,不一一列举了。

var OnlineDDLSchemes = map[string]func(*tcontext.Context, *config.SubTaskConfig) (OnlinePlugin, error){
    config.PT:    NewPT,
    config.GHOST: NewGhost,
    config.ALIRDS: NewAliRDS,
}

const (
    GHOST  = "gh-ost"
    PT     = "pt"
    ALIRDS = "ali-rds"
)

重新编译dm-master与dm-worker二进制文件,并替换掉原本通过TiUP部署的文件(具体路径因人而异)。

CGO_ENABLED=0 GOOS=linux GOARCH=amd64 make dm-master
CGO_ENABLED=0 GOOS=linux GOARCH=amd64 make dm-worker

在DM作业的配置文件中指定online-ddl-scheme参数。

online-ddl-scheme: "ali-rds"

使用DMS工具做无锁表变更操作,可以发现能够正常同步了。

The End

继续搬砖去了。

Enjoy~

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

推荐阅读更多精彩内容

  • author:sufei源码版本:8.0.18 说明:如果对于online ddl还不了解可以参考 一、Onlin...
    真之棒2016阅读 626评论 0 1
  • 来自公众号:51CTO技术栈作者:贺磊 当一张百亿数据量的表放在你面前,你将面临着什么?加列?哭吧,怎么也得等个几...
    夜空_2cd3阅读 327评论 0 4
  • 作者:贺磊链接:https://blog.51cto.com/suifu/2476758 我先说几个最让你兴奋和开...
    码农小光阅读 2,533评论 1 50
  • 我是黑夜里大雨纷飞的人啊 1 “又到一年六月,有人笑有人哭,有人欢乐有人忧愁,有人惊喜有人失落,有的觉得收获满满有...
    陌忘宇阅读 8,471评论 28 53
  • 信任包括信任自己和信任他人 很多时候,很多事情,失败、遗憾、错过,源于不自信,不信任他人 觉得自己做不成,别人做不...
    吴氵晃阅读 6,133评论 4 8