Golang で switch 文による型ごとの処理

Question

質問したいこと

interface を引数に持つ関数の内部で、switch 文を使って引数の型ごとに処理を分けるときにどのようにしたら簡潔にコードをかけるかわからない

環境

• Ubuntu 18.04
• go 1.14.2

問題の詳細

挿入ソートのコードを go を使って書いています。

package insertsort

func insertSort(x interface{}, size int) {
	switch value := x.(type) {
	case *[]int:
		for i := 0; i < size; i++ {
			tmp := (*value)[i]
			j := i - 1

			for j >= 0 {
				if tmp < (*value)[j] {
					(*value)[j+1] = (*value)[j]
					(*value)[j] = tmp
					j--
				} else {
					break
				}
			}
		}
	case *[]string:
		for i := 0; i < size; i++ {
			tmp := (*value)[i]
			j := i - 1

			for j >= 0 {
				if tmp < (*value)[j] {
					(*value)[j+1] = (*value)[j]
					(*value)[j] = tmp
					j--
				} else {
					break
				}
			}
		}
	default:

	}
}

引数には整数の配列や文字列の配列などを許容したくて interface{} にしています。中の処理では、型ごとに switch を使って分岐しているのですが、 *[]int のブランチと *[]string のブランチで中身は全く同じになってしまっています。

これを解決するために分岐の部分を

case *[]int, *[]string:

のように書いてみたのですが、invalid indirect of value (type interface {}) というエラーが出てしまいます。

この書き方だと上のブランチのほかに *[]float64 などどんどん分岐を増やしていくと重複だらけのコードになってしまうので困っているのですが、どのように書くのが正しいのでしょうか？

よろしくお願いいたします。

Answer 1

Goでのオススメの考え方は「必要なものだけを実装する」です。

標準のソート以外が必要になる型は実際には限られるはずです。
必要だとわかってるものだけを実装するのがGoの流儀かなと思います。

実際にやってみたことがあるけれど結局実装を複雑に膨らませただけでいいことは何もありませんでした。
使われるかわからない実装をして、そのためのテストを組むのはもったいないですしね。

それでも複数の型に対応する実装を展開する理由ができた場合、僕の場合はgo generateを活用します。

追記

現時点での僕のオススメの実装例を以下に示します。

以下のツールをインストールします。

go get github.com/cheekybits/genny

以下の２つのコードをメインパッケージにおきます
（必要ならパッケージを分けても構いません）

package main

import "github.com/cheekybits/genny/generic"

//go:generate genny -in=$GOFILE -out=gen-$GOFILE gen "Type=string,int"

// Type ...
type Type generic.Number

// SortType ...
func SortType(s []Type) {
	size := len(s)
	for i := 0; i < size; i++ {
		tmp := s[i]
		j := i - 1

		for j >= 0 {
			if tmp < s[j] {
				s[j+1] = s[j]
				s[j] = tmp
				j--
			} else {
				break
			}
		}
	}
}

package main

import "fmt"

// Sort ...
func Sort(s interface{}) {
	switch v := s.(type) {
	case []int:
		SortInt(v)
	case []string:
		SortString(v)
	}
}

func main() {
	is := []int{3, 2, 5}
	Sort(is)
	fmt.Println(is)
	ss := []string{"oge", "moge", "hoge"}
	Sort(ss)
	fmt.Println(ss)
}

以下のようにgo-generateを行えば、
SortInt、SortStringという関数定義が自動生成されます。

go generate .

以下のコマンドで実行したりできます

go run .

Comments

[nobonobo]

もちろん来年以降に使えるようになるかもしれないジェネリクス 機能を待ってもいいわけですが。

[Yhaya]

nobonoboさん、回答していただきありがとうございます。就活のコード面接対策でアルゴリズムを一通りさらい直していたところ、ふと複数の型への拡張は Go だとどうするのだろうと思って質問してみましたが、言われてみれば nobonobo さんの言うとおりですね。 go generate は初めて知りました。ありがとうございます。

[nobonobo]

go-generateの例を追記しました。

質問に書かれたコードでひとつ指摘するとすれば、
Goのスライスの値の実体は参照型なので僕の書いた実装例のように「スライス」を渡すだけでソートを実装することができます。
（スライスを渡した引数の内容を変更したとき、渡す前のスライスにも変更が反映されます。）
わざわざ「スライスへのポインタ」という概念を利用する必要はありません。
（ただし、要素数に関わる情報は値渡しなので、要素が増えたり減ったりする場合は「スライスへのポインタ」という概念が必要になる場合もあります）

[Yhaya]

ありがとうございます。go-generate 試してみました。スライスの関数への渡し方までご指摘いただきありがとうございます。