Golang-接口的总和{}

时间:2019-02-27 08:49:50

标签: go types interface type-assertion

我创建了一个通用的数据结构,并在Golang中添加了名称和通用数组。

package main

import "fmt"

type NamedArray struct {
  Name string
  values []interface{}
}

func main() {
  data := [...]int{1, 2, 3, 4, 5}
  interfaced_data := make([]interface{}, len(data))
  for i, v := range data{
    interfaced_data[i] = v
  }
  int_arr := NamedArray{Name: "Int Array", values: interfaced_data}
  fmt.Println(int_arr)
  // fmt.Println(int_arr.Sum()) -- uncomment to run Sum

  data_float := [...]float64{0.1, 0.2, 0.3, 0.4, 0.5}
  interfaced_data_float := make([]interface{}, len(data_float))
  for i, v := range data_float{
    interfaced_data_float[i] = v
  }
  float_arr := NamedArray{Name: "Float Array", values: interfaced_data_float}
  fmt.Println(float_arr)
  // fmt.Println(int_arr.Sum()) -- uncomment to run Sum
}

现在,我想定义一个方法,允许我对数组中的所有值求和。我知道它们是数字的(尽管它们是int还是float取决于上下文),但是我遇到了一些严重的麻烦。

func (arr NamedArray) Sum() interface{} {
  data := arr.values
  sum := 0
  for i, v := range data {
    sum += v
  }
  return sum
}

不过,我似乎无法完成这项工作。当我取消注释第18和27行(fmt.Println(int_arr.Sum()fmt.Println(int_arr.Sum())并尝试运行代码时,我得到

34:9: invalid operation: sum += v (mismatched types int and interface {})

在编译期间。

假设我们知道数字类型,有人知道如何添加泛型类型吗?

谢谢!

1 个答案:

答案 0 :(得分:1)

+运算符未在类型interface{}的值上定义。您必须先从int个值中获取一个类型为interface{}的值,然后才能将其用作数字。

为此,您可以使用type assertion。参见以下示例:

s := []interface{}{1, 2, 3, "invalid"}

sum := 0
for _, v := range s {
    if i, ok := v.(int); ok {
        sum += i
    } else {
        fmt.Println("Not int:", v)
    }
}
fmt.Println("Sum:", sum)

输出(在Go Playground上尝试):

Not int: invalid
Sum: 6

上面的示例仅处理int数字,而没有其他内容。如果要“支持”多种数字类型,一种更方便的方法是使用type switch

s := []interface{}{1, int32(2), int8(3), "invalid"}

sum := 0
for _, v := range s {
    switch i := v.(type) {
    case int:
        sum += i
    case int32:
        sum += int(i)
    case int8:
        sum += int(i)
    default:
        fmt.Println("Not int:", v)
    }
}
fmt.Println("Sum:", sum)

输出是相同的。在Go Playground上尝试这个。