Golang - [...]接口的总和{}

Cle*_*son 5 types interface go 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
}
Run Code Online (Sandbox Code Playgroud)

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

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

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

34:9: invalid operation: sum += v (mismatched types int and interface {})
Run Code Online (Sandbox Code Playgroud)

编译期间。

鉴于我们知道它们是数字,有谁知道如何添加泛型类型?

谢谢!

icz*_*cza 7

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

为此,您可以使用类型断言。看这个例子:

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)
Run Code Online (Sandbox Code Playgroud)

输出(在Go Playground上尝试):

Not int: invalid
Sum: 6
Run Code Online (Sandbox Code Playgroud)

上面的例子只处理int数字,不处理其他的。如果你想“支持”多种数字类型,更方便的方法是使用类型开关

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)
Run Code Online (Sandbox Code Playgroud)

输出是一样的。在Go Playground上试试这个。