58. ジェネリックな型 — 型安全なスタック

型にも型パラメータを持たせられます。type Stack[T any] struct { items []T } のように書くと、要素型をあとから決められる汎用コンテナになります。Stack[int] と Stack[string] は別々の型として扱われ、それぞれ型チェックが効きます。

メソッドのレシーバには、型を宣言したときと同じ型パラメータを書きます(func (s *Stack[T]) Push(v T))。メソッド自身に新しい型パラメータは追加できない点に注意してください。これにより interface{} を使った汎用コンテナと違い、取り出すときに型アサーションが不要になります。

下の例では int 専用のスタックを作って Push / Pop します。型が固定されているので、誤った型を入れようとするとコンパイルエラーになります。出力は操作順どおりで安定します。

▶ サンプルコードを実行してみよう(自由に書き換えてOK)

package main

import "fmt"

type Stack[T any] struct {
	items []T
}

func (s *Stack[T]) Push(v T) {
	s.items = append(s.items, v)
}

func (s *Stack[T]) Pop() (T, bool) {
	var zero T
	if len(s.items) == 0 {
		return zero, false
	}
	last := s.items[len(s.items)-1]
	s.items = s.items[:len(s.items)-1]
	return last, true
}

func main() {
	var s Stack[int]
	s.Push(1)
	s.Push(2)
	v, ok := s.Pop()
	fmt.Println("Pop =", v, ok) // 2 true
}

公式ドキュメントで詳しく ↗