はじめに

[]int でも []string でも、どのような型の配列でも、どのような型のスライスでも、引数として受け取れる関数を作りたいことがありました。
作ろうとしたら一筋縄ではいかなかったので、その方法を書き留めておきます。

環境

terminal

% go version
go version go1.17.6 darwin/amd64

失敗例

はじめに、引数の型に interface{} を使うとなんでも受け取れるので []interface{} を使えばどのような型の配列・スライスでも受け取れるだろうと思い、以下のコードを書きました。

main.go

package main

import (
    "fmt"
)

func printElements(array []interface{}) { // []int や []string など、配列・スライスならなんでも受け取りたい
    for _, e := range array {
        fmt.Println(e)
    }
}

func main() {
    numbers := []int{2, 3, 5, 7, 11}
    printElements(numbers)

    texts := []string{"foo", "bar", "baz", "qux", "quux"}
    printElements(texts)
}

実行結果

./main.go:15:15: cannot use numbers (type []int) as type []interface {} in argument to printElements
./main.go:18:15: cannot use texts (type []string) as type []interface {} in argument to printElements

しかし、これを実行しようとするとコンパイルに失敗します。
どうやら []interface{} 型の引数は interface{} 型の配列・スライスしか受け取れないようです。

解決方法

やっていることの本質は同じですが、一応2パターンあるので分けて記載します。

引数の型に `interface{}` を使用する

どのような型の配列・スライスでも受け取れるように、引数の型をなんでも受け取れる interface{} にします。
そして、リフレクションを使用して引数の値を判別し、配列またはスライスでない場合はエラーを返すようにします。

main.go

package main

import (
    "fmt"
    "reflect"
)

func printElements(array interface{}) error {
    arr := reflect.ValueOf(array)
    if arr.Kind() != reflect.Array && arr.Kind() != reflect.Slice {
        return fmt.Errorf(`"%#v" is not array`, array)
    }

    for i := 0; i < arr.Len(); i++ {
        fmt.Println(arr.Index(i))
    }

    return nil
}

func main() {
    numbers := []int{2, 3, 5, 7, 11}
    _ = printElements(numbers)

    texts := []string{"foo", "bar", "baz", "qux", "quux"}
    _ = printElements(texts)
}

実行結果

2
3
5
7
11
foo
bar
baz
qux
quux

配列・スライスを `[]interface{}` 型に変換する

まず、どのような型の配列・スライスでも []interface{} 型に変換する関数を用意します。
この関数はリフレクションを使用して引数の値を判別し、配列またはスライスでない場合にエラーを返します。
そうしたら、行いたい処理は []interface{} 型を引数にとる関数として実装します。

main.go

package main

import (
    "fmt"
    "reflect"
)

func convert(array interface{}) ([]interface{}, error) {
    arr := reflect.ValueOf(array)
    if arr.Kind() != reflect.Array && arr.Kind() != reflect.Slice {
        return nil, fmt.Errorf(`"%#v" is not array`, array)
    }

    res := make([]interface{}, arr.Len())
    for i := 0; i < arr.Len(); i++ {
        res[i] = arr.Index(i).Interface()
    }

    return res, nil
}

func printElements(array []interface{}) {
    for _, e := range array {
        fmt.Println(e)
    }
}

func main() {
    numbers := []int{2, 3, 5, 7, 11}
    ns, _ := convert(numbers)
    printElements(ns)

    texts := []string{"foo", "bar", "baz", "qux", "quux"}
    ts, _ := convert(texts)
    printElements(ts)
}

実行結果

2
3
5
7
11
foo
bar
baz
qux
quux

おわりに

これで便利にどんな配列・スライスでも扱えますね。
リフレクション様様です。

関連