想要在程式裡監控資料庫的操作耗時，想要在底層框架中自動傳遞鏈路跟蹤資訊，這些需求經常會碰到，特別是在構建基礎框架的時候。 核心目標只有一個，那就是在底層封裝好，不用上層使用人員關心。今天跟大家聊聊常用的底層擴充套件埋點方式是怎麼處理的。 ## 框架自帶擴充套件點 如果你使用的框架在設計的時候，就預留了擴充套件點就很方便了。比如 Mybatis 的攔截器，我們可以在攔截器中對 Sql 進行監控，改寫。 比如阿里的 Sentinel 框架，可以通過 SPI 來擴充套件 Slot，調整編排順序，新增自定義的 Slot 來實現限流告警等。 開源框架的質量參差不齊，有在早期設計比較好的，留足了各種擴充套件點，方便使用者。也有一些沒有考慮那麼全面，導致你在使用的時候需要進行擴充套件，發現找不到擴充套件點，對於框架本身沒有提供擴充套件點的場景，請接著看下面。 ## 修改原始碼 如果框架沒有擴充套件點，最直接的方式就是修改開源框架的原始碼來擴充套件自己想要的功能，通常的做法就是克隆原始碼到自己的私有倉庫中，然後修改，測試，重新打包使用。 像我們之前用了 XXL-JOB 做任務排程，也是修改了某些程式碼，在介面上擴充套件了監控通知的配置資訊，預設是隻支援郵箱，可以擴展出手機，釘釘等。 修改原始碼不好的點在於需要跟原框架的版本進行對齊，如果不對齊，隨便改都沒事。不對齊意味著修復了某些 bug 和新增了某些功能，就無法使用了。要對齊，就需要不斷的將新版本的程式碼合併到你自己的分支上。 還有很多公司，就是基於開源的版本，構建了公司內部自己的版本，後續直接就是跟著內部的使用需求去擴充套件和維護了，不會跟社群的版本進行同步，如果你們有專門的團隊去做這件事情，也是一種方式。 ## 同名檔案覆蓋 改原始碼的方式需要經常同步新版本的程式碼，有的時候往往只想修改某一個類而已，比如對底層的某些操作進行埋點監控，如果框架本身沒有提供擴充套件點的話只能改原始碼來實現。 其實還有個投機取巧的方式，就是在專案中建立一個跟你要修改的一模一樣的類，包名+類目都一樣，方法名也一樣，方法的實現你可以改。這樣就能覆蓋 jar 包中的類了，還是跟類載入順序有關係，先載入你自己定義的。 這樣的方式好處在於不用經常去同步新版本的程式碼，如果你用的框架版本升級了，只要包名和類名不變，你這個覆蓋的只是那個類而已，新增的功能和修復的 bug 都不會有影響。 ## 切面攔截 切面在做很多統一處理的時候非常有用，同樣在做底層埋點的場景也適用。 比如我們要對專案中 Mongodb 的所有操作都進行埋點監控，可以修改 MongoDB 的驅動原始碼，可以建立同名檔案進行覆蓋，方式有很多種，找到一個合適，又能實現需求的最重要。 以 Spring 中操作 Mongodb 來說明，在 Spring Data Mongodb 中會 MongoTemplate 來操作 Mongodb。最簡單的方式就是直接對 MongoTemplate 類進行埋點，這樣所有的操作都可以監控起來。 用切面直接切到 MongoTemplate 的所有方法上，然後進行埋點，就很簡單了。 ```plain @Aspect public class MongoTemplateAspect { @Pointcut("execution(* org.springframework.data.mongodb.core.MongoTemplate.*(..))") public void pointcut() {} @Around("pointcut()") public Object around(final ProceedingJoinPoint pjp) throws Throwable { String callMethod = pjp.getSignature().getDeclaringType().getSimpleName() + "." + pjp.getSignature().getName();