Go开发：复用CRUD Handler适配PostgreSQL与MongoDB的方案咨询

不需要重新编写MongoDB对应的Handler，通过接口抽象+依赖注入的方式可以完全复用现有Handler，同时优雅支持PostgreSQL（生产）和MongoDB（开发）两种数据库环境，这是Go后端开发中解耦业务与数据库实现的标准实践。

1. 核心思路：抽象Repository层接口 #

将数据库操作的核心逻辑（如检查邮箱、插入用户）抽象为通用接口，让PostgreSQL和MongoDB的Repository分别实现该接口。Service层依赖抽象接口而非具体实现，Handler层则完全不需要修改。

步骤1：定义统一的Repository接口 #

// 定义通用的用户Repository接口
type UserRepository interface {
    CheckIfEmailExists(email string) bool
    Insert(user *models.User) error
}

步骤2：统一数据模型（可选但推荐） #

将原有的models.User和models.UserB合并为一个结构体，通过标签同时适配GORM（PostgreSQL）和MongoDB：

// models/user.go
type User struct {
    ID       uint   `gorm:"primaryKey" bson:"_id,omitempty"` // 适配PG主键和Mongo ObjectID
    Email    string `gorm:"unique;not null" bson:"email"`
    Password string `gorm:"not null" bson:"password"`
    // 其他业务字段...
}

2. 让PostgreSQL和MongoDB Repository实现接口 #

PostgreSQL Repository实现 #

type PGUserRepository struct {
    db *gorm.DB
}

func (repo *PGUserRepository) CheckIfEmailExists(email string) bool {
    var user models.User
    err := repo.db.Debug().Model(&models.User{}).Where("email = ?", email).First(&user).Error
    return !errors.Is(err, gorm.ErrRecordNotFound) // 修正原逻辑：返回邮箱是否存在
}

func (repo *PGUserRepository) Insert(user *models.User) error {
    if err := repo.db.Debug().Create(user).Error; err != nil {
        log.Printf("PG insert user failed: %v", err)
        return err
    }
    return nil
}

MongoDB Repository实现 #

type MongoUserRepository struct {
    coll *mongo.Collection
}

func (repo *MongoUserRepository) CheckIfEmailExists(email string) bool {
    filter := bson.D{{Key: "email", Value: email}}
    count, err := repo.coll.CountDocuments(context.TODO(), filter)
    if err != nil {
        log.Printf("Mongo check email failed: %v", err)
        return false // 避免panic，返回安全默认值
    }
    return count > 0
}

func (repo *MongoUserRepository) Insert(user *models.User) error {
    _, err := repo.coll.InsertOne(context.TODO(), user)
    if err != nil {
        log.Printf("Mongo insert user failed: %v", err)
        return err
    }
    return nil
}

3. 重构Service层依赖抽象接口 #

让Service层依赖UserRepository接口而非具体实现，实现业务逻辑与数据库解耦：

type UserService struct {
    userRepo UserRepository
}

// 通过构造函数注入不同的Repository
func NewUserService(repo UserRepository) *UserService {
    return &UserService{userRepo: repo}
}

// 复用原有的Register逻辑，完全不需要修改
func (svc *UserService) Register(user *models.User) error {
    if svc.userRepo.CheckIfEmailExists(user.Email) {
        return errors.New("user already registered")
    }
    hash, err := svc.generatePasswordHash(user.Password)
    if err != nil {
        return errors.New("failed to hash password")
    }
    user.Password = hash
    return svc.userRepo.Insert(user)
}

4. 根据环境初始化对应组件，复用原Handler #

程序启动时通过环境变量判断数据库类型，初始化对应的Repository和Service，原Handler代码完全复用：

func main() {
    r := gin.Default()

    // 从环境变量获取数据库类型（开发环境设为mongo，生产设为pg）
    dbType := os.Getenv("DB_TYPE")
    if dbType == "" {
        dbType = "pg" // 默认生产环境
    }

    var userRepo UserRepository
    switch dbType {
    case "pg":
        // 初始化PostgreSQL连接
        db, err := gorm.Open(postgres.Open(os.Getenv("PG_DSN")), &gorm.Config{})
        if err != nil {
            log.Fatalf("PG connect failed: %v", err)
        }
        userRepo = &PGUserRepository{db: db}
    case "mongo":
        // 初始化MongoDB连接
        client, err := mongo.Connect(context.TODO(), options.Client().ApplyURI(os.Getenv("MONGO_URI")))
        if err != nil {
            log.Fatalf("Mongo connect failed: %v", err)
        }
        coll := client.Database(os.Getenv("MONGO_DB")).Collection("users")
        userRepo = &MongoUserRepository{coll: coll}
    default:
        log.Fatalf("unsupported DB_TYPE: %s", dbType)
    }

    userService := NewUserService(userRepo)
    userController := &UserController{userService: userService}

    // 复用原Handler
    r.POST("/register", userController.Register)
    r.Run()
}

为什么不直接在Handler里判断数据库类型？ #

• 违反开闭原则：新增数据库类型时需修改Handler/Service代码，耦合度高
• 维护成本高：每个CRUD接口都要加判断逻辑，代码冗余
• 测试困难：无法单独测试某一种数据库的业务逻辑

而接口抽象的方式完全解耦了业务与数据库实现，后续新增数据库（如MySQL）只需添加新的Repository实现，Service和Handler无需任何修改。

内容的提问来源于stack exchange，提问作者Shtefan Gusan