上下文

GORM 的上下文支持由 WithContext 方法启用，是一项强大的功能，可增强 Go 应用程序中数据库操作的灵活性和控制力。它允许跨不同的操作模式、超时设置，甚至集成到钩子/回调和中间件中进行上下文管理。让我们深入探讨这些不同的方面。

单会话模式

单会话模式适用于执行单个操作。它确保在上下文的范围内执行特定操作，从而实现更好的控制和监控。

db.WithContext(ctx).Find(&users)

连续会话模式

连续会话模式非常适合执行一系列相关操作。它在这些操作之间维护上下文，这在事务等场景中特别有用。

tx := db.WithContext(ctx)
tx.First(&user, 1)
tx.Model(&user).Update("role", "admin")

上下文超时

在传递给 db.WithContext 的上下文中设置超时可以控制长时间运行查询的持续时间。这对于维护性能和避免数据库交互中的资源锁定至关重要。

ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
defer cancel()

db.WithContext(ctx).Find(&users)

钩子/回调中的上下文

也可以在 GORM 的钩子/回调中访问上下文。这使得在这些生命周期事件期间可以使用上下文信息。

func (u *User) BeforeCreate(tx *gorm.DB) (err error) {
  ctx := tx.Statement.Context
  // ... use context
  return
}

与 Chi 中间件集成

GORM 的上下文支持扩展到 Web 服务器中间件，例如 Chi 路由器中的中间件。这允许在 Web 请求范围内为所有数据库操作设置带有超时的上下文。

func SetDBMiddleware(next http.Handler) http.Handler {
  return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    timeoutContext, _ := context.WithTimeout(context.Background(), time.Second)
    ctx := context.WithValue(r.Context(), "DB", db.WithContext(timeoutContext))
    next.ServeHTTP(w, r.WithContext(ctx))
  })
}

// Router setup
r := chi.NewRouter()
r.Use(SetDBMiddleware)

// Route handlers
r.Get("/", func(w http.ResponseWriter, r *http.Request) {
  db, ok := r.Context().Value("DB").(*gorm.DB)
  // ... db operations
})

r.Get("/user", func(w http.ResponseWriter, r *http.Request) {
  db, ok := r.Context().Value("DB").(*gorm.DB)
  // ... db operations
})

注意：使用 WithContext 设置 Context 是协程安全的。这确保了跨多个协程安全地管理数据库操作。有关更多详细信息，请参阅 GORM 中的会话文档。

日志记录器集成

GORM 的日志记录器也接受 Context，可用于日志跟踪和与现有日志记录基础设施集成。

有关更多详细信息，请参阅日志记录器文档。