GORM 的泛型 API 的 ctx := context.Background() 很啰嗦，有没有缓解的方法？

Question

我灵机一动写下了下面的代码：

package bc

import (
    "context"
)

var C = context.Background()

给 DeepSeek 看了一下，它说这种方法牺牲了上下文机制的所有好处，而且难以测试。不过我没看出来哪里难测试。另外如果需要超时不用这个 bc.C 不就行了吗？另外我发现写下 bc.C 之后保存可以自动在 import 里加入包，美滋滋。我之前以为只有标准库有这个待遇。

我是笨蛋，所以我对我的灵机一动很怀疑。但是这个是在是太啰嗦了，忍不了。一想到可能要写那条语句几十次，还有占用宝贵的视野就难受。而且我总感觉这个啰嗦是可以避免且没什么用的，不像错误处理没法避免且有用。

Answer 1

package bc

import "context"

func Ctx() context.Context {
    return context.Background()
}

这样你在业务代码中写 bc.Ctx()，仍然简洁，而且在测试中可以 mock bc.Ctx = func() context.Context { return context.WithTimeout(...) }（如果你用接口或函数变量）。

建议定义一个 ContextProvider 接口 + 默认实现

1、用一个接口来提供 context.Context，默认实现返回 context.Background()，

package contextx

import "context"

// ContextProvider 是一个提供 context 的接口
type ContextProvider interface {
    Context() context.Context
}

// defaultProvider 是默认实现，返回 context.Background()
type defaultProvider struct{}

func (d defaultProvider) Context() context.Context {
    return context.Background()
}

// Provider 是当前使用的上下文提供者，默认是 defaultProvider
var Provider ContextProvider = defaultProvider{}

// Ctx 是对外暴露的简洁调用方式
func Ctx() context.Context {
    return Provider.Context()
}

2、使用方式（业务代码中）

import "your_project/contextx"

db.WithContext(contextx.Ctx()).Find(&data)

3、测试中替换 Provider

type testProvider struct{}

func (t testProvider) Context() context.Context {
    ctx, _ := context.WithTimeout(context.Background(), time.Second*1)
    return ctx
}

func TestSomething(t *testing.T) {
    contextx.Provider = testProvider{}
    // 测试逻辑
}

Answer 2

type Repository[T any] struct {
    db *gorm.DB
}

func NewRepository[T any](db *gorm.DB) *Repository[T] {
    return &Repository[T]{db: db}
}

func (r *Repository[T]) FindAll(conds ...interface{}) ([]T, error) {
    var results []T
    err := r.db.WithContext(context.Background()).Find(&results, conds...).Error
    return results, err
}

func (r *Repository[T]) FindByID(id uint) (*T, error) {
    var result T
    err := r.db.WithContext(context.Background()).First(&result, id).Error
    return &result, err
}

// 使用
userRepo := NewRepository[User](db)
users, err := userRepo.FindAll("age > ?", 18)