摘要：作者張學(xué)程本文為源碼閱讀系列文章的第二篇，第一篇文章簡單介紹了源碼閱讀的目的和規(guī)劃，以及的源碼結(jié)構(gòu)以及工具鏈。通過注冊，并將該作為各的。在本篇文章中，我們暫時只關(guān)注架構(gòu)相關(guān)的實現(xiàn)，上述各功能的具體實現(xiàn)將在后續(xù)的相關(guān)文章中展開介紹。

作者：張學(xué)程

本文為 DM 源碼閱讀系列文章的第二篇，第一篇文章 簡單介紹了 DM 源碼閱讀的目的和規(guī)劃，以及 DM 的源碼結(jié)構(gòu)以及工具鏈。從本篇文章開始，我們會正式開始閱讀 DM 的源碼。

本篇文章主要介紹 DM 的整體架構(gòu)，包括 DM 有哪些組件、各組件分別實現(xiàn)什么功能、組件之間交互的數(shù)據(jù)模型和 RPC 實現(xiàn)。

通過上面的 DM 架構(gòu)圖，我們可以看出，除上下游數(shù)據(jù)庫及 Prometheus 監(jiān)控組件外，DM 自身有 DM-master、DM-worker 及 dmctl 這 3 個組件。其中，DM-master 負(fù)責(zé)管理和調(diào)度數(shù)據(jù)同步任務(wù)的各項操作，DM-worker 負(fù)責(zé)執(zhí)行具體的數(shù)據(jù)同步任務(wù)，dmctl 提供用于管理 DM 集群與數(shù)據(jù)同步任務(wù)的各項命令。

DM-master 的入口代碼在 cmd/dm-master/main.go，其中主要操作包括：

調(diào)用 cfg.Parse 解析命令行參數(shù)與參數(shù)配置文件

調(diào)用 log.SetLevelByString 設(shè)置進(jìn)程的 log 輸出級別

調(diào)用 signal.Notify 注冊系統(tǒng) signal 通知，用于接受到指定信號時退出進(jìn)程等

調(diào)用 server.Start 啟動 RPC server，用于響應(yīng)來自 dmctl 與 DM-worker 的請求

在上面的操作中，可以看出其中最關(guān)鍵的是步驟 4，其對應(yīng)的實現(xiàn)代碼在 dm/master/server.go 中，其核心為 Server 這個 struct，其中的主要 fields 包括：

rootLis, svr：監(jiān)聽網(wǎng)絡(luò)連接，分發(fā) RPC 請求給對應(yīng)的 handler。

workerClients：維護(hù)集群各 DM-worker ID 到對應(yīng)的 RPC client 的映射關(guān)系。

taskWorkers：維護(hù)用于執(zhí)行各同步（子）任務(wù)的 DM-worker ID 列表。

lockKeeper：管理在協(xié)調(diào)處理 sharding DDL 時的 lock 信息。

sqlOperatorHolder：管理手動 skip/replace 指定 sharding DDL 時的 SQL operator 信息。

在本篇文章中，我們暫時不會關(guān)注 lockKeeper 與 sqlOperatorHolder，其具體的功能與代碼實現(xiàn)會在后續(xù)相關(guān)文章中進(jìn)行介紹。

在 DM-master Server 的入口方法 Start 中：

通過 net.Listen 初始化 rootLis 并用于監(jiān)聽 TCP 連接（借助 soheilhy/cmux，我們在同一個 port 同時提供 gRPC 與 HTTP 服務(wù)）。

根據(jù)讀取的配置信息（DeployMap），初始化用于連接到各 DM-worker 的 RPC client 并保存在 workerClients 中。

通過 pb.RegisterMasterServer 注冊 gRPC server（svr），并將該 Server 作為各 services 的 implementation。

調(diào)用 m.Serve 開始提供服務(wù)。

DM-master 提供的 RPC 服務(wù)包括 DM 集群管理、同步任務(wù)管理等，對應(yīng)的 service 以 Protocol Buffers 格式定義在 dm/proto/dmmaster.proto 中，對應(yīng)的 generated 代碼在 dm/pb/dmmaster.pb.go 中。各 service 的具體實現(xiàn)在 dm/master/server.go 中（*Server）。

DM-worker 的結(jié)構(gòu)與 DM-master 類似，其入口代碼在 cmd/dm-worker/main.go 中。各 RPC services 的 Protocol Buffers 格式定義在 dm/proto/dmworker.proto 中，對應(yīng)的 generated 代碼在 dm/pb/dmworker.pb.go 中，對應(yīng)的實現(xiàn)代碼在 dm/worker/server.go 中（*Server）。DM-worker 的啟動流程與 DM-master 類似，在此不再額外說明。

Server 這個 struct 的主要 fields 除用于處理 RPC 的 rootLis 與 svr 外，另一個是用于管理同步任務(wù)與 relay log 的 worker（相關(guān)代碼在 dm/worker/worker.go 中）。

在 Worker 這個 struct 中，主要 fields 包括：

subTasks：維護(hù)該 DM-worker 上的所有同步子任務(wù)信息。

relayHolder：對 relay 處理單元相關(guān)操作進(jìn)行簡單封裝，轉(zhuǎn)發(fā)相關(guān)操作請求給 relay 處理單元，獲取 relay 處理單元的狀態(tài)信息。

relayPurger：根據(jù)用戶配置及相關(guān)策略，嘗試定期對 relay log 進(jìn)行 purge 操作。

數(shù)據(jù)同步子任務(wù)管理的代碼實現(xiàn)主要在 dm/worker/subtask.go 中， relay 處理單元管理的代碼實現(xiàn)主要在 dm/worker/relay.go 中，對 relay log 進(jìn)行 purge 操作的代碼實現(xiàn)主要在 relay/purger pkg 中。在本篇文章中，我們暫時只關(guān)注 DM 架構(gòu)相關(guān)的實現(xiàn)，上述各功能的具體實現(xiàn)將在后續(xù)的相關(guān)文章中展開介紹。

Worker 的入口方法為 Start，其中的主要操作包括：

通過 w.relayHolder.Start 啟動 relay 處理單元，開始從上游拉取 binlog。

通過 w.relayPurger.Start 啟動后臺 purge 線程，嘗試對 relay log 進(jìn)行定期 purge。

其他的操作主要還包括處理 Server 轉(zhuǎn)發(fā)而來的同步任務(wù)管理、relay 處理單元管理、狀態(tài)信息查詢等。

dmctl 的入口代碼在 cmd/dm-ctl/main.go，其操作除參數(shù)解析與 signal 處理外，主要為調(diào)用 loop 進(jìn)入命令處理循環(huán)、等待用戶輸入操作命令。

在 loop 中，我們借助 chzyer/readline 提供命令行交互環(huán)境，讀取用戶輸入的命令并輸出命令執(zhí)行結(jié)果。一個命令的處理流程為：

調(diào)用 l.Readline 讀取用戶輸入的命令

判斷是否需要退出命令行交互環(huán)境（exit 命令）或需要進(jìn)行處理

調(diào)用 ctl.Start 進(jìn)行命令分發(fā)與處理

dmctl 的具體命令處理實現(xiàn)在 dm/ctl pkg 中，入口為 dm/ctl/ctl.go 中的 Start 方法，命令的分發(fā)與參數(shù)解析借助于 spf13/cobra。命令的具體功能實現(xiàn)在相應(yīng)的子 pkg 中：

master：dmctl 與 DM-master 交互的命令，是當(dāng)前 DM 推薦的命令交互方式。

worker：dmctl 與 DM-worker 交互的命令，主要用于開發(fā)過程中進(jìn)行 debug，當(dāng)前并沒有實現(xiàn)所有 DM-worker 支持的命令，未來可能廢棄。

common：多個命令依賴的通用操作及 dmctl 依賴的配置信息等。

每個 dmctl 命令，其主要對應(yīng)的實現(xiàn)包括 3 個部分：

在各命令對應(yīng)的實現(xiàn)源文件中，通過 New***Cmd 形式的方法創(chuàng)建 cobra.Command 對象。

在 dm/ctl/ctl.go 中通過調(diào)用 rootCmd.AddCommand 添加該命令。

在各命令對應(yīng)的實現(xiàn)源文件中，通過 ***Func 形式的方法實現(xiàn)參數(shù)驗證、RPC 調(diào)用等具體功能。