1. ホーム
  2. スクリプト・コラム
  3. ゴラン

Golangのリフレクションを使った構造体のエレガントなソート機能

2022-02-14 15:21:20

私は最近インターンシップを始め、PythonとGolangをメインの技術スタックとして仕事をしています。コード構造は同じなのですが、比較項目が違うので、受け入れられません。

関数を比較します。

func reflectCmp(i, j interface{}, fieldName string) bool { // sort by incoming fieldName
	valI := reflect.ValueOf(i).FieldByName(fieldName).Interface()
	ValueOf(j).FieldByName(fieldName).Interface()
	switch s := valI.(type) {
	case string:
		return s < valJ.(string)
	case float64:
		return s < valJ.(float64)
	case int:
		return s < valJ.(int)
	default:
		fmt.Println("The type is unknown")
	}
	return true
}

比較したいタイプをスイッチに追加するだけです。

という構造体があるとします。

type person struct {
	Age int
	Name string
Name string}

では、どのように仕分けるのでしょうか?

func main() {
	pSlice := []person{{20,"a"},{15,"be"},{32,"cc"},{2,"aa"}}
	var p person
	typeOfA := reflect.TypeOf(p)
 
	// Iterate through all members of the structure
	for i := 0; i < typeOfA.NumField(); i++ {
		// Get the structure field type for each member
		fieldType := typeOfA.Field(i)
		sort.Slice(pSlice,func(i, j int) bool {
			return reflectCmp(pSlice[i],pSlice[j],fieldType.Name)
		})
		fmt.Println(pSlice)
	}
}

 結果を実行する

リフレクションを使えば、構造体がいくつのフィールドを持っていても、for sortで優雅にソートすることができるのです。

Golangで構造体をエレガントにソートするためにリフレクションを使用する方法についてのこの記事は終わりです。Golangで構造体をリフレクションする方法については、Script Houseの過去の記事を検索するか、以下の記事を引き続きご覧ください。