标签导航：

go语言中如何复用不同结构体的相同操作？

Go语言中复用不同结构体的相同操作，提高代码效率和可维护性

在Go语言开发中，经常遇到对不同结构体执行相同操作的情况，例如删除操作。本文介绍如何利用接口特性，将重复代码封装成公共方法，提升代码可重用性和可维护性。

假设我们有两个结构体models.scheme和models.tag，都需要执行删除操作，其核心逻辑相似，仅数据类型不同。 传统方法是为每个结构体编写单独的删除函数，导致代码冗余。

解决方案：使用接口

我们可以定义一个接口，声明一个Del方法，该方法接收一个上下文参数（例如*gin.Context）并返回错误。然后，让models.scheme和models.tag都实现这个接口。这样，我们就可以编写一个通用的删除函数，该函数接收接口类型的参数，从而实现代码复用。

代码示例:

  1. 定义接口:
type Deletable interface {
    Del(c *gin.Context) error
}
  1. 实现接口: 以下代码片段展示了如何让models.scheme和models.tag实现Deletable接口。 注意，Del方法的具体实现需要根据实际数据库操作进行调整，避免硬编码ID字段。 权限验证（例如checkbranchrole函数）也应包含在Del方法内部。
type scheme struct {
    // ... other fields
    ID      int
    BranchID int
    IsDel    int
}

func (s *scheme) Del(c *gin.Context) error {
    // ... 获取ID，通常从c.Param("id")或请求体中获取 ...
    // ... 数据库查询和更新逻辑，例如使用gorm或sqlx ...
    // ... 权限验证：checkbranchrole(c, s.BranchID) ...
    return nil // 或返回具体的错误
}


type tag struct {
    // ... other fields
    ID      int
    BranchID int
    IsDel    int
}

func (t *tag) Del(c *gin.Context) error {
    // ... 获取ID，通常从c.Param("id")或请求体中获取 ...
    // ... 数据库查询和更新逻辑，例如使用gorm或sqlx ...
    // ... 权限验证：checkbranchrole(c, t.BranchID) ...
    return nil // 或返回具体的错误
}
  1. 通用删除函数:
func Delete(c *gin.Context, deletable Deletable) {
    // ... 解析请求参数，例如批量删除的ID列表 ...
    // ...  错误处理 ...
    err := deletable.Del(c)
    if err != nil {
        // ... 错误处理 ...
        return
    }
    // ... 成功响应 ...
}

使用方法:

// 删除scheme
Delete(c, &models.scheme{ID: 1, BranchID: 101, IsDel: 0})

// 删除tag
Delete(c, &models.tag{ID: 2, BranchID: 102, IsDel: 0})

通过这种方式，我们成功地将不同结构体的删除操作抽象成一个通用的Delete函数，提高了代码的可重用性和可维护性。 关键在于灵活运用Go语言的接口特性，将公共逻辑提取出来，避免代码重复。 记住，数据库操作和权限验证的具体实现需要根据实际情况进行调整。