如何为GORM全局配置所有查询的默认超时时间？

可以实现全局默认超时配置，无需为每个查询单独设置，以下是两种实用方案：

方案一：利用GORM回调自动注入超时上下文 #

通过GORM的回调机制，在所有查询执行前自动为未指定上下文的请求添加默认超时上下文，同时支持业务代码通过WithContext覆盖全局设置。

import (
  "context"
  "time"
  "gorm.io/gorm"
)

// 定义全局默认超时时间
const defaultDBTimeout = 5 * time.Second

// 注册全局超时回调
func RegisterDefaultTimeoutCallback(db *gorm.DB) error {
  // 为所有CRUD及原生查询类型注册回调
  operations := []string{"create", "query", "update", "delete", "raw"}
  for _, op := range operations {
    err := db.Callback().Create().Before("*").Register(op+"_timeout", func(db *gorm.DB) {
      // 若当前无上下文，先设为默认背景上下文
      if db.Statement.Context == nil {
        db.Statement.Context = context.Background()
      }
      // 基于已有上下文添加超时，避免丢失原有上下文信息（如trace ID）
      ctx, cancel := context.WithTimeout(db.Statement.Context, defaultDBTimeout)
      defer cancel()
      db.Statement.Context = ctx
    })
    if err != nil {
      return err
    }
  }
  return nil
}

// 初始化DB时注册回调
func InitDB() (*gorm.DB, error) {
  // 初始化DB连接（替换为你的实际连接代码）
  db, err := gorm.Open(/* 你的DB驱动配置 */, "")
  if err != nil {
    return nil, err
  }

  // 注册全局超时回调
  if err := RegisterDefaultTimeoutCallback(db); err != nil {
    return nil, err
  }

  return db, nil
}

使用时直接用初始化后的db实例执行查询即可，无需额外设置上下文：

// 自动携带默认超时上下文
res := db.Model(&Stock{}).Update(myMap)

// 若需自定义超时，可通过WithContext覆盖全局设置
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
res := db.WithContext(ctx).Model(&Stock{}).Update(myMap)

方案二：封装DB实例统一处理超时 #

通过封装自定义的DB结构体，将所有查询方法包装为自动携带超时上下文的版本，强制所有查询走超时逻辑，适合需要严格管控的场景。

import (
  "context"
  "time"
  "gorm.io/gorm"
)

type TimedDB struct {
  *gorm.DB
  timeout time.Duration
}

// 创建带默认超时的DB实例
func NewTimedDB(rawDB *gorm.DB, timeout time.Duration) *TimedDB {
  return &TimedDB{DB: rawDB, timeout: timeout}
}

// 封装Model方法，自动注入超时上下文
func (td *TimedDB) Model(value interface{}) *gorm.DB {
  ctx, cancel := context.WithTimeout(context.Background(), td.timeout)
  defer cancel()
  return td.DB.WithContext(ctx).Model(value)
}

// 封装Create方法
func (td *TimedDB) Create(value interface{}) *gorm.DB {
  ctx, cancel := context.WithTimeout(context.Background(), td.timeout)
  defer cancel()
  return td.DB.WithContext(ctx).Create(value)
}

// 按需封装Update、Delete、Raw等其他常用方法

使用时直接调用封装后的方法：

// 初始化带超时的DB实例
timedDB := NewTimedDB(rawDB, 5*time.Second)

// 自动携带默认超时
res := timedDB.Model(&Stock{}).Update(myMap)

内容的提问来源于stack exchange，提问作者Hậu Chử