feat: 新增注入上下文的方法

pkg/db/db.go

@@ -1,6 +1,7 @@
 package db
 
 import (
+	"context"
 	"database/sql"
 	"errors"
 	"gorm.io/driver/mysql"
@@ -69,6 +70,10 @@ func DB() *gorm.DB {
 	return db
 }
 
+func WithContext(ctx context.Context) *gorm.DB {
+	return DB().WithContext(ctx)
+}
+
 // SetDB 自定义操作引擎
 func SetDB(engine *gorm.DB) {
 	lock.Lock()