接受通用结构的函数

时间:2017-08-09 17:00:58

标签: go refactoring

是否可以让我的函数定义接受任何类型的结构?

我试过像这样重构:

// This method should accept any type of struct
// Once I receive my response from the database,
// I scan the rows to create a slice of type struct.

func generateResponse(rows *sqlx.Rows, structSlice []struct{}, structBody struct{}) ([]struct{}, error) {
    for rows.Next() {

        err := rows.StructScan(&structBody)

        if err != nil {
            return nil, err
        }

        structSlice = append(structSlice, structBody)

    }

    err := rows.Err()
    if err != nil {
        return nil, err
    }

    return structSlice, nil
}

假设我的结构类型为OrderRevenue

当我调用上面的函数时:

structSlice, err := generateResponse(rows, []OrderRevenue{}, OrderRevenue{})

我得到的错误是:

cannot use []OrderRevenue literal as type []struct{} in argument...

我是以错误的方式解决这个问题吗?

1 个答案:

答案 0 :(得分:2)

这被认为是Go类型系统的基石(或更多限制)。 struct{}是一种与struct{ field1 int }不同的未命名类型,当然与OrderRevenue{}不同。

Go强调通过接口进行抽象,也许你应该尝试一下。这是第一次:

  type OrderRevenue interface {
          MarshalMyself() ([]byte, error)
  }

  type Anonymous struct {}
  func (a Anonymous) MarshalMyself() ([]byte, error) {
          // implementation's up to you
          return []byte{}, nil
  }

  // the function signature
  generateResponse(rows *sqlx.Rows, structSlice []OrderRevenue, structBody Body) ([]Body, error) {
          // ...
  }

在这种情况下,您还可以使用所有类型都实现的空接口interface{},但您必须递归遍历结构以执行手动类型断言。 Go中最好的方法是提前知道数据的形状,至少部分地知道。