简化代码，合并 if 和 else 条件

一分耕耘，一分收获！既然打开了这篇文章《简化代码，合并 if 和 else 条件》，就坚持看下去吧！文中内容包含等等知识点...希望你能在阅读本文后，能真真实实学到知识或者帮你解决心中的疑惑，也欢迎大佬或者新人朋友们多留言评论，多给建议！谢谢！

if category == 0 {
        rows, err := h.repo.getalllatestproducts(c.context())
        if err != nil {
            return c.status(fiber.statusinternalservererror).sendstring(err.error())
        }
        result := make([]interface{}, len(rows))
        for i, product := range rows {
            result[i] = dbrow.converttoalllatestproducts(product)
        }
    } else {
        rows, err := h.repo.getlatestproductsbycategory(c.context(), int16(category))
        if err != nil {
            return c.status(fiber.statusinternalservererror).sendstring(err.error())
        }
        result := make([]interface{}, len(rows))
        for i, product := range rows {
            result[i] = dbrow.converttocategorylatestproducts(product)
        }
    }

if和else条件都遵循相同的代码流程，只是函数和结构体不同，如何合并它们，使代码更小。我的意思是：

var rows []postgres.getalllatestproductsrow
    var rows []postgres.getlatestproductsbycategoryrow
    if category == 0 {
        rows, err = h.repo.getalllatestproducts(c.context())
    } else {
        rows, err = h.repo.getlatestproductsbycategory(c.context(), int16(category))
    }
    //rest of the code ...

无法触摸 h.repo.getalllatestproducts 或 h.repo.getlatestproductsbycategory，因为这些是外部函数。类型安全也很重要。

可以有多个函数，例如featuredproducts、newproducts，我想创建一个通用函数，根据动态选择的sql函数以json形式返回产品。

你遇到了问题，几十个具有相同代码结构的函数很糟糕，至少对我来说，它的可读性与否并不重要，它只是重复的复制/粘贴，没有任何意义， 主要是复制/粘贴，仅更改函数名称/sqlc 生成的函数名称和结构名称，其余代码流程相同。

sqlc 根据 sql 查询生成自动代码，现在编写重复的代码只是将结果转换为 json 并将其返回给客户端是浪费时间。 可能有几十个 sql 函数来返回最新产品、特色产品、类别中的产品、心愿单产品等等……

网站所理解的只是 product 结构，但是 sqlc 返回不同的结构，因此没有单一的 dbresult 之类的东西。无论如何，映射并不是什么大事，使用反射我们可以在映射函数中检查同名字段，并将sql.nullstring转换为字符串等。

对我来说真正的问题是 if/else 语句。您已将代码移动到不同的函数中，但对于我来说，在这种情况下它没有意义。因为网络处理程序无论如何都必须检查 请求是否有效，是否定义了category，然后检查category是否为0，然后调用不同的函数，然后得到结果并返回给客户端。 对于单个函数来说，它可能看起来更好，但对于实际生产来说，它会让事情变得更糟，现在每个 api 有 3 个函数，而不是单个函数和 if/else 块。

我希望将 sqlc 结果映射到路由处理程序。代码流程始终相同， 仅函数名称和结构名称发生变化。如何使其动态，以便在我的 http 处理程序中，我可以简单地编写：

return sqlcresult(c.query("category"), getallfeaturedproducts, getfeaturedproductsbycategory)

然后根据 c.query("category") 中的类别值，sqlcresult 将自动调用 getallfeaturedproducts 或 getfeaturedproductsbycategory。 类似函数作为回调，但函数签名不同，这是一个问题。

func (q *queries) getallfeaturedproducts(ctx context.context) ([]getallfeaturedproductsrow, error)
func (q *queries) getfeaturedproductsbycategory(ctx context.context, idcategory int16)

映射函数不是必需的，因为在 sqlcresult 中，我们可以执行以下操作：

mapdbstructtorestapistruct(&product{}, &row, mapfields(&row))

这将创建字段名称和索引的映射，并传递 dbresult 行，并使用反射将其转换为 product 结构并返回相同的结构，即在修改其字段后将第一个参数作为结果返回。 p>

我仍在寻找如何编写 sqlcresult 函数，以 sqlc 函数名称作为输入，然后返回结果，或者可以通过将 product{} 结构本身放入 sqlcresult 函数中来使其更通用，例如：

var result := sqlcresult(&product{}, c.query("category") == 0, getallfeaturedproducts, getfeaturedproductsbycategory)
return c.status(fiber.statusok).json(result)

其中 sqlcresult 将根据布尔条件调用 getallfeaturedproducts 或 getfeaturedproductsbycategory 并创建将函数结果映射到作为第一个参数传递的结构，然后返回该结构。

或者可能这就是最终目标：

func (h *Handlers) GetLatestProducts(c *fiber.Ctx) error {
  if c.Query("category") == 0
    return c.JSON(SQLCResult(&Product{}, GetAllLatestProducts)
  else
    return c.JSON(SQLCResult(&Product{}, GetLatestProductsByCategory, c.Query("category"))
}

func (h *Handlers) GetFeaturedProducts(c *fiber.Ctx) error {
  if c.Query("category") == 0
    return c.JSON(SQLCResult(&Product{}, GetAllFeaturedProducts)
  else
    return c.JSON(SQLCResult(&Product{}, GetFeaturedProductsByCategory, c.Query("category"))
}

解决方案

有很多需要考虑的地方，代码确实没有问题，但可能很难维护，如果预期的类似场景更多，可能会更困难，它的扩展性很差，从长远来看可能会出现更多规则轻松制作意大利面条。

我们想要的是关注点分离、相似部分可重用以及清晰。我们可以轻松实现这一点。

鉴于我们无法更改存储库 api（这可能是一种直接的方法），我们必须包装存储库，更像是装饰器或 go 术语中的“影子”。

// producthandler shadows product repository
type producthandler struct {
    *repo
}

这使我们能够更好地封装每次调用的兴趣

func (ph producthandler) getlatestproductsbycategory(ctx context.context, cat int) ([]interface{}, error) {
    if cat == 0 {
        return nil, nil
    }

    l, err := ph.repo.getlatestproductsbycategory(ctx, cat)

    return resultset(&productsbycategory{}, l), err
}

func (ph producthandler) getalllatestproducts(ctx context.context) ([]interface{}, error) {
    l, err := ph.repo.getalllatestproducts(ctx)

    return resultset(&products{}, l), err
}

这样，我们就可以通过自己的方法委托检索或不检索类别的责任，并自动将结果集包装为其自己的类型，从而相应地分离映射责任。

type products struct {
    id   string
    name string
}

type productsbycategory struct {
    category string
}

为了实现数据库结果集到特定类型的转换，我们必须公开一个公共接口，因此任何实现该接口的类型都可以进行自身转换（translate、map、 hydro 是同义词）

type transformer interface {
    transform(interface{}) interface{}
}

现在每种类型都可以有自己的从 -> 到 的转换

func (p products) transform(i interface{}) interface{} {
    v, _ := i.(*dbresult)

    p.name = v.rawname
    p.id = v.rawid

    return p
}

func (p productsbycategory) transform(i interface{}) interface{} {
    v, _ := i.(*dbresult)

    p.category = v.categoryname

    return p
}

有一个函数可以帮助我们转换数据列表，我们可以随时重复使用

func resultset(t transformer, d []interface{}) []interface{} {
    result := make([]interface{}, len(d))
    for i, p := range d {
        result[i] = t.transform(p)
    }

    return result
}

现在我们的实现可以简单地看起来像这样，并且所有这些部分都可以重用

func main() {
    var category int
    // h.Repo
    repo := Repo{}
    ph := ProductHandler{&repo}

    pcat, _ := ph.GetLatestProductsByCategory(context.Background(), category)
    products, _ := ph.GetAllLatestProducts(context.Background())
    products = append(pcat, products...)

    for _, product := range products {
        fmt.Printf("%v\n", product)
    }
}

虽然代码使用了 interface{} ，但这并没有什么不好，但最终你的数据已经像这样来自数据库了，而我们只是传递它们。如果做得不好，断言它们的类型可能会代价高昂，但这里没有这种情况，直到调用 json marshal。

你可以找到working copy here 为了支持这些情况，有一个可能的数据库响应调用的模拟。 尝试给 category 赋值，看看会发生什么

今天关于《简化代码，合并 if 和 else 条件》的内容介绍就到此结束，如果有什么疑问或者建议，可以在golang学习网公众号下多多回复交流；文中若有不正之处，也希望回复留言以告知！