Go 语言接口编程的最佳实践和错误处理模式

2023-10-26 06:38:10

03-Go 语言接口编程最佳实践方式、错误处理思想与 Book 表 CRUD - 2

引言

接口是 Go 语言中用于定义契约和实现多态性的强大机制。在上一节中，我们探讨了接口编程的基础知识。在这篇文章中，我们将使用接口编程的思想来实现我们项目中的 dbrepo 包所使用的功能。

提取接口

我们首先需要提取几个接口来定义 dbrepo 包中的操作：

// CRUD 接口定义基本的 CRUD 操作
type CRUD interface {
    Create(ctx context.Context, entity interface{}) error
    Read(ctx context.Context, id int64) (interface{}, error)
    Update(ctx context.Context, entity interface{}) error
    Delete(ctx context.Context, id int64) error
}

// TxHandler 接口定义事务处理程序
type TxHandler interface {
    // 在事务中执行函数
    Handle(ctx context.Context, fn func(ctx context.Context) error) error
}

优雅的错误处理

错误处理是 Go 语言中至关重要的方面。我们使用 errors.Is() 和 errors.As() 函数来优雅地处理错误：

func (repo *dbrepo) Read(ctx context.Context, id int64) (interface{}, error) {
    book := &Book{}
    err := repo.db.Where("id = ?", id).First(book).Error
    if errors.Is(err, gorm.ErrRecordNotFound) {
        return nil, ErrNotFound
    }
    if err != nil {
        return nil, fmt.Errorf("failed to read book: %w", err)
    }
    return book, nil
}

基础 CRUD

我们实现了所有 CRUD 操作，它们使用 Create()、Read()、Update() 和 Delete() 方法：

func (repo *dbrepo) Create(ctx context.Context, entity interface{}) error {
    return repo.db.Create(entity).Error
}

func (repo *dbrepo) Read(ctx context.Context, id int64) (interface{}, error) {
    // ...
}

func (repo *dbrepo) Update(ctx context.Context, entity interface{}) error {
    return repo.db.Save(entity).Error
}

func (repo *dbrepo) Delete(ctx context.Context, id int64) error {
    return repo.db.Delete(&Book{}, id).Error
}

批量添加数据

我们还实现了批量添加数据的功能：

func (repo *dbrepo) BulkCreate(ctx context.Context, entities []interface{}) error {
    return repo.db.Create(entities).Error
}

事务管理

事务管理对于确保数据库操作的原子性至关重要。我们使用 repo.db.Transaction() 函数来实现事务：

func (repo *dbrepo) Handle(ctx context.Context, fn func(ctx context.Context) error) error {
    return repo.db.Transaction(func(tx *gorm.DB) error {
        if err := fn(ctx); err != nil {
            return err
        }
        return tx.Commit().Error
    })
}