Golang 中的泛型方法参数

Question

我需要帮助才能使这项工作适用于任何类型。

我有一个函数，我需要接受具有 ID 属性的其他类型。

我尝试过使用接口，但这对我的 ID 属性案例不起作用。这是代码：

package main


import (
  "fmt"
  "strconv"
  )

type Mammal struct{
  ID int
  Name string 
}

type Human struct {  
  ID int
  Name string 
  HairColor string
}

func Count(ms []Mammal) *[]string { // How can i get this function to accept any type not just []Mammal
   IDs := make([]string, len(ms))
   for i, m := range ms {
     IDs[i] = strconv.Itoa(int(m.ID))
   }
   return &IDs
}

func main(){
  mammals := []Mammal{
    Mammal{1, "Carnivorious"},
    Mammal{2, "Ominivorious"},
  }

  humans := []Human{
    Human{ID:1, Name: "Peter", HairColor: "Black"},
    Human{ID:2, Name: "Paul", HairColor: "Red"},
  } 
  numberOfMammalIDs := Count(mammals)
  numberOfHumanIDs := Count(humans)
  fmt.Println(numberOfMammalIDs)
  fmt.Println(numberOfHumanIDs)
}

我明白了

error prog.go:39: cannot use humans (type []Human) as type []Mammal in argument to Count

在此处查看 Go Playground 了解更多详细信息http://play.golang.org/p/xzWgjkzcmH

Answer 1

使用接口而不是具体类型，并使用 embedded interfaces 因此常用方法不必在两种类型中都列出：

type Mammal interface {
    GetID() int
    GetName() string
}

type Human interface {
    Mammal

    GetHairColor() string
}

下面是基于您使用 embedded types（结构）的代码实现的这些接口：

type MammalImpl struct {
    ID   int
    Name string
}

func (m MammalImpl) GetID() int {
    return m.ID
}

func (m MammalImpl) GetName() string {
    return m.Name
}

type HumanImpl struct {
    MammalImpl
    HairColor string
}

func (h HumanImpl) GetHairColor() string {
    return h.HairColor
}

但是当然在你的 Count() 函数中你只能引用方法而不是实现的字段：

IDs[i] = strconv.Itoa(m.GetID())  // Access ID via the method: GetID()

并创建您的哺乳动物和人类切片：

mammals := []Mammal{
    MammalImpl{1, "Carnivorious"},
    MammalImpl{2, "Ominivorious"},
}

humans := []Mammal{
    HumanImpl{MammalImpl: MammalImpl{ID: 1, Name: "Peter"}, HairColor: "Black"},
    HumanImpl{MammalImpl: MammalImpl{ID: 2, Name: "Paul"}, HairColor: "Red"},
}

这是 Go Playground 上的完整工作代码。

Answer 2

在 Go 中，您无法完全按照您的要求进行操作。在 Go 中最接近的做事方式如下面的代码所示：

type Ids interface{
  Id() int
}

func (this Mammal) Id() int{
  return this.ID
} 

func (this Human) Id() int{
  return this.ID
} 


func Count(ms []Ids) *[]string {
...
    IDs[i] = strconv.Itoa(int(m.Id()))
...
}

func main(){
  mammals := []Ids{
    Mammal{1, "Carnivorious"},
    Mammal{2, "Ominivorious"},
  }

  humans := []Ids{
    Human{ID:1, Name: "Peter", HairColor: "Black"},
    Human{ID:2, Name: "Paul", HairColor: "Red"},
  }
  ...
}

这是工作example

Golang 中的泛型方法参数

Generic Method Parameters in Golang

generics

struct

interface

go